aboutsummaryrefslogtreecommitdiffstats
path: root/api/app/api/deps.py
diff options
context:
space:
mode:
Diffstat (limited to 'api/app/api/deps.py')
-rw-r--r--api/app/api/deps.py18
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