forked from Chris3691/Dash-User-Management
-
Notifications
You must be signed in to change notification settings - Fork 0
/
users_mgt.py
70 lines (48 loc) · 1.67 KB
/
users_mgt.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
from sqlalchemy import Table
from sqlalchemy.sql import select
from flask_sqlalchemy import SQLAlchemy
from werkzeug.security import generate_password_hash
from config import engine
db = SQLAlchemy()
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(15), unique=True)
email = db.Column(db.String(50), unique=True)
password = db.Column(db.String(80))
admin = db.Column(db.Boolean)
userTable = Table('user', User.metadata)
def create_user_table():
User.metadata.create_all(engine)
def add_user(username, password, email, admin):
hashed_password = generate_password_hash(password, method='sha256')
insert_stmt = userTable.insert().values(
username=username, email=email, password=hashed_password, admin=admin
)
conn = engine.connect()
conn.execute(insert_stmt)
conn.close()
def update_password(username, password):
hashed_password = generate_password_hash(password, method='sha256')
update = userTable.update().\
values(password=hashed_password).\
where(userTable.c.username==username)
conn = engine.connect()
conn.execute(update)
conn.close()
def show_users():
select_stmt = select([userTable.c.id,
userTable.c.username,
userTable.c.email,
userTable.c.admin])
conn = engine.connect()
results = conn.execute(select_stmt)
users = []
for result in results:
users.append({
'id' : result[0],
'username' : result[1],
'email' : result[2],
'admin' : str(result[3])
})
conn.close()
return users