diff --git a/papi/main.py b/papi/main.py index fc67284..a1fc243 100644 --- 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):