db like in tutorial

2021-09-29 11:05:09 +02:00 · 2021-09-29 11:05:09 +02:00 · 6b4c8b69b8
parent b5710b3dfb
commit 6b4c8b69b8
1 changed files with 8 additions and 0 deletions
--- a/papi/main.py
+++ b/papi/main.py
@ -19,6 +19,14 @@ notifications = defaultdict(list)
 Base.metadata.create_all(bind=engine)


+def get_db():
+    db = SessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+
 class Notifier:
    @staticmethod
    def __call__(idsonde, changes):