refactor: wip

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -189,6 +189,9 @@ def __init__(
             org=None,
             database=None,
             token=None,
+            auth_scheme=None,
+            enable_gzip=False,
+            gzip_threshold=None,
             write_client_options=None,
             flight_client_options=None,
             write_port_overwrite=None,
@@ -229,6 +232,8 @@ def __init__(
         :key urllib3.util.retry.Retry retries: Set the default retry strategy that is used for all HTTP requests
                                                except batching writes. As a default there is no one retry strategy.
         :key str query_timeout: int value used to set the client query API timeout in milliseconds.
+        :key str enable_gzip: TODO ???
+        :key str gzip_threshold: TODO ???
         :key str write_timeout: int value used to set the client write API timeout in milliseconds.
         :key bool write_accept_partial: allow partial writes when some lines fail.
         :key bool write_use_v2_api: route writes through /api/v2/write compatibility endpoint.
@@ -293,14 +298,19 @@ def __init__(
         if write_port_overwrite is not None:
             port = write_port_overwrite
 
-        self._client = _InfluxDBClient(
-            url=f"{scheme}://{hostname}:{port}",
+        # TODO point_settings??
+        # TODO enable_gzip and gzip_threshold be in WriteOptions
+        self._write_api = _WriteApi(
+            base_url=f"{scheme}://{hostname}:{port}",
             token=self._token,
+            bucket=self._database,
             org=self._org,
+            gzip_threshold=gzip_threshold,
+            enable_gzip=enable_gzip,
+            auth_scheme=auth_scheme,
             timeout=write_timeout,
-            **kwargs)
-
-        self._write_api = _WriteApi(influxdb_client=self._client, **self._write_client_options)
+            **self._write_client_options
+        )
 
         if query_port_overwrite is not None:
             port = query_port_overwrite
@@ -702,7 +712,6 @@ def close(self):
         """Close the client and clean up resources."""
         self._write_api.close()
         self._query_api.close()
-        self._client.close()
 
     def __enter__(self):
         return self