File: _base_service.py

package info (click to toggle)
python-influxdb-client 1.40.0-3
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 7,216 kB
  • sloc: python: 60,236; sh: 64; makefile: 53
file content (67 lines) | stat: -rw-r--r-- 2,517 bytes parent folder | download
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"