Add initdb cmd

authlib · Jan 7, 2018 · 88379e7 · 88379e7
1 parent 8987747
commit 88379e7
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/app.py b/app.py
@@ -14,8 +14,10 @@ def add_header(resp):
         resp.headers['Cache-Control'] = 'no-store'
         resp.headers['Pragma'] = 'no-cache'
         return resp
-
-    with app.app_context():
-        db.create_all()
 else:
     app = create_app()
+
+
+@app.cli.command()
+def initdb():
+    db.create_all()