新增自定義命令

下面是一個使用命令列新增管理員賬號的虛擬碼

from info.models import User
from flask_script import Manager

app = Flask(__neme__)
manage = Manager(app)

# 新增自定義命令
@manage.option('-n', '-name', dest='name')
@manage.option('-p', '-password', dest='password')
def createadmin(name, password):
    if not all([name, password]):
        print('引數不全'
 
)
    user = User()
    user.mobile = name
    user.nick_name = name
    user.password = password
    user.is_admin = 1
    try:
        db.session.add(user)
        db.session.commit()
    except Exception as e:
        print(e)
        print('建立失敗')
        db.session.rollback()
        return
    print('建立管理員使用者成功'
 
)

# 使用示例(命令列,manage.py存在的路徑)
python manage -n admin_user -p asd258852