ITQC/pydase_service_base
24
0
Fork 0
mirror of https://github.com/tiqi-group/pydase_service_base.git synced 2025-04-20 00:20:01 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix: optional database dependencies work without each other now

Browse Source
This commit is contained in:
Mose Mueller 2024-02-13 11:18:22 +01:00
parent f708b6d159
commit 44909f52d9
1 changed files with 31 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
pydase_service_base/database/__init__.py
View File

@ -1,4 +1,33 @@
from .influxdb_session import InfluxDBSession
from .postgres_session import PostgresDatabaseSession
class OptionalDependencyError(Exception):
"""Exception raised when an optional dependency is not installed."""
try:
import influxdb_client # type: ignore # noqa
from .influxdb_session import InfluxDBSession # type: ignore
except ImportError:
class InfluxDBSession: # type: ignore
def __init__(self) -> None:
raise OptionalDependencyError(
"InfluxDBSession requires the 'influxdbv2' extra. "
"Please refer to https://gitlab.phys.ethz.ch/tiqi-projects/qchub/icon-services/pydase_service_base."
)
try:
import sqlmodel # noqa
from .postgres_session import PostgresDatabaseSession # type: ignore
except ImportError:
class PostgresDatabaseSession: # type: ignore
def __init__(self) -> None:
raise OptionalDependencyError(
"PostgresDatabaseSession requires the 'postgresql' extra. "
"Please refer to https://gitlab.phys.ethz.ch/tiqi-projects/qchub/icon-services/pydase_service_base."
)
__all__ = ["InfluxDBSession", "PostgresDatabaseSession"]