feat: add possibility to specify default timezone for datetimes without tzinfo

Author: bednar

influxdb_client/client/util/date_utils.py

@@ -10,6 +10,15 @@
 class DateHelper:
     """DateHelper to groups different implementations of date operations."""
 
+    def __init__(self, timezone: datetime.tzinfo = UTC) -> None:
+        """
+        Initialize defaults.
+
+        :param timezone: Default timezone used for serialization "datetime" without "tzinfo".
+                         Default value is "UTC".
+        """
+        self.timezone = timezone
+
     def parse_date(self, date_string: str):
         """
         Parse string into Date or Timestamp.
@@ -40,7 +49,7 @@ def to_utc(self, value: datetime):
         :return: datetime in UTC
         """
         if not value.tzinfo:
-            return UTC.localize(value)
+            return self.to_utc(value.replace(tzinfo=self.timezone))
         else:
             return value.astimezone(UTC)
 

tests/test_DateHelper.py

@@ -0,0 +1,28 @@
+# -*- coding: utf-8 -*-
+
+import unittest
+from datetime import datetime, timezone
+
+from pytz import UTC, timezone
+
+from influxdb_client.client.util import date_utils
+from influxdb_client.client.util.date_utils import DateHelper, get_date_helper
+
+
+class DateHelperTest(unittest.TestCase):
+
+    def setUp(self) -> None:
+        date_utils.date_helper = DateHelper()
+
+    def test_to_utc(self):
+        date = get_date_helper().to_utc(datetime(2021, 4, 29, 20, 30, 10, 0))
+        self.assertEqual(datetime(2021, 4, 29, 20, 30, 10, 0, UTC), date)
+
+    def test_to_utc_different_timezone(self):
+        date_utils.date_helper = DateHelper(timezone=timezone('ETC/GMT+2'))
+        date = get_date_helper().to_utc(datetime(2021, 4, 29, 20, 30, 10, 0))
+        self.assertEqual(datetime(2021, 4, 29, 22, 30, 10, 0, UTC), date)
+
+
+if __name__ == '__main__':
+    unittest.main()