Flask-Migrate init

Flask-Migrate init

mpwo_api/mpwo_api/__init__.py

@@ -2,11 +2,13 @@ import logging
 
 from flask import Flask
 from flask_bcrypt import Bcrypt
+from flask_migrate import Migrate
 from flask_sqlalchemy import SQLAlchemy
 
 
 db = SQLAlchemy()
 bcrypt = Bcrypt()
+migrate = Migrate()
 appLog = logging.getLogger('mpwo_api')
 
 # instantiate the app
@@ -19,6 +21,7 @@ with app.app_context():
 # set up extensions
 db.init_app(app)
 bcrypt.init_app(app)
+migrate.init_app(app, db)
 
 from .users.auth import auth_blueprint  # noqa
 from .users.users import users_blueprint  # noqa

mpwo_api/requirements.txt

@@ -6,6 +6,7 @@ flake8-isort==2.2.2
 flake8-polyfill==1.0.1
 Flask==0.12.2
 Flask-Bcrypt==0.7.1
+Flask-Migrate==2.1.1
 Flask-SQLAlchemy==2.3.2
 Flask-Testing==0.6.2
 isort==4.2.15

mpwo_api/server.py

@@ -5,10 +5,15 @@ from mpwo_api.users.models import User
 
 
 @app.cli.command()
-def init_db():
-    """Init the database."""
+def drop_db():
     db.drop_all()
-    db.create_all()
+    db.session.commit()
+    print('Database dropped.')
+
+
+@app.cli.command()
+def init_data():
+    """Init the database."""
     admin = User(
         username='admin',
         email='admin@example.com',
@@ -16,7 +21,7 @@ def init_db():
     admin.admin = True
     db.session.add(admin)
     db.session.commit()
-    print('Database initialization done.')
+    print('Admin created.')
 
 
 def run_test(test_path='mpwo_api/tests'):