-
Notifications
You must be signed in to change notification settings - Fork 0
/
app.py
69 lines (58 loc) · 1.9 KB
/
app.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 click
from dynamic_qr import app, db
from dynamic_qr.models import User, QRCode
app.run()
@app.shell_context_processor
def make_shell_context():
return dict(db=db, User=User, QRCode=QRCode)
@app.cli.command("create-user")
@click.argument("username")
@click.argument("email")
@click.argument("password")
def create_user(username, email, password):
try:
user = User(username=username, email=email)
user.set_password(password=password)
db.session.add(user)
db.session.commit()
print("New user {} with email {} created".format(user, email))
except:
print("Something went wrong")
@app.cli.command("update-password")
@click.argument("username")
@click.argument("email")
@click.argument("password")
def update_password(username, email, password):
try:
user = User.query.filter_by(email=email).first()
user.set_password(password=password)
db.session.add(user)
db.session.commit()
print("User {} with email {} updated password".format(user, email))
except:
print("Something went wrong")
@app.cli.command("update-email")
@click.argument("username")
@click.argument("email")
@click.argument("password")
def update_email(username, email, password):
try:
user = User.query.filter_by(email=email).first()
user.email = email
db.session.add(user)
db.session.commit()
print("User {} with email {} updated password".format(user, email))
except:
print("Something went wrong")
@app.cli.command("delete-user")
@click.argument("username")
@click.argument("email")
@click.argument("password")
def delete_user(username, email, password):
try:
user = User.query.filter_by(email=email).first()
db.session.delete(user)
db.session.commit()
print("User {} with email {} deleted".format(user, email))
except:
print("Something went wrong")