refactor: write rest client

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -305,14 +305,12 @@ def __init__(
         # TODO fix retries
         retries = None
 
-        self.base_url = f"{scheme}://{hostname}:{port}"
-
-        # TODO refactor
         auth_schema = 'Token' if auth_scheme is None else auth_scheme
         default_header = {
             'Authorization': f'{auth_schema} {self._token}',
             'User-Agent': USER_AGENT
         }
+        self.base_url = f"{scheme}://{hostname}:{port}"
         self.default_header = default_header
         self.rest_client = rest.RestClient(
             base_url=self.base_url,
@@ -706,10 +704,9 @@ def get_server_version(self) -> Any | None:
             if key.lower() == "x-influxdb-version":
                 return value
 
-        # TODO refactor this
-        string = resp.get_string_body()
-        if string is not None:
-            return json.loads(string)['version']
+        string_body = resp.get_string_body()
+        if string_body is not None:
+            return json.loads(string_body)['version']
         return None
 
     def flush(self):

influxdb_client_3/write_client/_sync/rest_client.py

@@ -32,7 +32,6 @@ def getheader(self, name, default=None):
         """Return a given response header."""
         return self.urllib3_response.headers.get(name, default)
 
-    # TODO is this the best way?
     def get_string_body(self):
         string = self.urllib3_response.data.decode('utf-8')
         if string is None or string == '':

influxdb_client_3/write_client/client/write_api.py

@@ -70,10 +70,7 @@
 
 """Commons function for Sync and Async client."""
 
-# TODO should this file be refactored and remove?
 try:
-    # import dataclasses
-
     _HAS_DATACLASS = True
 except ModuleNotFoundError:
     _HAS_DATACLASS = False