feat: add API to check build type of InfluxDB instance

Author: bednar

influxdb_client/client/_base.py

@@ -98,10 +98,10 @@ def __init__(self, url, token, debug=None, timeout=10_000, enable_gzip=False, or
         self.profilers = kwargs.get('profilers', None)
         pass
 
-    def _version(self, response) -> str:
+    def _response_header(self, response, header_name='X-Influxdb-Version') -> str:
         if response is not None and len(response) >= 3:
-            if 'X-Influxdb-Version' in response[2]:
-                return response[2]['X-Influxdb-Version']
+            if header_name in response[2]:
+                return response[2][header_name]
 
         return "unknown"
 

influxdb_client/client/influxdb_client.py

@@ -407,7 +407,19 @@ def version(self) -> str:
 
         response = ping_service.get_ping_with_http_info(_return_http_data_only=False)
 
-        return self._version(response)
+        return self._response_header(response)
+
+    def build(self) -> str:
+        """
+        Return the build type of the connected InfluxDB Server.
+
+        :return: The type of InfluxDB build.
+        """
+        ping_service = PingService(self.api_client)
+
+        response = ping_service.get_ping_with_http_info(_return_http_data_only=False)
+
+        return self._response_header(response, header_name='X-Influxdb-Build')
 
     def ready(self) -> Ready:
         """

influxdb_client/client/influxdb_client_async.py

@@ -251,7 +251,18 @@ async def version(self) -> str:
         ping_service = PingService(self.api_client)
 
         response = await ping_service.get_ping_async(_return_http_data_only=False)
-        return self._version(response)
+        return self._response_header(response)
+
+    async def build(self) -> str:
+        """
+        Return the build type of the connected InfluxDB Server.
+
+        :return: The type of InfluxDB build.
+        """
+        ping_service = PingService(self.api_client)
+
+        response = await ping_service.get_ping_async(_return_http_data_only=False)
+        return self._response_header(response, header_name='X-Influxdb-Build')
 
     def query_api(self, query_options: QueryOptions = QueryOptions()) -> QueryApiAsync:
         """

tests/test_InfluxDBClient.py

@@ -316,6 +316,10 @@ def test_version(self):
         version = self.client.version()
         self.assertTrue(len(version) > 0)
 
+    def test_build(self):
+        build = self.client.build()
+        self.assertEqual('oss', build.lower())
+
     def test_version_not_running_instance(self):
         client_not_running = InfluxDBClient("http://localhost:8099", token="my-token", debug=True)
         with self.assertRaises(NewConnectionError):

tests/test_InfluxDBClientAsync.py

@@ -49,6 +49,11 @@ async def test_version(self):
         version = await self.client.version()
         self.assertTrue(len(version) > 0)
 
+    @async_test
+    async def test_build(self):
+        build = await self.client.build()
+        self.assertEqual('oss', build.lower())
+
     def test_create_query_api(self):
         query_api = self.client.query_api()
         self.assertIsNotNone(query_api)