wagtail/wagtail/users/models.py

115 wiersze
3.4 KiB
Python

import os
import uuid
from django.conf import settings
from django.db import models
from django.utils.translation import get_language
from django.utils.translation import gettext_lazy as _
def upload_avatar_to(instance, filename):
filename, ext = os.path.splitext(filename)
return os.path.join(
"avatar_images",
"avatar_{uuid}_{filename}{ext}".format(
uuid=uuid.uuid4(), filename=filename, ext=ext
),
)
class UserProfile(models.Model):
user = models.OneToOneField(
settings.AUTH_USER_MODEL,
on_delete=models.CASCADE,
related_name="wagtail_userprofile",
)
submitted_notifications = models.BooleanField(
verbose_name=_("submitted notifications"),
default=True,
help_text=_("Receive notification when a page is submitted for moderation"),
)
approved_notifications = models.BooleanField(
verbose_name=_("approved notifications"),
default=True,
help_text=_("Receive notification when your page edit is approved"),
)
rejected_notifications = models.BooleanField(
verbose_name=_("rejected notifications"),
default=True,
help_text=_("Receive notification when your page edit is rejected"),
)
updated_comments_notifications = models.BooleanField(
verbose_name=_("updated comments notifications"),
default=True,
help_text=_(
"Receive notification when comments have been created, resolved, or deleted on a page that you have subscribed to receive comment notifications on"
),
)
preferred_language = models.CharField(
verbose_name=_("preferred language"),
max_length=10,
help_text=_("Select language for the admin"),
default="",
)
current_time_zone = models.CharField(
verbose_name=_("current time zone"),
max_length=40,
help_text=_("Select your current time zone"),
default="",
)
avatar = models.ImageField(
verbose_name=_("profile picture"),
upload_to=upload_avatar_to,
blank=True,
)
dismissibles = models.JSONField(default=dict, blank=True)
class AdminColorThemes(models.TextChoices):
SYSTEM = "system", _("System default")
LIGHT = "light", _("Light")
DARK = "dark", _("Dark")
theme = models.CharField(
verbose_name=_("admin theme"),
choices=AdminColorThemes.choices,
default=AdminColorThemes.SYSTEM,
max_length=40,
)
class AdminDensityThemes(models.TextChoices):
DEFAULT = "default", _("Default")
SNUG = "snug", _("Snug")
density = models.CharField(
# Translators: "Density" is the term used to describe the amount of space between elements in the user interface
verbose_name=_("density"),
choices=AdminDensityThemes.choices,
default=AdminDensityThemes.DEFAULT,
max_length=40,
)
@classmethod
def get_for_user(cls, user):
return cls.objects.get_or_create(user=user)[0]
def get_preferred_language(self):
return self.preferred_language or get_language()
def get_current_time_zone(self):
return self.current_time_zone or settings.TIME_ZONE
def __str__(self):
return self.user.get_username()
class Meta:
verbose_name = _("user profile")
verbose_name_plural = _("user profiles")