1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
|
# noinspection PyMethodMayBeStatic
class _BaseService(object):
def __init__(self, api_client=None):
"""Init common services operation."""
if api_client is None:
raise ValueError("Invalid value for `api_client`, must be defined.")
self.api_client = api_client
self._build_type = None
def _check_operation_params(self, operation_id, supported_params, local_params):
supported_params.append('async_req')
supported_params.append('_return_http_data_only')
supported_params.append('_preload_content')
supported_params.append('_request_timeout')
supported_params.append('urlopen_kw')
for key, val in local_params['kwargs'].items():
if key not in supported_params:
raise TypeError(
f"Got an unexpected keyword argument '{key}'"
f" to method {operation_id}"
)
local_params[key] = val
del local_params['kwargs']
def _is_cloud_instance(self) -> bool:
if not self._build_type:
self._build_type = self.build_type()
return 'cloud' in self._build_type.lower()
async def _is_cloud_instance_async(self) -> bool:
if not self._build_type:
self._build_type = await self.build_type_async()
return 'cloud' in self._build_type.lower()
def build_type(self) -> str:
"""
Return the build type of the connected InfluxDB Server.
:return: The type of InfluxDB build.
"""
from influxdb_client import PingService
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')
async def build_type_async(self) -> str:
"""
Return the build type of the connected InfluxDB Server.
:return: The type of InfluxDB build.
"""
from influxdb_client import PingService
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 response_header(self, response, header_name='X-Influxdb-Version') -> str:
if response is not None and len(response) >= 3:
if header_name in response[2]:
return response[2][header_name]
return "unknown"
|