Misago/misago/legal/models.py

from django.db import models
from django.utils import timezone
from django.utils.translation import gettext_lazy as _

from misago.conf import settings
from misago.core.cache import cache


CACHE_KEY = "misago_agreements"


class AgreementManager(models.Manager):
    def invalidate_cache(self):
        cache.delete(CACHE_KEY)

    def get_agreements(self):
        agreements = self.get_agreements_from_cache()
        if agreements == "nada":
            agreements = self.get_agreements_from_db()
            cache.set(CACHE_KEY, agreements)
        return agreements

    def get_agreements_from_cache(self):
        return cache.get(CACHE_KEY, "nada")

    def get_agreements_from_db(self):
        agreements = {}
        for agreement in Agreement.objects.filter(is_active=True):
            agreements[agreement.type] = {
                "id": agreement.id,
                "title": agreement.get_final_title(),
                "link": agreement.link,
                "text": bool(agreement.text),
            }
        return agreements


class Agreement(models.Model):
    TYPE_TOS = "terms_of_service"
    TYPE_PRIVACY = "privacy_policy"
    TYPE_CHOICES = [
        (TYPE_TOS, _("Terms of service")),
        (TYPE_PRIVACY, _("Privacy policy")),
    ]

    type = models.CharField(
        max_length=20, default=TYPE_TOS, choices=TYPE_CHOICES, db_index=True
    )
    title = models.CharField(max_length=255, null=True, blank=True)
    link = models.URLField(max_length=255, null=True, blank=True)
    text = models.TextField(null=True, blank=True)
    is_active = models.BooleanField(default=False)
    created_on = models.DateTimeField(default=timezone.now)
    created_by = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.SET_NULL,
        blank=True,
        null=True,
        related_name="+",
    )
    created_by_name = models.CharField(max_length=255, null=True, blank=True)
    last_modified_on = models.DateTimeField(null=True, blank=True)
    last_modified_by = models.ForeignKey(
        settings.AUTH_USER_MODEL,
        on_delete=models.SET_NULL,
        blank=True,
        null=True,
        related_name="+",
    )
    last_modified_by_name = models.CharField(max_length=255, null=True, blank=True)

    objects = AgreementManager()

    def get_final_title(self):
        return self.title or self.get_type_display()

    def set_created_by(self, user):
        self.created_by = user
        self.created_by_name = user.username

    def set_last_modified_by(self, user):
        self.last_modified_by = user
        self.last_modified_by_name = user.username


class UserAgreement(models.Model):
    user = models.ForeignKey(settings.AUTH_USER_MODEL)
    agreement = models.ForeignKey(Agreement, related_name="accepted_by")
    accepted_on = models.DateTimeField(default=timezone.now)

    class Meta:
        ordering = ["-pk"]