merge makedb.py into manage.py

cloverfeed · Oct 27, 2014 · 0d1a5e0 · 0d1a5e0
1 parent 3b02916
commit 0d1a5e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/makedb.py b/makedb.py
diff --git a/manage.py b/manage.py
@@ -1,15 +1,22 @@
 """
 Flask-script stuff
 """
-from flask.ext.migrate import MigrateCommand
+from flask.ext.migrate import MigrateCommand, stamp
 from flask.ext.script import Manager
 from app.factory import create_app
+from app.models import db
 
 def main():
     manager = Manager(create_app)
     manager.add_command('db', MigrateCommand)
     manager.add_option('-c', '--config', dest='config_file', required=False)
 
+    @manager.command
+    def create():
+        "Creates database tables from sqlalchemy models"
+        db.create_all()
+        stamp()
+
     manager.run()
 
 if __name__ == '__main__':