Fix method definition for udt registration

This commit is contained in:
Tyler Hobbs
2014-07-23 15:37:11 -05:00
parent 38d757fafb
commit 8e63506c38

View File

@@ -1585,10 +1585,10 @@ class Session(object):
raise UserTypeDoesNotExist(
'User type %s does not exist in keyspace %s' % (user_type, keyspace))
def encode(encoder_self, val):
def encode(val):
return '{ %s }' % ' , '.join('%s : %s' % (
field_name,
encoder_self.cql_encode_all_types(getattr(val, field_name))
self.encoder.cql_encode_all_types(getattr(val, field_name))
) for field_name in type_meta.field_names)
self.encoder.mapping[klass] = encode