From 6b4c8b69b8f75fbba9494eb154166179e27cead5 Mon Sep 17 00:00:00 2001
From: Colin Goutte <cgte@bk.ru>
Date: Wed, 29 Sep 2021 11:05:09 +0200
Subject: [PATCH] db like in tutorial

---
 papi/main.py | 8 ++++++++
 1 file changed, 8 insertions(+)

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):