Corvus/server/atheneum/service/user_service.py

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:
    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:
    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:
    user.last_login_time = datetime.now()
    db.session.commit()


def update_password(user: User, password: str) -> None:
    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]:
    return User.query.filter_by(name=name).first()