Chapter 17: Deploy command (17a)

flasky.py

@@ -8,7 +8,7 @@
 
 import sys
 import click
-from flask_migrate import Migrate
+from flask_migrate import Migrate, upgrade
 from app import create_app, db
 from app.models import User, Follow, Role, Permission, Post, Comment
 
@@ -61,4 +61,17 @@ def profile(length, profile_dir):
  from werkzeug.contrib.profiler import ProfilerMiddleware
  app.wsgi_app = ProfilerMiddleware(app.wsgi_app, restrictions=[length],
  profile_dir=profile_dir)
- app.run(debug=False)
+ app.run()
+
+
+@app.cli.command()
+def deploy():
+ """Run deployment tasks."""
+ # migrate database to latest revision
+ upgrade()
+
+ # create or update user roles
+ Role.insert_roles()
+
+ # ensure all users are following themselves
+ User.add_self_follows()