Switch from flask_script to click, add CLI unit tests.

manage.py

@@ -4,60 +4,73 @@ CLI to manage redash.
 """
 import json
 
-from flask_script import Manager
 
-from redash import settings, models, __version__
-from redash.wsgi import app
+import click
+from flask.cli import FlaskGroup, run_command
+
+from redash import create_app, settings, __version__
 from redash.cli import users, groups, database, data_sources, organization
 from redash.monitor import get_status
 
-manager = Manager(app)
-manager.add_command("database", database.manager)
-manager.add_command("users", users.manager)
-manager.add_command("groups", groups.manager)
-manager.add_command("ds", data_sources.manager)
-manager.add_command("org", organization.manager)
+
+def create(group):
+    app = create_app()
+    group.app = app
+    return app
 
 
+@click.group(cls=FlaskGroup, create_app=create)
+def manager():
+    "Management script for redash"
 
-@manager.command
+
+manager.add_command(database.manager, "database")
+manager.add_command(users.manager, "users")
+manager.add_command(groups.manager, "groups")
+manager.add_command(data_sources.manager, "ds")
+manager.add_command(organization.manager, "org")
+manager.add_command(run_command, "runserver")
+
+
+@manager.command()
 def version():
     """Displays re:dash version."""
     print __version__
 
-@manager.command
+
+@manager.command()
 def status():
     print json.dumps(get_status(), indent=2)
 
-@manager.command
+
+@manager.command()
 def runworkers():
     """Start workers (deprecated)."""
     print "** This command is deprecated. Please use Celery's CLI to control the workers. **"
 
 
-@manager.shell
-def make_shell_context():
-    from redash.models import db
-    return dict(app=app, db=db, models=models)
-
-
-@manager.command
+@manager.command()
 def check_settings():
     """Show the settings as re:dash sees them (useful for debugging)."""
     for name, item in settings.all_settings().iteritems():
         print "{} = {}".format(name, item)
 
 
-@manager.option('email', default=None, help="Email address to send test message to (default: the address you defined in MAIL_DEFAULT_SENDER)")
+@manager.command()
+@click.argument('email', default=settings.MAIL_DEFAULT_SENDER, required=False)
 def send_test_mail(email=None):
+    """
+    Send test message to EMAIL (default: the address you defined in MAIL_DEFAULT_SENDER)
+    """
     from redash import mail
     from flask_mail import Message
 
     if email is None:
         email = settings.MAIL_DEFAULT_SENDER
 
-    mail.send(Message(subject="Test Message from re:dash", recipients=[email], body="Test message."))
+    mail.send(Message(subject="Test Message from re:dash", recipients=[email],
+                      body="Test message."))
 
 
 if __name__ == '__main__':
-    manager.run()
+    manager()