mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-17 04:59:37 -04:00
68 lines
1.5 KiB
Python
68 lines
1.5 KiB
Python
from notifications_python_client import HTTPError
|
|
|
|
from app import login_manager
|
|
from app import user_api_client
|
|
|
|
#
|
|
# TODO fix up this, do we really need this class why not just use the clients
|
|
# directly??
|
|
#
|
|
|
|
|
|
@login_manager.user_loader
|
|
def load_user(user_id):
|
|
return get_user_by_id(user_id)
|
|
|
|
|
|
# TODO Would be better to have a generic get and update for user
|
|
# something that replicates the sql functionality.
|
|
def get_user_by_id(id):
|
|
return user_api_client.get_user(id)
|
|
|
|
|
|
def get_all_users():
|
|
return user_api_client.get_users()
|
|
|
|
|
|
def get_user_by_email(email_address):
|
|
return user_api_client.get_user_by_email(email_address)
|
|
|
|
|
|
def verify_password(user_id, password):
|
|
return user_api_client.verify_password(user_id, password)
|
|
|
|
|
|
def update_user(user):
|
|
return user_api_client.update_user(user)
|
|
|
|
|
|
def increment_failed_login_count(id):
|
|
user = get_user_by_id(id)
|
|
user.failed_login_count += 1
|
|
return user_api_client.update_user(user)
|
|
|
|
|
|
def activate_user(user):
|
|
user.state = 'active'
|
|
return user_api_client.update_user(user)
|
|
|
|
|
|
def is_email_unique(email_address):
|
|
try:
|
|
if user_api_client.get_user_by_email(email_address):
|
|
return False
|
|
return True
|
|
except HTTPError as ex:
|
|
if ex.status_code == 404:
|
|
return True
|
|
else:
|
|
raise ex
|
|
|
|
|
|
def send_verify_code(user_id, code_type, to):
|
|
return user_api_client.send_verify_code(user_id, code_type, to)
|
|
|
|
|
|
def check_verify_code(user_id, code, code_type):
|
|
return user_api_client.check_verify_code(user_id, code, code_type)
|