Suffix byte arguments with _bytes

Author: blakeembrey

kafka/producer/kafka.py

@@ -496,14 +496,14 @@ def _max_usable_produce_magic(self):
         else:
             return 0
 
-    def _estimate_size_in_bytes(self, key, value, headers=[]):
+    def _estimate_size_in_bytes(self, key_bytes, value_bytes, headers=[]):
         magic = self._max_usable_produce_magic()
         if magic == 2:
             return DefaultRecordBatchBuilder.estimate_size_in_bytes(
-                key, value, headers)
+                key_bytes, value_bytes, headers)
         else:
             return LegacyRecordBatchBuilder.estimate_size_in_bytes(
-                magic, self.config['compression_type'], key, value)
+                magic, self.config['compression_type'], key_bytes, value_bytes)
 
     def send(self, topic, value=None, key=None, partition=None, timestamp_ms=None):
         """Publish a message to a topic.