-
Notifications
You must be signed in to change notification settings - Fork 0
/
manage.py
38 lines (30 loc) · 929 Bytes
/
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
#coding: utf-8
from __future__ import unicode_literals
from main import app
from models import db, User, Menu, Article, Showcase
from flask.ext.script import Manager
from flask.ext.migrate import Migrate, MigrateCommand
manager = Manager(app)
migrate = Migrate(app, db)
manager.add_command('db', MigrateCommand)
@manager.command
def create_all():
db.create_all()
@manager.command
def add_user():
with app.app_context():
username = raw_input("Username: ")
nickname = raw_input("Nickname: ")
password = raw_input("Password: ")
permission = raw_input("Permission: ")
user = User()
user.name = username
user.nickname = nickname
user.password = password
user.permission = permission
user.locked = False
user.failed_times = 0
str = user.regenerate_otp_token()
print str
if __name__ == '__main__':
manager.run()