Initial commit

env/lib/python3.10/site-packages/wagtail/admin/forms/account.py

@@ -0,0 +1,143 @@
+import warnings
+from operator import itemgetter
+
+import l18n
+from django import forms
+from django.contrib.auth import get_user_model
+from django.db.models.fields import BLANK_CHOICE_DASH
+from django.utils.translation import get_language_info
+from django.utils.translation import gettext_lazy as _
+
+from wagtail.admin.localization import (
+    get_available_admin_languages,
+    get_available_admin_time_zones,
+)
+from wagtail.admin.widgets import SwitchInput
+from wagtail.permissions import page_permission_policy
+from wagtail.users.models import UserProfile
+
+User = get_user_model()
+
+
+class NotificationPreferencesForm(forms.ModelForm):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        permission_policy = page_permission_policy
+        if not permission_policy.user_has_permission(self.instance.user, "publish"):
+            del self.fields["submitted_notifications"]
+        if not permission_policy.user_has_permission(self.instance.user, "change"):
+            del self.fields["approved_notifications"]
+            del self.fields["rejected_notifications"]
+            del self.fields["updated_comments_notifications"]
+
+    class Meta:
+        model = UserProfile
+        fields = [
+            "submitted_notifications",
+            "approved_notifications",
+            "rejected_notifications",
+            "updated_comments_notifications",
+        ]
+        widgets = {
+            "submitted_notifications": SwitchInput(),
+            "approved_notifications": SwitchInput(),
+            "rejected_notifications": SwitchInput(),
+            "updated_comments_notifications": SwitchInput(),
+        }
+
+
+def _get_language_choices():
+    language_choices = [
+        (lang_code, get_language_info(lang_code)["name_local"])
+        for lang_code, lang_name in get_available_admin_languages()
+    ]
+    return sorted(
+        BLANK_CHOICE_DASH + language_choices,
+        key=lambda language_choice: language_choice[1].lower(),
+    )
+
+
+def _get_time_zone_choices():
+    time_zones = [
+        (tz, str(l18n.tz_fullnames.get(tz, tz)))
+        for tz in get_available_admin_time_zones()
+    ]
+    time_zones.sort(key=itemgetter(1))
+    return BLANK_CHOICE_DASH + time_zones
+
+
+class LocalePreferencesForm(forms.ModelForm):
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+
+        if len(get_available_admin_languages()) <= 1:
+            del self.fields["preferred_language"]
+
+        if len(get_available_admin_time_zones()) <= 1:
+            del self.fields["current_time_zone"]
+
+    preferred_language = forms.ChoiceField(
+        required=False, choices=_get_language_choices, label=_("Preferred language")
+    )
+
+    current_time_zone = forms.ChoiceField(
+        required=False, choices=_get_time_zone_choices, label=_("Current time zone")
+    )
+
+    class Meta:
+        model = UserProfile
+        fields = ["preferred_language", "current_time_zone"]
+
+
+class NameEmailForm(forms.ModelForm):
+    first_name = forms.CharField(required=True, label=_("First Name"))
+    last_name = forms.CharField(required=True, label=_("Last Name"))
+    email = forms.EmailField(required=True, label=_("Email"))
+
+    def __init__(self, *args, **kwargs):
+        from wagtail.admin.views.account import email_management_enabled
+
+        super().__init__(*args, **kwargs)
+
+        if not email_management_enabled():
+            del self.fields["email"]
+
+    class Meta:
+        model = User
+        fields = ["first_name", "last_name", "email"]
+
+
+class AvatarPreferencesForm(forms.ModelForm):
+    avatar = forms.ImageField(label=_("Upload a profile picture"), required=False)
+
+    def __init__(self, *args, **kwargs):
+        super().__init__(*args, **kwargs)
+        self._original_avatar = self.instance.avatar
+
+    def save(self, commit=True):
+        if (
+            commit
+            and self._original_avatar
+            and (self._original_avatar != self.cleaned_data["avatar"])
+        ):
+            # Call delete() on the storage backend directly, as calling self._original_avatar.delete()
+            # will clear the now-updated field on self.instance too
+            try:
+                self._original_avatar.storage.delete(self._original_avatar.name)
+            except OSError:
+                # failure to delete the old avatar shouldn't prevent us from continuing
+                warnings.warn(
+                    "Failed to delete old avatar file: %s" % self._original_avatar.name
+                )
+        super().save(commit=commit)
+
+    class Meta:
+        model = UserProfile
+        fields = ["avatar"]
+
+
+class ThemePreferencesForm(forms.ModelForm):
+    class Meta:
+        model = UserProfile
+        fields = ["theme", "density"]