harmacist/DashMachine
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

working on user system

Browse Source
This commit is contained in:
Ross Mountjoy 2020-02-08 09:12:40 -05:00
parent 82c7afb470
commit 071d12a285
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
dashmachine/user_system/utils.py
View File

@ -2,14 +2,12 @@ from dashmachine import db, bcrypt
from dashmachine.user_system.models import User from dashmachine.user_system.models import User
def add_edit_user(username, password, user_id=None, role=None, new=False): def add_edit_user(username, password, user_id=None, role=None):
if user_id: if user_id:
user = User.query.filter_by(id=user_id).first() user = User.query.filter_by(id=user_id).first()
elif new: if not user:
user = User() user = User()
else: else:
user = User.query.first()
if not user:
user = User() user = User()
hashed_password = bcrypt.generate_password_hash(password).decode("utf-8") hashed_password = bcrypt.generate_password_hash(password).decode("utf-8")