-
Notifications
You must be signed in to change notification settings - Fork 1
/
manage.py
141 lines (113 loc) · 2.93 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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
# coding: utf-8
"""
project management
-- database management
-- python manage.py db init: create migrations folder
-- python manage.py db migrate: database migrate
-- python manage.py db upgrate: upgrade database
-- python manage.py shell
>> Role.insert_roles() : create user roles
-- add administrator
-- python manage.py admin
\_admin username:
\_admin email:
\_admin password:
-- add users
-- python manage.py adduser
\_username:
\_password:
\_email:
\_[1:moderator 2:admin 3:user]:
-- run project
-- python manage.py runserver
-- shell environment
-- python manage.py shell
-- run your unit tests
-- python manage.py test
"""
import sys
import os
from flask_script import Manager, Shell
from flask_migrate import Migrate, MigrateCommand
from app import db, app
from app.models import User, Role
# 编码设置
"""
manage.py
~~~~~~~~
shell:
python manage.py shell
run server:
python manage.py runserver
-d 开启调试模式
"""
import sys
from app import app
from flask_script import Manager, Shell
"""编码设置"""
# >>>>>>> ec2645b42e8d0b874bf3ca3e57dd7dd3e98d9fb0
reload(sys)
sys.setdefaultencoding('utf-8')
manager = Manager(app)
# <<<<<<< HEAD
migrate = Migrate(app, db)
# =======
# >>>>>>> ec2645b42e8d0b874bf3ca3e57dd7dd3e98d9fb0
def make_shell_context():
"""自动加载环境"""
return dict(
# <<<<<<< HEAD
app = app,
db = db,
User = User,
Role = Role
# =======
# app=app
#>>>>>>> ec2645b42e8d0b874bf3ca3e57dd7dd3e98d9fb0
)
manager.add_command("shell", Shell(make_context=make_shell_context))
# <<<<<<< HEAD
manager.add_command('db', MigrateCommand)
@manager.command
def test():
"""run your unit tests"""
import unittest
tests = unittest.TestLoader().discover('test')
unittest.TextTestRunner(verbosity=2).run(tests)
@manager.command
def admin():
"""add administrator"""
from getpass import getpass
username = raw_input("\_admin username: ")
email = raw_input("\_admin email: ")
password = getpass("\_admin password: ")
u = User(
email = email,
username = username,
password = password,
role_id = 2
)
db.session.add(u)
db.session.commit()
print ("<admin user %s add in database>") % username
@manager.command
def adduser():
"""add user"""
from getpass import getpass
username = raw_input("\_username: ")
email = raw_input("\_email: ")
role_id = raw_input("\_[1:moderator 2:admin 3:user]: ")
password = getpass("\_password: ")
u = User(
email = email,
username = username,
password = password,
role_id = role_id
)
db.session.add(u)
db.session.commit()
print ("<user %s add in database>") % username
# =======
# >>>>>>> ec2645b42e8d0b874bf3ca3e57dd7dd3e98d9fb0
if __name__ == '__main__':
manager.run()