fix: formatting nanoseconds to Flux AST

Author: bednar

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:

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()