From f5eafb0ca0af3ed064202abbe99bfbeef8bbb74d Mon Sep 17 00:00:00 2001 From: Andrew Godwin Date: Thu, 17 Nov 2022 08:21:42 -0700 Subject: Add image/icon upload --- users/models/identity.py | 33 ++++++++++----- users/views/admin.py | 52 +---------------------- users/views/settings.py | 105 +++++++++++++++++++++++++++++++++++++++++++++-- 3 files changed, 126 insertions(+), 64 deletions(-) (limited to 'users') diff --git a/users/models/identity.py b/users/models/identity.py index 4bbaeaf..d4ab720 100644 --- a/users/models/identity.py +++ b/users/models/identity.py @@ -1,5 +1,3 @@ -import base64 -import uuid from functools import partial from typing import Optional, Tuple from urllib.parse import urlparse @@ -10,9 +8,11 @@ from asgiref.sync import async_to_sync, sync_to_async from cryptography.hazmat.primitives import serialization from cryptography.hazmat.primitives.asymmetric import rsa from django.db import models +from django.templatetags.static import static from django.utils import timezone from core.ld import canonicalise +from core.uploads import upload_namer from stator.models import State, StateField, StateGraph, StatorModel from users.models.domain import Domain @@ -33,15 +33,6 @@ class IdentityStates(StateGraph): return "updated" -def upload_namer(prefix, instance, filename): - """ - Names uploaded images etc. - """ - now = timezone.now() - filename = base64.b32encode(uuid.uuid4().bytes).decode("ascii") - return f"{prefix}/{now.year}/{now.month}/{now.day}/{filename}" - - class Identity(StatorModel): """ Represents both local and remote Fediverse identities (actors) @@ -128,6 +119,26 @@ class Identity(StatorModel): else: return f"/@{self.username}@{self.domain_id}/" + def local_icon_url(self): + """ + Returns an icon for us, with fallbacks to a placeholder + """ + if self.icon: + return self.icon.url + elif self.icon_uri: + return self.icon_uri + else: + return static("img/unknown-icon-128.png") + + def local_image_url(self): + """ + Returns a background image for us, returning None if there isn't one + """ + if self.image: + return self.image.url + elif self.image_uri: + return self.image_uri + ### Alternate constructors/fetchers ### @classmethod diff --git a/users/views/admin.py b/users/views/admin.py index c1210f1..165572c 100644 --- a/users/views/admin.py +++ b/users/views/admin.py @@ -1,6 +1,4 @@ import re -from functools import partial -from typing import ClassVar, Dict from django import forms from django.db import models @@ -11,6 +9,7 @@ from django.views.generic import FormView, RedirectView, TemplateView from core.models import Config from users.decorators import admin_required from users.models import Domain, Identity, User +from users.views.settings import SettingsPage @method_decorator(admin_required, name="dispatch") @@ -19,7 +18,7 @@ class AdminRoot(RedirectView): @method_decorator(admin_required, name="dispatch") -class AdminSettingsPage(FormView): +class AdminSettingsPage(SettingsPage): """ Shows a settings page dynamically created from our settings layout at the bottom of the page. Don't add this to a URL directly - subclass! @@ -27,32 +26,6 @@ class AdminSettingsPage(FormView): template_name = "admin/settings.html" options_class = Config.SystemOptions - section: ClassVar[str] - options: Dict[str, Dict[str, str]] - - def get_form_class(self): - # Create the fields dict from the config object - fields = {} - for key, details in self.options.items(): - config_field = self.options_class.__fields__[key] - if config_field.type_ is bool: - form_field = partial( - forms.BooleanField, - widget=forms.Select( - choices=[(True, "Enabled"), (False, "Disabled")] - ), - ) - elif config_field.type_ is str: - form_field = forms.CharField - else: - raise ValueError(f"Cannot render settings type {config_field.type_}") - fields[key] = form_field( - label=details["title"], - help_text=details.get("help_text", ""), - required=details.get("required", False), - ) - # Create a form class dynamically (yeah, right?) and return that - return type("SettingsForm", (forms.Form,), fields) def load_config(self): return Config.load_system() @@ -60,27 +33,6 @@ class AdminSettingsPage(FormView): def save_config(self, key, value): Config.set_system(key, value) - def get_initial(self): - config = self.load_config() - initial = {} - for key in self.options.keys(): - initial[key] = getattr(config, key) - return initial - - def get_context_data(self): - context = super().get_context_data() - context["section"] = self.section - return context - - def form_valid(self, form): - # Save each key - for field in form: - self.save_config( - field.name, - form.cleaned_data[field.name], - ) - return redirect(".") - class BasicPage(AdminSettingsPage): diff --git a/users/views/settings.py b/users/views/settings.py index 877ad01..c3c166b 100644 --- a/users/views/settings.py +++ b/users/views/settings.py @@ -1,9 +1,14 @@ +from functools import partial +from typing import ClassVar, Dict + +from django import forms +from django.shortcuts import redirect from django.utils.decorators import method_decorator -from django.views.generic import RedirectView +from django.views.generic import FormView, RedirectView +from PIL import Image, ImageOps from core.models import Config from users.decorators import identity_required -from users.views.admin import AdminSettingsPage @method_decorator(identity_required, name="dispatch") @@ -11,7 +16,8 @@ class SettingsRoot(RedirectView): url = "/settings/interface/" -class SettingsPage(AdminSettingsPage): +@method_decorator(identity_required, name="dispatch") +class SettingsPage(FormView): """ Shows a settings page dynamically created from our settings layout at the bottom of the page. Don't add this to a URL directly - subclass! @@ -19,6 +25,32 @@ class SettingsPage(AdminSettingsPage): options_class = Config.IdentityOptions template_name = "settings/settings.html" + section: ClassVar[str] + options: Dict[str, Dict[str, str]] + + def get_form_class(self): + # Create the fields dict from the config object + fields = {} + for key, details in self.options.items(): + config_field = self.options_class.__fields__[key] + if config_field.type_ is bool: + form_field = partial( + forms.BooleanField, + widget=forms.Select( + choices=[(True, "Enabled"), (False, "Disabled")] + ), + ) + elif config_field.type_ is str: + form_field = forms.CharField + else: + raise ValueError(f"Cannot render settings type {config_field.type_}") + fields[key] = form_field( + label=details["title"], + help_text=details.get("help_text", ""), + required=details.get("required", False), + ) + # Create a form class dynamically (yeah, right?) and return that + return type("SettingsForm", (forms.Form,), fields) def load_config(self): return Config.load_identity(self.request.identity) @@ -26,6 +58,27 @@ class SettingsPage(AdminSettingsPage): def save_config(self, key, value): Config.set_identity(self.request.identity, key, value) + def get_initial(self): + config = self.load_config() + initial = {} + for key in self.options.keys(): + initial[key] = getattr(config, key) + return initial + + def get_context_data(self): + context = super().get_context_data() + context["section"] = self.section + return context + + def form_valid(self, form): + # Save each key + for field in form: + self.save_config( + field.name, + form.cleaned_data[field.name], + ) + return redirect(".") + class InterfacePage(SettingsPage): @@ -37,3 +90,49 @@ class InterfacePage(SettingsPage): "help_text": "If enabled, changes all 'Post' buttons to 'Toot!'", } } + + +@method_decorator(identity_required, name="dispatch") +class ProfilePage(FormView): + """ + Lets the identity's profile be edited + """ + + template_name = "settings/profile.html" + + class form_class(forms.Form): + name = forms.CharField(max_length=500) + summary = forms.CharField(widget=forms.Textarea, required=False) + icon = forms.ImageField(required=False) + image = forms.ImageField(required=False) + + def get_initial(self): + return { + "name": self.request.identity.name, + "summary": self.request.identity.summary, + } + + def get_context_data(self): + context = super().get_context_data() + context["section"] = "profile" + return context + + def form_valid(self, form): + # Update identity name and summary + self.request.identity.name = form.cleaned_data["name"] + self.request.identity.summary = form.cleaned_data["summary"] + # Resize images + icon = form.cleaned_data.get("icon") + image = form.cleaned_data.get("image") + if icon: + resized_image = ImageOps.fit(Image.open(icon), (400, 400)) + icon.open() + resized_image.save(icon) + self.request.identity.icon = icon + if image: + resized_image = ImageOps.fit(Image.open(image), (1500, 500)) + image.open() + resized_image.save(image) + self.request.identity.image = image + self.request.identity.save() + return redirect(".") -- cgit v1.2.3