/
manage.py
69 lines (46 loc) · 1.55 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
import os
from flask.ext.script import Manager
from my_app import app
# terminal prompt to get password from user
from getpass import getpass
from werkzeug.security import generate_password_hash
from my_app.models import User
from my_app.database import session
from flask.ext.migrate import Migrate, MigrateCommand
from my_app.database import Base, engine, session
manager = Manager(app)
class DB(object):
def __init__(self, metadata):
self.metadata = metadata
migrate = Migrate(app, DB(Base.metadata))
manager.add_command('db', MigrateCommand)
@manager.command
def run():
port = int(os.environ.get('PORT', 5000))
app.run(host='0.0.0.0', port=port)
@manager.command
def adduser():
name = raw_input("Name: ")
email = raw_input("Email: ")
if session.query(User).filter_by(email=email).first():
print "User with that email address already exists"
return
#add get_pass to hide password
password = raw_input("Password ")
password_2 = raw_input("Password ")
while not (password and password_2) or password != password_2:
# use get pass from the get pass modelue to add a user
password = getpass("Password: ")
password_2 = getpass("Re-enter password: ")
user = User(name=name, email=email,
password=generate_password_hash(password))
session.add(user)
session.commit()
@manager.command
def createtables():
Base.metadata.create_all(engine)
@manager.command
def deletetables():
Base.metadata.drop_all(engine)
if __name__ == "__main__":
manager.run()