""" Service to handle user operations """ from datetime import datetime from typing import Optional from atheneum import APP, db from atheneum.model import User from atheneum.service import authentication_service def register(name: str, password: str, role: str) -> User: """ Register a new user :param name: Desired user name. Must be unique and not already registered :param password: Password to be hashed and stored for the user :param role: Role to assign the user [ROLE_USER, ROLE_ADMIN] :return: """ pw_hash, pw_revision = authentication_service.get_password_hash(password) new_user = User( name=name, role=role, password_hash=pw_hash, password_revision=pw_revision, creation_time=datetime.now(), version=0) db.session.add(new_user) db.session.commit() APP.logger.info('Registered new user: %s with role: %s', name, role) return new_user def delete(user: User) -> bool: """ Delete a user :param user: :return: """ existing_user = db.session.delete(user) if existing_user is None: db.session.commit() return True return False def update_last_login_time(user: User) -> None: """ Bump the last login time for the user :param user: :return: """ user.last_login_time = datetime.now() db.session.commit() def update_password(user: User, password: str) -> None: """ Change the user password :param user: :param password: :return: """ pw_hash, pw_revision = authentication_service.get_password_hash( password) user.password_hash = pw_hash user.password_revision = pw_revision db.session.commit() def find_by_name(name: str) -> Optional[User]: """ Find a user by name :param name: :return: """ return User.query.filter_by(name=name).first()