22 lines
761 B
Python

from app import user_datastore
def upsert_user(email, password="password"):
result = user_datastore.find_user(email=email)
if result is None:
result = user_datastore.create_user(email=email, password=hash_password(password))
return result
def add_roles_to_user(user_name, role_names):
user = upsert_user(user_name)
for role_name in role_names:
user_datastore.add_role_to_user(user, role_name)
def upsert_user_role(role_name, role_title, permissions):
role = user_datastore.find_or_create_role(role_name)
role.title = role_title
role.remove_permissions(role.get_permissions())
role.add_permissions(permissions)
return role
def find_user_by_email(email):
return user_datastore.find_user(email=email)