issue #55: database schema is created, if it does not exist

sphericalpm · Sep 1, 2019 · cadade3 · cadade3
1 parent 0f48e9c
commit cadade3
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/app/models.py b/app/models.py
@@ -97,3 +97,14 @@ def get_link(self):
 
 def get_random_jokes_from_db(number=1):
     return Joke.query.order_by(sqlalchemy.func.random()).limit(number)
+
+
+def check_database_schema_existence():
+    try:
+        test_user_id = 1
+        User.query.get(test_user_id)
+    except sqlalchemy.exc.OperationalError:
+        db.create_all()
+
+
+check_database_schema_existence()