You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
39 lines
1.1 KiB
39 lines
1.1 KiB
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
from flask.ext.script import Manager, Shell, Server
|
|
from {{cookiecutter.repo_name }} import models
|
|
from {{cookiecutter.repo_name }}.app import create_app
|
|
from {{cookiecutter.repo_name}}.models import db
|
|
|
|
env = os.environ.get("{{cookiecutter.repo_name | upper }}_ENV", 'prod')
|
|
app = create_app("{{cookiecutter.repo_name}}.settings.{0}Config"
|
|
.format(env.capitalize()), env)
|
|
|
|
manager = Manager(app)
|
|
TEST_CMD = "nosetests"
|
|
|
|
def _make_context():
|
|
'''Return context dict for a shell session so you can access
|
|
app, db, and models by default.
|
|
'''
|
|
return {'app': app, 'db': db, 'models': models}
|
|
|
|
@manager.command
|
|
def test():
|
|
'''Run the tests.'''
|
|
status = subprocess.call(TEST_CMD, shell=True)
|
|
sys.exit(status)
|
|
|
|
@manager.command
|
|
def createdb():
|
|
'''Create a database from the tables defined in models.py.'''
|
|
db.create_all()
|
|
|
|
manager.add_command("runserver", Server())
|
|
manager.add_command("shell", Shell(make_context=_make_context))
|
|
|
|
if __name__ == '__main__':
|
|
manager.run()
|
|
|