set migrate in manage.py

flask/manage-example.py

@@ -0,0 +1,23 @@
+from flask.cli import FlaskGroup
+
+from project import app, db, User
+
+
+cli = FlaskGroup(app)
+
+
+@cli.command("create_db")
+def create_db():
+    db.drop_all()
+    db.create_all()
+    db.session.commit()
+
+
+@cli.command("seed_db")
+def seed_db():
+    db.session.add(User(email="daniel@deflax.net"))
+    db.session.commit()
+
+
+if __name__ == "__main__":
+    cli()

flask/manage.py

@@ -1,10 +1,16 @@
+#!/usr/bin/env python
+
+import os
+import subprocess, shlex
+from forest import app, db, User
+
 from flask.cli import FlaskGroup
-
-from project import app, db, User
-
+from flask_migrate import Migrate, MigrateCommand
 
 cli = FlaskGroup(app)
+migrate = Migrate(app, db)
 
+cli.add_command('db', MigrateCommand)
 
 @cli.command("create_db")
 def create_db():
@@ -12,12 +18,32 @@ def create_db():
     db.create_all()
     db.session.commit()
 
-
 @cli.command("seed_db")
 def seed_db():
     db.session.add(User(email="daniel@deflax.net"))
     db.session.commit()
 
+@cli.command("upgrade_db")
+def upgrade_db():
+    """Run deployment tasks."""
+    from flask_migrate import upgrade
+    from app.models import Role, User
+
+    # migrate database to latest revision
+    upgrade()
+
+    # create user roles
+    Role.insert_roles()
+
+@cli.command("restore_db")
+def restore_db(restore_file):
+    """ recreate db from grid export with python3 manage.py restore /path/grid.tar.bz2 """
+    print(str(restore_file))
+    #TODO
+    from app.models import User
+    db.session.add(User(email=str(user), password=str(password), confirmed=True, confirmed_on=datetime.datetime.now()))
+    db.session.commit()
 
 if __name__ == "__main__":
     cli()
+