Centralized db initialization in flask cli

Author: =

accessmapapi/__init__.py

@@ -6,7 +6,7 @@
 
 from . import auth
 from . import jwt
-from .models import db
+from .models import init_app as db_init_app
 from . import blueprints
 
 
@@ -30,12 +30,7 @@ def create_app():
     )
 
     # Attach database
-    db.init_app(app)
-
-    # Migrate database
-    # FIXME: move db creation stuff into a separate, trackable migration framework
-    with app.app_context():
-        db.create_all()
+    db_init_app(app)
 
     # Add oauth interface
     auth.init_app(app)

accessmapapi/db.py

@@ -1,4 +1,6 @@
+import click
 from flask import g, current_app
+from flask.cli import with_appcontext
 from flask_sqlalchemy import SQLAlchemy
 from werkzeug.local import LocalProxy
 from werkzeug.contrib.cache import FileSystemCache
@@ -7,6 +9,22 @@
 db = SQLAlchemy()
 
 
+def init_db():
+    db.create_all()
+
+
+@click.command("init-db")
+@with_appcontext
+def init_db_command():
+    init_db()
+    click.echo("Initialized the database.")
+
+
+def init_app(app):
+    app.cli.add_command(init_db_command)
+    db.init_app(app)
+
+
 class User(db.Model):
     user_id = db.Column(db.Integer, primary_key=True)