feat: refactor to enum

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -209,8 +209,8 @@ def __init__(
         self._database = database
         self._token = token
 
-        write_type = DefaultWriteOptions['write_type']
-        write_precision = DefaultWriteOptions['write_precision']
+        write_type = DefaultWriteOptions.write_type.value
+        write_precision = DefaultWriteOptions.write_precision.value
         gzip_threshold = None
         if isinstance(write_client_options, dict) and write_client_options.get('write_options') is not None:
             write_opts = write_client_options['write_options']

influxdb_client_3/write_client/client/write_api.py

@@ -41,10 +41,14 @@ class WriteType(Enum):
 DEFAULT_GZIP_THRESHOLD = 1000
 
 #todo: convert to enum
-DefaultWriteOptions = {
-    'write_type': WriteType.synchronous,
-    'write_precision': WritePrecision.NS
-}
+# DefaultWriteOptions = {
+#     'write_type': WriteType.synchronous,
+#     'write_precision': WritePrecision.NS
+# }
+class DefaultWriteOptions(Enum):
+    write_type = WriteType.synchronous
+    write_precision = WritePrecision.NS
+
 
 
 class WriteOptions(object):

tests/test_influxdb_client_3.py

@@ -75,8 +75,8 @@ async def test_query_async(self):
             assert {'data': 'query_type', 'reference': 'sql', 'value': -1.0} in result_list
 
     def test_default_client(self):
-        expected_precision = DefaultWriteOptions['write_precision']
-        expected_write_type = DefaultWriteOptions['write_type']
+        expected_precision = DefaultWriteOptions.write_precision.value
+        expected_write_type = DefaultWriteOptions.write_type.value
         expected_gzip_threshold = None
         expected_gzip_enabled = False