byceps/byceps

"""
byceps.blueprints.site.ticketing.forms
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

:Copyright: 2014-2024 Jochen Kupperschmidt
:License: Revised BSD (see `LICENSE` file for details)
"""

from flask import g
from flask_babel import gettext, lazy_gettext
from wtforms import StringField
from wtforms.validators import InputRequired, ValidationError

from byceps.services.consent import consent_service, consent_subject_service
from byceps.services.user import user_service
from byceps.util.l10n import LocalizedForm


def validate_user(form, field):
    screen_name = field.data.strip()

    db_user = user_service.find_db_user_by_screen_name(
        screen_name, case_insensitive=True
    )

    if db_user is None:
        raise ValidationError(gettext('Unknown username'))

    user = user_service.get_user(db_user.id)

    if not user.initialized:
        raise ValidationError(gettext('The user account is not active.'))

    if user.suspended or user.deleted:
        raise ValidationError(gettext('The user account is not active.'))

    required_consent_subjects = (
        consent_subject_service.get_subjects_required_for_brand(g.brand_id)
    )
    required_consent_subject_ids = {
        subject.id for subject in required_consent_subjects
    }

    if not consent_service.has_user_consented_to_all_subjects(
        user.id, required_consent_subject_ids
    ):
        raise ValidationError(
            gettext(
                'User "%(screen_name)s" has not yet given all necessary '
                'consents. Logging in again is required.',
                screen_name=user.screen_name,
            )
        )

    field.data = user


class SpecifyUserForm(LocalizedForm):
    user = StringField(
        lazy_gettext('Username'), [InputRequired(), validate_user]
    )