moderator/moderate/models.py

from datetime import timedelta

from django.contrib.auth.models import User
from django.core.validators import MaxLengthValidator, MinLengthValidator
from django.db import models
from django.db.models import signals as dbsignals
from django.dispatch import receiver
from django.utils.timezone import now as django_now
from uuslug import uuslug


def default_date():
    """Default date."""
    return django_now().date() + timedelta(days=1)


class MozillianProfile(models.Model):
    """Mozillians User Profile"""

    user = models.OneToOneField(
        User, related_name="userprofile", on_delete=models.CASCADE
    )
    slug = models.SlugField(blank=True, max_length=100)
    username = models.CharField(max_length=40)
    avatar_url = models.URLField(max_length=400, default="", blank=True)
    is_nda_member = models.BooleanField(default=False)

    def __str__(self):
        return self.username

    class Meta:
        ordering = ["username"]

    def save(self, *args, **kwargs):
        if not self.pk:
            self.username = self.user.email.split("@")[0]
        super(MozillianProfile, self).save(*args, **kwargs)

    @property
    def is_admin(self):
        return self.user.groups.filter(name="Admin").exists() or self.user.is_superuser


def get_user_full_name(self):
    if self.first_name or self.last_name:
        return f"{self.first_name} {self.last_name}"
    return self.username


User.add_to_class("__str__", get_user_full_name)


@receiver(dbsignals.post_save, sender=User, dispatch_uid="create_user_profile_sig")
def create_user_profile(sender, instance, created, raw, **kwargs):
    if not raw:
        up, created = MozillianProfile.objects.get_or_create(user=instance)
        if not created:
            dbsignals.post_save.send(
                sender=MozillianProfile, instance=up, created=created, raw=raw
            )


class Event(models.Model):
    """Event model."""

    name = models.CharField(max_length=400)
    body = models.TextField(
        help_text="Optional: Helpful links, additional information",
        blank=True,
        default="",
    )
    archived = models.BooleanField(default=False)
    slug = models.SlugField(max_length=400, blank=True)
    created_at = models.DateTimeField(auto_now_add=True)
    event_date = models.DateField(null=True, blank=True, default=default_date)
    is_nda = models.BooleanField(default=False)
    created_by = models.ForeignKey(
        User, null=True, blank=True, on_delete=models.SET_NULL
    )
    moderators = models.ManyToManyField(User, related_name="events_moderated")
    is_moderated = models.BooleanField(default=False)
    users_can_vote = models.BooleanField(default=True)

    class Meta:
        ordering = ["-created_at"]

    def __str__(self):
        return self.name

    def save(self, *args, **kwargs):
        if not self.pk:
            self.slug = uuslug(self.name, instance=self)
        super().save(*args, **kwargs)

    @property
    def questions_count(self):
        return self.questions.all().count()


class Question(models.Model):
    """Question relational model."""

    ACCEPTANCE_CHOICES = [
        (True, "Accepted"),
        (False, "Rejected"),
        (None, "Pending"),
    ]
    asked_by = models.ForeignKey(User, null=True, blank=True, on_delete=models.SET_NULL)
    event = models.ForeignKey(Event, related_name="questions", on_delete=models.CASCADE)
    question = models.TextField(
        validators=[MaxLengthValidator(280), MinLengthValidator(10)]
    )
    answer = models.TextField(
        validators=[MaxLengthValidator(2500)], default="", blank=True
    )
    addressed = models.BooleanField(default=False)
    is_anonymous = models.BooleanField(default=False, blank=False)
    submitter_contact_info = models.EmailField(max_length=256, default="", blank=True)
    # Default value is None, which means that moderation is still pending
    is_accepted = models.BooleanField(
        blank=True, null=True, default=None, choices=ACCEPTANCE_CHOICES
    )
    rejection_reason = models.TextField(
        default="",
        blank=True,
        validators=[MinLengthValidator(10)],
    )

    def __str__(self):
        return "Question {pk} from {user}".format(pk=self.id, user=self.asked_by)

    @property
    def is_moderated(self):
        return self.event.is_moderated and (
            self.is_accepted is False or self.is_accepted is True
        )

    @property
    def has_contact_info(self):
        """Whether contact info has been provided upon submission."""
        return bool(self.submitter_contact_info)


class Vote(models.Model):
    """Vote model."""

    user = models.ForeignKey(User, on_delete=models.CASCADE)
    question = models.ForeignKey(
        Question, related_name="votes", on_delete=models.CASCADE
    )
    date_voted = models.DateField(auto_now_add=True)

    class Meta:
        unique_together = ("user", "question")

    def __str__(self):
        return "Vote of {user} for {question}".format(
            user=self.user, question=self.question
        )