mirror of
https://github.com/boostorg/website-v2.git
synced 2026-01-19 04:42:17 +00:00
274777a2e6
This also hides the pop up notifications after 6 seconds For review, a review and merge of #1311 on which this is built would make this easier to review.
137 lines
4.4 KiB
Python
137 lines
4.4 KiB
Python
import os
|
|
|
|
from django.contrib.auth import get_user_model
|
|
from django import forms
|
|
|
|
from allauth.account.forms import ResetPasswordKeyForm
|
|
|
|
from .models import Preferences
|
|
from news.models import NEWS_MODELS
|
|
from news.acl import can_approve
|
|
|
|
|
|
User = get_user_model()
|
|
|
|
NEWS_ENTRY_CHOICES = [(m.news_type, m._meta.verbose_name.title()) for m in NEWS_MODELS]
|
|
|
|
|
|
class CustomResetPasswordFromKeyForm(ResetPasswordKeyForm):
|
|
def save(self, **kwargs):
|
|
"""Override default reset password form so we can mark unclaimed
|
|
users as claimed once they have reset their passwords."""
|
|
result = super().save(**kwargs)
|
|
self.user.claim()
|
|
return result
|
|
|
|
|
|
class PreferencesForm(forms.ModelForm):
|
|
allow_notification_own_news_approved = forms.MultipleChoiceField(
|
|
choices=NEWS_ENTRY_CHOICES,
|
|
widget=forms.widgets.CheckboxSelectMultiple,
|
|
label="Your own news is approved after moderation",
|
|
required=False,
|
|
)
|
|
allow_notification_others_news_posted = forms.MultipleChoiceField(
|
|
choices=NEWS_ENTRY_CHOICES,
|
|
widget=forms.widgets.CheckboxSelectMultiple,
|
|
label="Other users publish their news",
|
|
required=False,
|
|
)
|
|
allow_notification_others_news_needs_moderation = forms.MultipleChoiceField(
|
|
choices=NEWS_ENTRY_CHOICES,
|
|
widget=forms.widgets.CheckboxSelectMultiple,
|
|
label="There are new entries pending moderation",
|
|
required=False,
|
|
)
|
|
|
|
def __init__(self, *args, instance=None, **kwargs):
|
|
if instance is not None:
|
|
is_moderator = can_approve(instance.user)
|
|
initial = kwargs.pop("initial", {})
|
|
for field in self.Meta.fields:
|
|
initial[field] = getattr(instance, field)
|
|
kwargs["initial"] = initial
|
|
else:
|
|
is_moderator = False
|
|
all_news = Preferences.ALL_NEWS_TYPES
|
|
kwargs["initial"] = {i: all_news for i in self.Meta.fields}
|
|
|
|
super().__init__(*args, instance=instance, **kwargs)
|
|
|
|
if not is_moderator:
|
|
self.fields.pop("allow_notification_others_news_needs_moderation")
|
|
self.initial.pop("allow_notification_others_news_needs_moderation")
|
|
|
|
def save(self, *args, **kwargs):
|
|
for field, value in self.cleaned_data.items():
|
|
setattr(self.instance, field, value)
|
|
return super().save(*args, **kwargs)
|
|
|
|
class Meta:
|
|
model = Preferences
|
|
fields = [
|
|
"allow_notification_own_news_approved",
|
|
"allow_notification_others_news_posted",
|
|
"allow_notification_others_news_needs_moderation",
|
|
]
|
|
|
|
|
|
class UserProfileForm(forms.ModelForm):
|
|
class Meta:
|
|
model = User
|
|
fields = ["email", "first_name", "last_name", "indicate_last_login_method"]
|
|
|
|
|
|
class CustomClearableFileInput(forms.ClearableFileInput):
|
|
"""
|
|
Overrides the template for clearable file input so that we can display
|
|
the widget without the filename/path displayed and change the checkbox
|
|
to clear the field.
|
|
"""
|
|
|
|
template_name = "users/clearable_file_input.html"
|
|
|
|
|
|
class UserProfilePhotoForm(forms.ModelForm):
|
|
image = forms.FileField(widget=CustomClearableFileInput, required=False)
|
|
|
|
class Meta:
|
|
model = User
|
|
fields = ["image"]
|
|
|
|
def clean(self):
|
|
"""Ensure a user can't update their photo if they
|
|
don't have permission."""
|
|
cleaned_data = super().clean()
|
|
if not self.instance.can_update_image:
|
|
raise forms.ValidationError(
|
|
"You do not have permission to update your profile photo."
|
|
)
|
|
return cleaned_data
|
|
|
|
def save(self, commit=True):
|
|
# Temporarily store the old image
|
|
old_image = self.instance.image
|
|
# Save the new image
|
|
user = super().save(commit=False)
|
|
|
|
if old_image:
|
|
# Delete the old image file if there's a new image being uploaded
|
|
if self.cleaned_data["image"] != old_image:
|
|
old_image.delete(save=False)
|
|
|
|
if self.cleaned_data.get("image"):
|
|
new_image = self.cleaned_data["image"]
|
|
_, file_extension = os.path.splitext(new_image.name)
|
|
|
|
# Strip the leading period from the file extension.
|
|
file_extension = file_extension.lstrip(".")
|
|
|
|
new_image.name = f"{user.profile_image_filename_root}.{file_extension}"
|
|
user.image = new_image
|
|
|
|
if commit:
|
|
user.save()
|
|
|
|
return user
|