fix: formatting nanoseconds to Flux AST (influxdata#457)

bednar · web-flow · commit 465476b33648 · 2022-06-14T13:43:13.000+02:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -3,6 +3,9 @@
 ### Features
 1. [#440](https://github.com/influxdata/influxdb-client-python/pull/440): Add possibility to specify timestamp column and its timezone [DataFrame]
 
+### Bug Fixes
+1. [#457](https://github.com/influxdata/influxdb-client-python/pull/457): Formatting nanoseconds to Flux AST
+
 ### Dependencies
 1. [#449](https://github.com/influxdata/influxdb-client-python/pull/449): Remove `pytz` library
 
diff --git a/influxdb_client/client/_base.py b/influxdb_client/client/_base.py
@@ -341,7 +341,9 @@ def _parm_to_extern_ast(value) -> Union[Expression, None]:
             return FloatLiteral("FloatLiteral", value)
         elif isinstance(value, datetime):
             value = get_date_helper().to_utc(value)
-            return DateTimeLiteral("DateTimeLiteral", value.strftime('%Y-%m-%dT%H:%M:%S.%fZ'))
+            nanoseconds = getattr(value, 'nanosecond', 0)
+            fraction = f'{(value.microsecond * 1000 + nanoseconds):09d}'
+            return DateTimeLiteral("DateTimeLiteral", value.strftime('%Y-%m-%dT%H:%M:%S.') + fraction + 'Z')
         elif isinstance(value, timedelta):
             _micro_delta = int(value / timedelta(microseconds=1))
             if _micro_delta < 0:
diff --git a/tests/test_QueryApi.py b/tests/test_QueryApi.py
@@ -132,7 +132,7 @@ def test_parameter_ast(self):
                         },
                         "init": {
                             "type": "DateTimeLiteral",
-                            "value": "2021-03-20T15:59:10.607352Z"
+                            "value": "2021-03-20T15:59:10.607352000Z"
                         },
                         "type": "VariableAssignment"
                     },
@@ -150,7 +150,7 @@ def test_parameter_ast(self):
                         },
                         "init": {
                             "type": "DateTimeLiteral",
-                            "value": "2021-03-20T15:59:10.607352Z"
+                            "value": "2021-03-20T15:59:10.607352000Z"
                         },
                         "type": "VariableAssignment"
                     },
@@ -514,6 +514,22 @@ def test_profiler_mock(self):
                            'available': 5727718400, 'free': 35330048,
                            'used': 11452150784})
 
+    def test_time_to_ast(self):
+        from influxdb_client.extras import pd
+        import dateutil.parser
+
+        literals = [
+            (pd.Timestamp('1996-02-25T21:20:00.001001231Z'), '1996-02-25T21:20:00.001001231Z'),
+            (dateutil.parser.parse('1996-02-25T21:20:00.001001231Z'), '1996-02-25T21:20:00.001001000Z'),
+            (dateutil.parser.parse('1996-02-25'), '1996-02-25T00:00:00.000000000Z'),
+            (datetime.datetime(2021, 5, 24, 8, 40, 44, 785000, tzinfo=timezone.utc), '2021-05-24T08:40:44.785000000Z'),
+        ]
+
+        for literal in literals:
+            ast = QueryApi._build_flux_ast({'date': literal[0]})
+            self.assertEqual('DateTimeLiteral', ast.body[0].assignment.init.type)
+            self.assertEqual(literal[1], ast.body[0].assignment.init.value)
+
 
 if __name__ == '__main__':
     unittest.main()
