run.py

import os
from app import create_app, db
from app.models import Category, Recipe, User

app = create_app(os.getenv('FLASK_ENV') or 'default')


@app.shell_context_processor
def make_shell_context():
    return dict(db=db, Category=Category, Recipe=Recipe, User=User)


@app.cli.command()
def init_db():
    db.create_all()
    
    categories = [
        Category(name='荤菜', description='肉类菜品'),
        Category(name='素菜', description='蔬菜类菜品'),
        Category(name='汤类', description='汤品'),
        Category(name='主食', description='米饭、面食等'),
        Category(name='凉菜', description='凉拌菜'),
        Category(name='甜品', description='甜点饮品')
    ]
    
    for category in categories:
        existing = Category.query.filter_by(name=category.name).first()
        if not existing:
            db.session.add(category)
    
    if not User.query.filter_by(username='admin').first():
        admin = User(username='admin')
        admin.set_password('admin123')
        db.session.add(admin)
        print('已创建默认管理员账户: admin / admin123')
    
    db.session.commit()
    print('数据库初始化完成，已创建默认分类。')


if __name__ == '__main__':
    with app.app_context():
        db.create_all()
        if Category.query.count() == 0:
            categories = [
                Category(name='荤菜', description='肉类菜品'),
                Category(name='素菜', description='蔬菜类菜品'),
                Category(name='汤类', description='汤品'),
                Category(name='主食', description='米饭、面食等'),
                Category(name='凉菜', description='凉拌菜'),
                Category(name='甜品', description='甜点饮品')
            ]
            for category in categories:
                db.session.add(category)
            db.session.commit()
            print('已创建默认分类')
        
        if not User.query.filter_by(username='admin').first():
            admin = User(username='admin')
            admin.set_password('admin123')
            db.session.add(admin)
            db.session.commit()
            print('已创建默认管理员账户: admin / admin123')
    
    app.run(host='0.0.0.0', port=5000, debug=True)