from django import forms
from apps.core.models import User
from apps.core.constants import UserType


class UserAdminForm(forms.ModelForm):

    class Meta:
        model = User
        fields = "__all__"

    def __init__(self, *args, **kwargs):
        super().__init__(*args, **kwargs)
        self.fields["user_type"].widget = forms.Select(
            choices=UserType.choices()
        )