line_protocol: handle py2/py3 difference in _get_unicode().

anonymouzz · anonymouzz · commit f132aae9ce00 · 2016-02-17T10:34:01.000+07:00
Fix test_make_lines_unicode() test. _get_unicode() with flag fails
under py27 if tag contains unicode.
diff --git a/influxdb/line_protocol.py b/influxdb/line_protocol.py
@@ -7,7 +7,7 @@
 from numbers import Integral
 
 from dateutil.parser import parse
-from six import binary_type, text_type, integer_types
+from six import binary_type, text_type, integer_types, PY2
 
 
 def _convert_timestamp(timestamp, precision=None):
@@ -74,7 +74,10 @@ def _get_unicode(data, force=False):
     elif data is None:
         return ''
     elif force:
-        return str(data)
+        if PY2:
+            return unicode(data)
+        else:
+            return str(data)
     else:
         return data
 
