/
manage.py
50 lines (39 loc) · 1.32 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
49
50
#!/usr/bin/env python
import argparse
import code
from fabric.api import task
from app import app, db, user_datastore, Role, User
def build():
db.create_all()
user_role = user_datastore.create_role(name="user")
admin_role = user_datastore.create_role(name="admin")
admin_user = user_datastore.create_user(email="admin@sup.io", password="password")
user_datastore.add_role_to_user(admin_user, admin_role)
test_user1 = user_datastore.create_user(email="test@sup.io", password="password")
test_user2 = user_datastore.create_user(email="test2@sup.io", password="password")
user_datastore.add_role_to_user(test_user1, user_role)
user_datastore.add_role_to_user(test_user2, user_role)
db.session.add(admin_user)
db.session.add(test_user1)
db.session.add(test_user2)
db.session.commit()
def console():
context = locals()
context['app'] = app
context['db'] = db
context['user_datastore'] = user_datastore
context['Role'] = Role
context['User'] = User
code.interact(local=locals())
def run():
app.run(host='127.0.0.1', debug=True)
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('action')
args = parser.parse_args()
if args.action == 'build':
build()
elif args.action == 'console':
console()
elif args.action == 'run':
run()