[Amit|Ashwin] Made changes in make_kafka_safe methodto make it work i…

…n Kafka3
goibibo · Apr 24, 2018 · 4bdc1a6 · 4bdc1a6
1 parent 75a55e6
commit 4bdc1a6
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 7 deletions.
diff --git a/setup.py b/setup.py
@@ -28,7 +28,7 @@
     description="Messaging library ",
     long_description=README,
     keywords="apache kafka",
-    install_requires=['six', 'kafka-python', 'gevent'],
+    install_requires=['six', 'kafka', 'gevent'],
     classifiers=[
         "Development Status :: 3 - Alpha",
         "Intended Audience :: Developers",

diff --git a/woof/transactions.py b/woof/transactions.py
@@ -14,12 +14,8 @@ def make_kafka_safe(raw_data):
     This function was written to avoid non-unicode
     string data produced to Kafka
     """
-    if type(raw_data) != unicode:
-        raw_data = str(raw_data)
-        raw_data = raw_data.decode('utf-8')
-        return raw_data.encode('ascii', 'ignore')
-    else:
-        return raw_data.encode('ascii', 'ignore')
+    raw_data = str(raw_data)
+    return raw_data.encode('ascii', 'ignore')
 
 
 class TransactionLogger(object):