-
Notifications
You must be signed in to change notification settings - Fork 2
/
manager.py
43 lines (29 loc) · 852 Bytes
/
manager.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
# coding:utf-8
from flask.ext.script import Manager
from sqlalchemy.exc import IntegrityError
from mario.config import HOST, PORT, DEBUG, USERS
from mario.app import create_app
from mario.libs.extension import db
from mario.models.user import User
app = create_app()
manager = Manager(app)
@manager.command
def run():
app.run(host=HOST, port=PORT, debug=DEBUG)
@manager.shell
def make_shell_context():
return {'app': app, 'db': db, 'User': User}
@manager.command
def init_db():
db.drop_all()
db.create_all()
@manager.command
def init_data():
for name, display_name, count in USERS:
try:
user = User.add(name=name, display_name=display_name, count=count)
except IntegrityError:
continue
print 'User {0} is added'.format(user)
if __name__ == '__main__':
manager.run()