21 lines
567 B
Python
Executable File
21 lines
567 B
Python
Executable File
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):
|
|
if user_id:
|
|
user = User.query.filter_by(id=user_id).first()
|
|
elif new:
|
|
user = User()
|
|
else:
|
|
user = User.query.first()
|
|
if not user:
|
|
user = User()
|
|
|
|
hashed_password = bcrypt.generate_password_hash(password).decode("utf-8")
|
|
user.username = username
|
|
user.password = hashed_password
|
|
user.role = role
|
|
db.session.merge(user)
|
|
db.session.commit()
|