account for python2/3 with six in ThriftSerializationContext pythonToThriftMap

This commit is contained in:
Michael James 2018-10-30 11:11:42 -06:00
parent 48d133a136
commit 486a1ca820

View file

@ -255,10 +255,12 @@ class ThriftSerializationContext(object):
pyt = type(obj)
if pyt in pythonToThriftMap:
return pythonToThriftMap[pyt]
elif pyt.__module__[:DS_LEN - 1] == 'dynamicserialize.dstypes':
elif pyt.__module__[:DS_LEN - 1] == ('dynamicserialize.dstypes'):
if six.PY2:
return pythonToThriftMap[types.InstanceType]
return pythonToThriftMap[object]
raise dynamicserialize.SerializationException(
"Don't know how to serialize object of type: " + str(pyt))
"Don't know how to serialize object of type: " + str(pyt))
def serializeMessage(self, obj):
tt = self._lookupType(obj)