updates Readme with new config options

README.md

@@ -32,12 +32,12 @@ To utilize specific functionalities such as `IonizerServer`, `InfluxDBv1Session`
 or directly add the following line to the `pyproject.toml` file:
 
 ```toml
-pydase-service-base = {git = "https://github.com/tiqi-group/pydase_service_base.git", rev = "main", extras = ["ionizer", "postgresql", "ionizer"]}
+pydase_service_base = {git = "https://github.com/tiqi-group/pydase_service_base.git", rev = "main", extras = ["ionizer", "postgresql", "influxdbv1", "influxdbv2"]}
 ```
 
 ## Configuration
 
-Database connections rely on configurations provided either through environment variables or a specific configuration file. The package anticipates a `database_config` folder in the root directory of any module using this package. This folder should house the configuration files for the databases. Override the `database_config` folder's location by passing a different path to the database classes' constructor.
+Database connections rely on configurations provided by environment variables or a specific configuration file. The package anticipates a `database_config` folder in the root directory of any module using this package. This folder should house the database configuration files. You can override the `database_config` folder's location by passing a different path to the database classes' constructor.
 
 Structure of the `database_config` folder:
 
@@ -60,6 +60,8 @@ password: root
 database: my_database
 ssl: True  # defaults to True
 verify_ssl: True  # defaults to True
+headers:
+  Host: other-virtual-host.ethz.ch
 ```
 
 `influxdb_config.yaml`:
@@ -67,6 +69,9 @@ verify_ssl: True  # defaults to True
 url: https://database-url.ch
 org: your-org
 token: <influxdb-token>
+verify_ssl: True  # defaults to True
+headers:
+  Host: other-virtual-host.ethz.ch
 ```
 
 `postgres_development.yaml` / `postgres_production.yaml`:
@@ -160,7 +165,7 @@ with PostgresDatabaseSession() as session:
 
 ### Ionizer Interface
 
-The `IonizerServer` provides an interface to seamlessly integrate `pydase` applications with Ionizer. **This requires the `ionizer` optional dependency**.
+The `IonizerServer` provides an interface to integrate `pydase` applications with Ionizer seamlessly. **This requires the `ionizer` optional dependency**.
 
 To deploy `IonizerServer` with your service:
 
@@ -190,7 +195,7 @@ if __name__ == "__main__":
 
 This integration ensures that your primary `pydase` server and `YourServiceClass` service are set up. It also incorporates the `IonizerServer` on port `8002`.
 
-For more details on the `IonizerServer`, refer to the [official documentation](https://pydase.readthedocs.io/en/latest/) or get in touch with the maintainers.
+For more details on the `IonizerServer`, get in touch with the maintainers.
 
 ## License
 

pydase_service_base/database/config.py

@@ -29,6 +29,8 @@ class InfluxDBConfig(BaseConfig):  # type: ignore
     url: str
     org: str
     token: SecretStr
+    verify_ssl: bool = True
+    headers: dict[str, str] = {}  # noqa: RUF012
 
 
 class InfluxDBv1Config(BaseConfig):  # type: ignore
@@ -39,3 +41,4 @@ class InfluxDBv1Config(BaseConfig):  # type: ignore
     database: str
     ssl: bool = True
     verify_ssl: bool = True
+    headers: dict[str, str] = {}  # noqa: RUF012

pydase_service_base/database/influxdb_session.py

@@ -1,16 +1,17 @@
 from __future__ import annotations
 
 import logging
+import sys
 from typing import TYPE_CHECKING, Any, NamedTuple
 
-try:
-    from typing import Self  # type: ignore
-except ImportError:
+if sys.version_info < (3, 11):
     from typing_extensions import Self
+else:
+    from typing import Self
 
 
 from confz import FileSource
-from influxdb_client import (
+from influxdb_client import (  # type: ignore
     Bucket,
     BucketRetentionRules,
     BucketsApi,
@@ -19,9 +20,9 @@ from influxdb_client import (
     WriteApi,
     WritePrecision,
 )
-from influxdb_client.client.write.point import DEFAULT_WRITE_PRECISION
-from influxdb_client.client.write_api import SYNCHRONOUS
-from influxdb_client.rest import ApiException
+from influxdb_client.client.write.point import DEFAULT_WRITE_PRECISION  # type: ignore
+from influxdb_client.client.write_api import SYNCHRONOUS  # type: ignore
+from influxdb_client.rest import ApiException  # type: ignore
 
 from pydase_service_base.database.config import InfluxDBConfig, ServiceConfig
 
@@ -79,12 +80,20 @@ class InfluxDBSession:
         self.url = self._config.url
         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)
+
         self._write_api = self._client.write_api(write_options=SYNCHRONOUS)  # type: ignore
         return self
 

pydase_service_base/database/influxdbv1_session.py

@@ -65,6 +65,7 @@ class InfluxDBv1Session:
         self._database = self._config.database
         self._ssl = self._config.ssl
         self._verify_ssl = self._config.verify_ssl
+        self._headers = self._config.headers
 
     def __enter__(self) -> Self:
         self._client = influxdb.InfluxDBClient(
@@ -75,6 +76,7 @@ class InfluxDBv1Session:
             database=self._database,
             ssl=self._ssl,
             verify_ssl=self._verify_ssl,
+            headers=self._headers,
         )
         return self
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pydase-service-base"
-version = "0.3.1"
+version = "0.3.2"
 description = "Repository storing the code that is common to all pydase services."
 authors = ["Mose Mueller <mosmuell@ethz.ch>"]
 license = "MIT"