takahe/users/views/settings/__init__.py

141 lines
4.6 KiB
Python
Raw Normal View History

2022-11-17 15:21:42 +00:00
from functools import partial
2022-11-18 00:43:00 +00:00
from typing import ClassVar, Dict, List
2022-11-17 15:21:42 +00:00
from django import forms
from django.core.files import File
2022-11-17 15:21:42 +00:00
from django.shortcuts import redirect
from django.utils.decorators import method_decorator
2022-11-17 15:21:42 +00:00
from django.views.generic import FormView, RedirectView
from core.models.config import Config, UploadedImage
from users.decorators import identity_required
2022-11-19 20:38:25 +00:00
from users.views.settings.profile import ProfilePage # noqa
@method_decorator(identity_required, name="dispatch")
class SettingsRoot(RedirectView):
pattern_name = "settings_profile"
2022-11-17 15:21:42 +00:00
@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!
"""
options_class = Config.IdentityOptions
template_name = "settings/settings.html"
2022-11-17 15:21:42 +00:00
section: ClassVar[str]
options: Dict[str, Dict[str, str]]
2022-11-18 00:43:00 +00:00
layout: Dict[str, List[str]]
2022-11-17 15:21:42 +00:00
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 UploadedImage:
form_field = forms.ImageField
2022-11-17 15:21:42 +00:00
elif config_field.type_ is str:
if details.get("display") == "textarea":
form_field = partial(
forms.CharField,
widget=forms.Textarea,
)
else:
form_field = forms.CharField
2022-11-18 00:43:00 +00:00
elif config_field.type_ is int:
form_field = forms.IntegerField
2022-11-17 15:21:42 +00:00
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)
def save_config(self, key, value):
Config.set_identity(self.request.identity, key, value)
2022-11-17 15:21:42 +00:00
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
2022-11-18 00:43:00 +00:00
# Gather fields into fieldsets
context["fieldsets"] = {}
for title, fields in self.layout.items():
context["fieldsets"][title] = [context["form"][field] for field in fields]
2022-11-17 15:21:42 +00:00
return context
def form_valid(self, form):
# Save each key
for field in form:
if field.field.__class__.__name__ == "ImageField":
# These can be cleared with an extra checkbox
if self.request.POST.get(f"{field.name}__clear"):
self.save_config(field.name, None)
continue
# We shove the preview values in initial_data, so only save file
# fields if they have a File object.
if not isinstance(form.cleaned_data[field.name], File):
continue
2022-11-17 15:21:42 +00:00
self.save_config(
field.name,
form.cleaned_data[field.name],
)
return redirect(".")
class InterfacePage(SettingsPage):
section = "interface"
options = {
"toot_mode": {
"title": "I Will Toot As I Please",
2022-11-18 00:43:00 +00:00
"help_text": "Changes all 'Post' buttons to 'Toot!'",
}
}
2022-11-17 15:21:42 +00:00
2022-11-18 00:43:00 +00:00
layout = {"Posting": ["toot_mode"]}
2022-11-17 15:21:42 +00:00
2022-11-18 02:36:25 +00:00
@method_decorator(identity_required, name="dispatch")
class SecurityPage(FormView):
"""
Lets the identity's profile be edited
"""
template_name = "settings/login_security.html"
extra_context = {"section": "security"}
class form_class(forms.Form):
email = forms.EmailField(
disabled=True,
help_text="Your email address cannot be changed yet.",
)
def get_initial(self):
return {"email": self.request.user.email}
template_name = "settings/login_security.html"