Remove the dict requirement when serializing AVRO

kwilcox · kwilcox · commit ac8b4b2f4a18 · 2016-12-12T15:55:39.000-05:00
diff --git a/confluent_kafka/avro/__init__.py b/confluent_kafka/avro/__init__.py
@@ -87,9 +87,9 @@ def produce(self, **kwargs):
         """
             Sends message to kafka by encoding with specified avro schema
             @:param: topic: topic name
-            @:param: value: A dictionary object
+            @:param: value: An object to serialize
             @:param: value_schema : Avro schema for value
-            @:param: key: A dictionary object
+            @:param: key: An object to serialize
             @:param: key_schema : Avro schema for key
             @:exception: SerializerError
         """
diff --git a/confluent_kafka/avro/serializer/message_serializer.py b/confluent_kafka/avro/serializer/message_serializer.py
@@ -85,12 +85,10 @@ def encode_record_with_schema(self, topic, schema, record, is_key=False):
         The schema is registered with the subject of 'topic-value'
         @:param topic : Topic name
         @:param schema : Avro Schema
-        @:param record : A dictionary object
+        @:param record : An object to serialize
         @:param is_key : If the record is a key
         @:returns : Encoded record with schema ID as bytes
         """
-        if not isinstance(record, dict):
-            raise SerializerError("record must be a dictionary")
         serialize_err = KeySerializerError if is_key else ValueSerializerError
 
         subject_suffix = ('-key' if is_key else '-value')
@@ -112,12 +110,10 @@ def encode_record_with_schema_id(self, schema_id, record, is_key=False):
         Encode a record with a given schema id.  The record must
         be a python dictionary.
         @:param: schema_id : integer ID
-        @:param: record : A dictionary object
+        @:param: record : An object to serialize
         @:param is_key : If the record is a key
         @:returns: decoder function
         """
-        if not isinstance(record, dict):
-            raise SerializerError("record must be a dictionary")
         serialize_err = KeySerializerError if is_key else ValueSerializerError
 
         # use slow avro
