mybuddy/babybuddy/models.py

108 lines
3.7 KiB
Python
Raw Normal View History

2017-11-11 22:27:42 +00:00
# -*- coding: utf-8 -*-
2024-01-14 04:22:08 +00:00
import zoneinfo
2020-02-14 17:22:25 +00:00
from django.conf import settings
from django.contrib.auth import get_user_model
2017-11-11 22:27:42 +00:00
from django.db import models
from django.db.models.signals import post_save
from django.dispatch import receiver
from django.utils import timezone
from django.utils.text import format_lazy
2019-04-19 02:55:55 +00:00
from django.utils.translation import gettext_lazy as _
2017-11-11 22:27:42 +00:00
from rest_framework.authtoken.models import Token
2017-11-11 22:27:42 +00:00
class Settings(models.Model):
user = models.OneToOneField(settings.AUTH_USER_MODEL, on_delete=models.CASCADE)
2017-11-11 22:27:42 +00:00
dashboard_refresh_rate = models.DurationField(
2022-02-10 00:00:30 +00:00
verbose_name=_("Refresh rate"),
help_text=_(
"If supported by browser, the dashboard will only refresh "
"when visible, and also when receiving focus."
),
2017-11-11 22:27:42 +00:00
blank=True,
null=True,
2020-02-14 17:22:25 +00:00
default=timezone.timedelta(minutes=1),
2017-11-11 22:27:42 +00:00
choices=[
2022-02-10 00:00:30 +00:00
(None, _("disabled")),
(timezone.timedelta(minutes=1), _("1 min.")),
(timezone.timedelta(minutes=2), _("2 min.")),
(timezone.timedelta(minutes=3), _("3 min.")),
(timezone.timedelta(minutes=4), _("4 min.")),
(timezone.timedelta(minutes=5), _("5 min.")),
(timezone.timedelta(minutes=10), _("10 min.")),
(timezone.timedelta(minutes=15), _("15 min.")),
(timezone.timedelta(minutes=30), _("30 min.")),
],
)
dashboard_hide_empty = models.BooleanField(
2022-02-10 00:00:30 +00:00
verbose_name=_("Hide Empty Dashboard Cards"), default=False, editable=True
)
dashboard_hide_age = models.DurationField(
2022-02-10 00:00:30 +00:00
verbose_name=_("Hide data older than"),
help_text=_(
"This setting controls which data will be shown " "in the dashboard."
),
blank=True,
null=True,
default=None,
choices=[
2022-02-10 00:00:30 +00:00
(None, _("show all data")),
(timezone.timedelta(days=1), _("1 day")),
(timezone.timedelta(days=2), _("2 days")),
(timezone.timedelta(days=3), _("3 days")),
(timezone.timedelta(weeks=1), _("1 week")),
(timezone.timedelta(weeks=4), _("4 weeks")),
],
)
language = models.CharField(
choices=settings.LANGUAGES,
default=settings.LANGUAGE_CODE,
max_length=255,
2022-02-10 00:00:30 +00:00
verbose_name=_("Language"),
)
2020-02-14 17:22:25 +00:00
timezone = models.CharField(
2024-01-14 04:22:08 +00:00
choices=sorted(
tuple(zip(zoneinfo.available_timezones(), zoneinfo.available_timezones()))
),
2020-02-14 17:22:25 +00:00
default=timezone.get_default_timezone_name(),
max_length=100,
2022-02-10 00:00:30 +00:00
verbose_name=_("Timezone"),
2020-02-14 17:22:25 +00:00
)
2017-11-11 22:27:42 +00:00
def __str__(self):
2022-02-10 00:00:30 +00:00
return str(format_lazy(_("{user}'s Settings"), user=self.user))
2017-11-11 22:27:42 +00:00
def api_key(self, reset=False):
"""
Get or create an API key for the associated user.
:param reset: If True, delete the existing key and create a new one.
:return: The user's API key.
"""
if reset:
Token.objects.get(user=self.user).delete()
return Token.objects.get_or_create(user=self.user)[0]
2017-11-11 22:27:42 +00:00
@property
def dashboard_refresh_rate_milliseconds(self):
"""
Convert seconds to milliseconds to be used in a Javascript setInterval
function call.
:return: the refresh rate in milliseconds or None.
"""
if self.dashboard_refresh_rate:
return self.dashboard_refresh_rate.seconds * 1000
return None
@receiver(post_save, sender=get_user_model())
2017-11-11 22:27:42 +00:00
def create_user_settings(sender, instance, created, **kwargs):
if created:
Settings.objects.create(user=instance)
@receiver(post_save, sender=get_user_model())
2017-11-11 22:27:42 +00:00
def save_user_settings(sender, instance, **kwargs):
instance.settings.save()