-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathmanage.py
48 lines (34 loc) · 1.1 KB
/
manage.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
from flask_script import Manager
from flask_migrate import Migrate, MigrateCommand
from info import create_app, db, models
# manage.py是程序启动的入口,只关心启动的相关参数以及内容,不关心具体该
# 如果创建app或者相关业务逻辑
# 通过指定的配置名字创建对应配置的app
# create_app 就类似于工厂方法
from info.models import User
app = create_app('development')
manager = Manager(app)
# 将 app 与 db 关联
Migrate(app, db)
# 将迁移命令添加到manager中
manager.add_command('db', MigrateCommand)
@manager.option('-n', '-name', dest="name")
@manager.option('-p', '-password', dest="password")
def createsuperuser(name, password):
if not all([name, password]):
print("参数不足")
user = User()
user.nick_name = name
user.mobile = name
user.password = password
user.is_admin = True
try:
db.session.add(user)
db.session.commit()
except Exception as e:
db.session.rollback()
print(e)
print("添加成功")
if __name__ == '__main__':
print(app.url_map)
manager.run()