chore(line_protocol): update repr value of floats to properly handle precision (#813)

sebito91 · web-flow · commit 896f6237f921 · 2020-04-09T19:05:34.000-05:00
* chore(line_protocol): update repr value of floats to properly handle precision. Closes #488 * chore(line_protocol): fix repr and handle boolean values * chore(CHANGELOG): update to include reference to PR#488
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -25,6 +25,7 @@ and this project adheres to [Semantic Versioning](http://semver.org/spec/v2.0.0.
 - Update make_lines section in line_protocol.py to split out core function (#375 thx @aisbaa)
 - Fix nanosecond time resolution for points (#407 thx @AndreCAndersen && @clslgrnc)
 - Fix import of distutils.spawn (#805 thx @Hawk777)
+- Update repr of float values including properly handling of boolean (#488 thx @ghost)
 
 ### Removed
 
diff --git a/influxdb/line_protocol.py b/influxdb/line_protocol.py
@@ -112,8 +112,11 @@ def _escape_value(value):
     if isinstance(value, integer_types) and not isinstance(value, bool):
         return str(value) + 'i'
 
+    if isinstance(value, bool):
+        return str(value)
+
     if _is_float(value):
-        return repr(value)
+        return repr(float(value))
 
     return str(value)
 
diff --git a/influxdb/tests/test_line_protocol.py b/influxdb/tests/test_line_protocol.py
@@ -6,10 +6,12 @@
 from __future__ import print_function
 from __future__ import unicode_literals
 
-from datetime import datetime
 import unittest
-from pytz import UTC, timezone
 
+from datetime import datetime
+from decimal import Decimal
+
+from pytz import UTC, timezone
 from influxdb import line_protocol
 
 
@@ -166,3 +168,20 @@ def test_float_with_long_decimal_fraction(self):
             line_protocol.make_lines(data),
             'test float_val=1.0000000000000009\n'
         )
+
+    def test_float_with_long_decimal_fraction_as_type_decimal(self):
+        """Ensure precision is preserved when casting Decimal into strings."""
+        data = {
+            "points": [
+                {
+                    "measurement": "test",
+                    "fields": {
+                        "float_val": Decimal(0.8289445733333332),
+                    }
+                }
+            ]
+        }
+        self.assertEqual(
+            line_protocol.make_lines(data),
+            'test float_val=0.8289445733333332\n'
+        )
