flask flask-script自定义指令(flask 50)

from flask import Flask,render_template from views.admin import admin_bp from views.auth import auth_bp from views.blog import blog_bp from settings import config from extensions import bootstrap,db,login_manager,csrf,ckeditor,mail,moment,toolbar,migrate import os from flask_wtf.csrf import CSRFError from fakes import fake_admin, fake_categories, fake_posts, fake_comments from flask_script import Manager

app = Flask(name) basedir = os.path.abspath(os.path.dirname(file)) manager=Manager(app)

def create_app(config_name=None): if config_name is None: config_name = os.getenv('FLASK_CONFIG', 'development')

app.config.from_object(config[config_name])
app.jinja_env.trim_blocks = True
app.jinja_env.lstrip_blocks = True

register_blueprints(app)
register_extensions(app)
register_errors(app)
register_commands(app)
return app

def register_blueprints(app): app.register_blueprint(blog_bp) app.register_blueprint(admin_bp) app.register_blueprint(auth_bp)

def register_extensions(app): bootstrap.init_app(app) db.init_app(app) login_manager.init_app(app) csrf.init_app(app) ckeditor.init_app(app) mail.init_app(app) moment.init_app(app) toolbar.init_app(app) migrate.init_app(app, db)

def register_errors(app): @app.errorhandler(400) def bad_request(e): return render_template('errors/400.html'),400

@app.errorhandler(404)
def page_not_found(e):
    return render_template('errors/404.html'), 404

@app.errorhandler(500)
def internal_server_error(e):
    return render_template('errors/500.html'), 500

@app.errorhandler(CSRFError)
def handle_csrf_error(e):
    return render_template('errors/400.html', description=e.description), 400

def register_commands(app): @manager.option('-d','--drop',dest='drop',default=Flask,help='create after drop.') def initdb(drop): if drop: db.drop_all() print('Drop tables') db.create_all() print('Initialized database')

@manager.option('--category', default=10, help="Quantity of categories,default is 10")
@manager.option('--post', default=50, help='Quantity of posts, default is 50.')
@manager.option('--comment', default=500, help='Quantity of comments, default is 500.')
def forge(category, post, comment):
    db.drop_all()
    db.create_all()

    print('Generating the administrator...')
    fake_admin()

    print('Generating %d categories...' % category)
    fake_categories(category)

    print('Generating %d posts...' % post)
    fake_posts(post)

    print('Generating %d comments...' % comment)
    fake_comments(comment)

    print('Generating links...')
    #fake_links()

    print('Done.')

if name == 'main': create_app() manager.run()

python app.py forge