feat: handle URL path prefix in InfluxDB

Author: NguyenHoangSon96

influxdb_client_3/__init__.py

@@ -266,12 +266,16 @@ def __init__(
         hostname = parsed_url.hostname if parsed_url.hostname else host
         port = parsed_url.port if parsed_url.port else 443
 
+        path = ""
+        if parsed_url.scheme is not None and parsed_url.hostname is not None and parsed_url.port is not None:
+            path = parsed_url.path
+
         # Construct the clients using the parsed values
         if write_port_overwrite is not None:
             port = write_port_overwrite
 
         self._client = _InfluxDBClient(
-            url=f"{scheme}://{hostname}:{port}",
+            url=f"{scheme}://{hostname}:{port}{path}",
             token=self._token,
             org=self._org,
             timeout=write_timeout,

tests/test_influxdb_client_3.py

@@ -352,6 +352,33 @@ def test_get_version_fail(self):
                 host=f'http://{server.host}:{server.port}', org="ORG", database="DB", token="TOKEN"
             ).get_server_version()
 
+    def test_url_with_path_prefix(self):
+        server = self.http_server
+        server.expect_request('/prefix/ping').respond_with_json(
+            response_json={"version": "3.0"},
+        )
+        version = InfluxDBClient3(
+            host=f'http://{server.host}:{server.port}/prefix',
+            org="ORG",
+            database="DB",
+            token="TOKEN"
+        ).get_server_version()
+        assert version == "3.0"
+
+    def test_url_lack_path_prefix(self):
+        server = self.http_server
+        server.expect_request('/prefix/ping').respond_with_json(
+            response_json={"version": "3.0"},
+        )
+        try:
+            InfluxDBClient3(
+                host=f'http://{server.host}:{server.port}',
+                org="ORG",
+                database="DB",
+                token="TOKEN"
+            ).get_server_version()
+        except ApiException:
+            self.assertRaises(ApiException)
 
 if __name__ == '__main__':
     unittest.main()