import logging

from flask import current_app
from flask_script import Manager,Command
from flask_migrate import Migrate, MigrateCommand

from info import create_app,db
from config import DevelopementConfig,ProductionConfig
from flask_sqlalchemy import SQLAlchemy
from info import models

# 执行项目的初始化
from info.models import User

app = create_app(DevelopementConfig)

# 使用终端脚本工具启动和管理flask
manager = Manager(app)

# 启用数据迁移工具
Migrate(app, db)
# 添加数据迁移的命令到终端脚本工具中
manager.add_command('db', MigrateCommand)

# 通过终端命令创建测试数据[ 例如管理员 ]
# class CreateSupperUser(Command):  # 命令类
#     def run(self):
#         print("hello")
# manager.add_command('createsuperuser', CreateSupperUser())

@manager.option('-n', '-name', dest='name')
@manager.option('-p', '-password', dest='password')
def createsuperuser(name, password):
    """创建管理员用户"""
    if not all([name, password]):
        print('参数不足')
        return

    user = User()
    user.mobile = name
    user.nick_name = name
    user.password = password
    user.is_admin = True

    try:
        db.session.add(user)
        db.session.commit()
        print("创建成功")
    except Exception as e:
        print(e)
        db.session.rollback()


if __name__ == '__main__':
    app.run()        # 开发阶段使用的
    # manager.run()  # 终端下面运行