takahe/users/views/settings/profile.py

140 wiersze
5.4 KiB
Python

from django import forms
from django.contrib import messages
from django.contrib.auth.decorators import login_required
from django.core.files import File
from django.shortcuts import redirect
from django.utils.decorators import method_decorator
from django.views.generic import FormView
from core.html import FediverseHtmlParser
from core.models.config import Config
from users.models import IdentityStates
from users.services import IdentityService
from users.shortcuts import by_handle_or_404
@method_decorator(login_required, name="dispatch")
class ProfilePage(FormView):
"""
Lets the identity's profile be edited
"""
template_name = "settings/profile.html"
extra_context = {"section": "profile"}
class form_class(forms.Form):
name = forms.CharField(max_length=500)
summary = forms.CharField(
widget=forms.Textarea,
required=False,
help_text="Describe you and your interests",
label="Bio",
)
icon = forms.ImageField(
required=False, help_text="Shown next to all your posts and activities"
)
image = forms.ImageField(
required=False, help_text="Shown at the top of your profile"
)
discoverable = forms.BooleanField(
help_text="If this user is visible on the frontpage and in user directories",
widget=forms.Select(
choices=[(True, "Discoverable"), (False, "Not Discoverable")]
),
required=False,
)
visible_follows = forms.BooleanField(
help_text="Whether or not to show your following and follower counts in your profile",
widget=forms.Select(choices=[(True, "Visible"), (False, "Hidden")]),
required=False,
)
search_enabled = forms.BooleanField(
help_text="If a search feature is provided for your posts on the profile page\n(Disabling this will not prevent third-party search crawlers from indexing your posts)",
widget=forms.Select(choices=[(True, "Enabled"), (False, "Disabled")]),
required=False,
)
boosts_on_profile = forms.BooleanField(
help_text="Include your boosts with your posts on your profile page",
widget=forms.Select(choices=[(True, "Enabled"), (False, "Disabled")]),
required=False,
)
metadata = forms.JSONField(
label="Profile Metadata Fields",
help_text="These values will appear on your profile below your bio",
widget=forms.HiddenInput(attrs={"data-min-empty": 2}),
required=False,
)
def clean_metadata(self):
metadata = self.cleaned_data["metadata"]
if metadata:
metadata = [x for x in metadata if x["name"] and x["value"]]
if not metadata:
return None
return metadata
def dispatch(self, request, handle: str, *args, **kwargs):
self.identity = by_handle_or_404(self.request, handle, local=True, fetch=False)
return super().dispatch(request, *args, **kwargs)
def get_context_data(self, **kwargs):
context = super().get_context_data(**kwargs)
context["identity"] = self.identity
return context
def get_initial(self):
return {
"name": self.identity.name,
"summary": (
FediverseHtmlParser(self.identity.summary).plain_text
if self.identity.summary
else ""
),
"icon": self.identity.icon and self.identity.icon.url,
"image": self.identity.image and self.identity.image.url,
"discoverable": self.identity.discoverable,
"visible_follows": self.identity.config_identity.visible_follows,
"metadata": self.identity.metadata or [],
"search_enabled": self.identity.config_identity.search_enabled,
"boosts_on_profile": self.identity.config_identity.boosts_on_profile,
}
def form_valid(self, form):
# Update basic info
service = IdentityService(self.identity)
self.identity.name = form.cleaned_data["name"]
self.identity.discoverable = form.cleaned_data["discoverable"]
service.set_summary(form.cleaned_data["summary"])
# Resize images
icon = form.cleaned_data.get("icon")
image = form.cleaned_data.get("image")
if isinstance(icon, File):
service.set_icon(icon)
if isinstance(image, File):
service.set_image(image)
self.identity.metadata = form.cleaned_data.get("metadata")
# Clear images if specified
if "icon__clear" in self.request.POST:
self.identity.icon = None
if "image__clear" in self.request.POST:
self.identity.image = None
# Save and propagate
self.identity.save()
self.identity.transition_perform(IdentityStates.edited)
# Save profile-specific identity Config
Config.set_identity(
self.identity, "visible_follows", form.cleaned_data["visible_follows"]
)
Config.set_identity(
self.identity, "search_enabled", form.cleaned_data["search_enabled"]
)
Config.set_identity(
self.identity, "boosts_on_profile", form.cleaned_data["boosts_on_profile"]
)
messages.success(self.request, "Your profile has been updated.")
return redirect(".")