Database saving issue for docker container fix

1 files changed, 7 insertions, 1 deletions

diff --git a/app/db.py b/app/db.py
index 4d41dcc..454baae 100644
--- a/app/db.py
+++ b/app/db.py
@@ -1,6 +1,12 @@
 import sqlalchemy
+import os
+
+try:
+    os.mkdir('data')
+except FileExistsError:
+    pass
+engine = sqlalchemy.create_engine('sqlite:///data/data.db')
 
-engine = sqlalchemy.create_engine('sqlite:///data.db')
 
 def init_db():
     with engine.begin() as conn: