Source code for pas.forms.user

from __future__ import annotations
from onegov.form import merge_forms
from onegov.org.forms import ManageUserForm
from onegov.org.forms.user import PartialNewUserForm
from onegov.pas import _

from typing import TYPE_CHECKING




class ManageUserFormCustom(ManageUserForm):


    def on_request(self) -> None:
        super().on_request()
        self.role.choices = [
            ('admin', _('Admin')),
            ('parliamentarian', _('Parliamentarian')),
            ('commission_president', _('Commission President'))
        ]




# The problem is that NewUserFormCustom is a runtime-created form
# class from merge_forms(), which mypy can't properly type check.
if TYPE_CHECKING:


    class NewUserFormCustom(PartialNewUserForm, ManageUserFormCustom):
        pass

else:
    NewUserFormCustom = merge_forms(PartialNewUserForm, ManageUserFormCustom)