Source code for eucrim.users.forms

# SPDX-FileCopyrightText: 2024 Thomas Breitner <t.breitner@csl.mpg.de>
#
# SPDX-License-Identifier: EUPL-1.2

from django import forms
from wagtail.users.forms import UserEditForm, UserCreationForm
from .models import CustomUser




class CustomUserEditForm(UserEditForm):
    """Custom user attributes in the wagtail admin UserEditForm."""

    is_association_manager = forms.BooleanField(
        required=False, label="Is association manager?"
    )
    is_author = forms.BooleanField(required=False, label="Is author?")
    is_team = forms.BooleanField(required=False, label="Is team member?")
    team_role = forms.ChoiceField(
        required=False,
        choices=CustomUser.TEAM_ROLE_CHOICES,
        initial=CustomUser.NOROLE,
        help_text="Role in eucrim team, like stated in the imprint of the "
        'printed issue; e.g. "Managing Editor", defaults to a quasi '
        'non-role "Member"',
    )



    class Meta(UserEditForm.Meta):
        fields = UserEditForm.Meta.fields | {
            "is_association_manager",
            "is_author",
            "is_team",
            "team_role",
        }






class CustomUserCreationForm(UserCreationForm):
    """Custom user attributes in the wagtail admin UserCreationForm."""

    is_association_manager = forms.BooleanField(
        required=False, label="Is association manager?"
    )
    is_author = forms.BooleanField(required=False, label="Is author?")
    is_team = forms.BooleanField(required=False, label="Is team member?")
    team_role = forms.ChoiceField(
        required=False,
        choices=CustomUser.TEAM_ROLE_CHOICES,
        initial=CustomUser.NOROLE,
        help_text="Role in eucrim team, like stated in the imprint of the "
        'printed issue; e.g. "Managing Editor", defaults to a quasi '
        'non-role "Member"',
    )



    class Meta(UserCreationForm.Meta):
        fields = UserCreationForm.Meta.fields | {
            "is_association_manager",
            "is_author",
            "is_team",
            "team_role",
        }