diff --git a/dashmachine/user_system/utils.py b/dashmachine/user_system/utils.py index e01bf9e..b641955 100755 --- a/dashmachine/user_system/utils.py +++ b/dashmachine/user_system/utils.py @@ -2,14 +2,12 @@ from dashmachine import db, bcrypt 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: user = User.query.filter_by(id=user_id).first() - elif new: - user = User() + if not user: + user = User() else: - user = User.query.first() - if not user: user = User() hashed_password = bcrypt.generate_password_hash(password).decode("utf-8")