from django import forms
from django.contrib.auth.forms import ReadOnlyPasswordHashField
from django.contrib.auth.models import Group, User
from django.urls import reverse_lazy
from django.utils.text import format_lazy
from django.utils.translation import gettext_lazy as _

from accounts.models import UserAttributes

from .models import Permission


class GroupForm(forms.ModelForm):
 permissions = forms.ModelMultipleChoiceField(
 widget=forms.CheckboxSelectMultiple,
 queryset=Permission.objects.filter(content_type__model="permissionset"),
 required=False,
 )

 users = forms.ModelMultipleChoiceField(
 widget=forms.CheckboxSelectMultiple,
 queryset=User.objects.all(),
 required=False,
 )

 def __init__(self, *args, **kwargs):
 super(GroupForm, self).__init__(*args, **kwargs)
 instance = getattr(self, "instance", None)
 if instance and instance.id:
 self.fields["users"].initial = self.instance.user_set.all()

 def save_m2m(self):
 self.instance.user_set.set(self.cleaned_data["users"])

 def save(self, *args, **kwargs):
 instance = super(GroupForm, self).save()
 self.save_m2m()
 return instance

 class Meta:
 model = Group
 fields = "__all__"


class UserForm(forms.ModelForm):
 user_permissions = forms.ModelMultipleChoiceField(
 widget=forms.CheckboxSelectMultiple,
 queryset=Permission.objects.filter(content_type__model="permissionset"),
 label=_("Permissions"),
 required=False,
 )

 groups = forms.ModelMultipleChoiceField(
 widget=forms.CheckboxSelectMultiple,
 queryset=Group.objects.all(),
 label=_("Groups"),
 required=False,
 )

 class Meta:
 model = User
 fields = [
 "username",
 "groups",
 "first_name",
 "last_name",
 "email",
 "user_permissions",
 "is_staff",
 "is_active",
 "is_superuser",
 ]

 def __init__(self, *args, **kwargs):
 super(UserForm, self).__init__(*args, **kwargs)
 if self.instance.id:
 password = ReadOnlyPasswordHashField(
 label=_("Password"),
 help_text=format_lazy(
 _(
 """Raw passwords are not stored, so there is no way to see this user's password, 
 but you can change the password using <a href='{}'>this form</a>."""
 ),
 reverse_lazy(
 "admin:user_update_password",
 args=[
 self.instance.id,
 ],
 ),
 ),
 )
 self.fields["Password"] = password


class UserCreateForm(UserForm):
 password = forms.CharField(widget=forms.PasswordInput)

 class Meta:
 model = User
 fields = [
 "username",
 "password",
 "groups",
 "first_name",
 "last_name",
 "email",
 "user_permissions",
 "is_staff",
 "is_active",
 "is_superuser",
 ]


class UserAttributesForm(forms.ModelForm):
 class Meta:
 model = UserAttributes
 exclude = ["user", "can_clone_instances"]