diff options
Diffstat (limited to 'api/app/api/deps.py')
-rw-r--r-- | api/app/api/deps.py | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/api/app/api/deps.py b/api/app/api/deps.py new file mode 100644 index 0000000..473f9b4 --- /dev/null +++ b/api/app/api/deps.py @@ -0,0 +1,18 @@ +from collections.abc import Generator +from sqlmodel import Session +import redis + +from app.core.db import engine +from app.core.cache import pool +from app.core.netbox import nb + + +def get_db() -> Generator[Session, None, None]: + with Session(engine) as session: + yield session + +def get_cache(): + return redis.Redis(connection_pool=pool) + +def get_netbox(): + return nb |