selfprivacy-rest-api/selfprivacy_api/actions/users.py

"""Actions to manage the users."""

import re
from typing import Optional

from selfprivacy_api.models.user import UserDataUser

from selfprivacy_api.utils import hash_password, is_username_forbidden

from selfprivacy_api.repositories.users import ACTIVE_USERS_PROVIDER
from selfprivacy_api.repositories.users.exceptions import (
    PasswordIsEmpty,
    UsernameForbidden,
    UsernameNotAlphanumeric,
    UsernameTooLong,
)


def get_users(
    exclude_primary: bool = False,
    exclude_root: bool = False,
) -> list[UserDataUser]:
    return ACTIVE_USERS_PROVIDER.get_users(
        exclude_primary=exclude_primary, exclude_root=exclude_root
    )


def create_user(username: str, password: str) -> None:
    if password == "":
        raise PasswordIsEmpty("Password is empty")

    if is_username_forbidden(username):
        raise UsernameForbidden("Username is forbidden")

    if not re.match(r"^[a-z_][a-z0-9_]+$", username):
        raise UsernameNotAlphanumeric(
            "Username must be alphanumeric and start with a letter"
        )

    if len(username) >= 32:
        raise UsernameTooLong("Username must be less than 32 characters")

    hashed_password = hash_password(password)

    return ACTIVE_USERS_PROVIDER.create_user(
        username=username, hashed_password=hashed_password
    )


def delete_user(username: str) -> None:
    return ACTIVE_USERS_PROVIDER.delete_user(username=username)


def update_user(username: str, password: str) -> None:
    if password == "":
        raise PasswordIsEmpty("Password is empty")

    hashed_password = hash_password(password)

    return ACTIVE_USERS_PROVIDER.update_user(
        username=username, hashed_password=hashed_password
    )


def get_user_by_username(username: str) -> Optional[UserDataUser]:
    return ACTIVE_USERS_PROVIDER.get_user_by_username(username=username)