From b36a4e3d225da89f0581648a8086f3c5294242c8 Mon Sep 17 00:00:00 2001 From: Mose Mueller Date: Mon, 5 May 2025 11:44:56 +0200 Subject: [PATCH] feat: adds verify_ssl to InfluxDBSession (v2) --- pydase_service_base/database/config.py | 1 + pydase_service_base/database/influxdb_session.py | 5 ++++- 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/pydase_service_base/database/config.py b/pydase_service_base/database/config.py index 65cb77a..f4c4b25 100644 --- a/pydase_service_base/database/config.py +++ b/pydase_service_base/database/config.py @@ -29,6 +29,7 @@ class InfluxDBConfig(BaseConfig): # type: ignore url: str org: str token: SecretStr + verify_ssl: bool = True headers: dict[str, str] = {} # noqa: RUF012 diff --git a/pydase_service_base/database/influxdb_session.py b/pydase_service_base/database/influxdb_session.py index 2c905ef..5365547 100644 --- a/pydase_service_base/database/influxdb_session.py +++ b/pydase_service_base/database/influxdb_session.py @@ -81,12 +81,15 @@ class InfluxDBSession: self.token = self._config.token.get_secret_value() self.org = self._config.org self.headers = self._config.headers + self.verify_ssl = self._config.verify_ssl self._client: InfluxDBClient self._write_api: WriteApi self._buckets_api: BucketsApi def __enter__(self) -> Self: - self._client = InfluxDBClient(url=self.url, token=self.token, org=self.org) + self._client = InfluxDBClient( + url=self.url, token=self.token, org=self.org, verify_ssl=self.verify_ssl + ) for header_name, header_value in self.headers.items(): self._client.api_client.set_default_header(header_name, header_value)