Source code for intranet.apps.events.models

from django.conf import settings
from django.contrib.auth.models import Group as DjangoGroup
from django.core.cache import cache
from django.db import models
from django.db.models import Manager, Q
from django.db.models.signals import post_delete, post_save
from django.utils import timezone

from ...utils.date import get_date_range_this_year, is_current_year
from ...utils.deletion import set_historical_user
from ..announcements.models import Announcement
from ..eighth.models import EighthScheduledActivity
from .notifications import event_approval_request





[docs]class EventQuerySet(models.query.QuerySet):
[docs] def this_year(self): """Get Events created during this school year. Returns: Events created during this school year. """ start_date, end_date = get_date_range_this_year() return self.filter(added__gte=start_date, added__lte=end_date)
[docs]class EventManager(Manager):
[docs] def get_queryset(self): return EventQuerySet(self.model, using=self._db)
[docs] def visible_to_user(self, user): """Get a list of visible events for a given user (usually request.user). These visible events will be those that either have no groups assigned to them (and are therefore public) or those in which the user is a member. It also includes all events created by the user. Args: user (User): A User to check for Returns: Events that either have no groups assigned to them (and are therefore public), were created by the user, or are in a group. """ return Event.objects.filter(approved=True).filter(Q(groups__in=user.groups.all()) | Q(groups__isnull=True) | Q(user=user))
[docs] def hidden_events(self, user): """Get a list of events marked as hidden for a given user (usually request.user). These are all events visible to the user -- they have just decided to hide them. Args: user (User): A User to check for Returns: Events a user has hid. """ return user.events_hidden.all()
[docs]class EventUserMap(models.Model): """Represents a mapping between events and users. These attributes would be a part of the Event model, but if they are, the last updated date is changed whenever a student sees or hides an event. Access these through event.user_map If you are checking to see whether a user has hidden an event, use: >>> Event.objects.hidden_events(user) Attributes: event (Event): The one-to-one mapping between this object and the Event it is for users_hidden (UserQuerySet): A many-to-many field of Users who have hidden this event """ event = models.OneToOneField("Event", related_name="_user_map", on_delete=models.CASCADE) users_hidden = models.ManyToManyField(settings.AUTH_USER_MODEL, blank=True, related_name="events_hidden") def __str__(self): return "UserMap: {}".format(self.event.title)
[docs]class Event(models.Model): """An event available to the TJ community. Attributes: title (str): The title for the event. description (str): A description about the event. links (LinksQuerySet): Links to be attached to the event. Not currently used. added (datetime.datetime): Time created (automatically set). updated (datetime.datetime): Time last modified (automatically set). time (datetime.datetime): The date and time of the event. location (str): Where the event is located. user (User): The user who created the event. scheduled_activity (EighthScheduledActivity): An EighthScheduledActivity that should be linked with the event. announcement (Announcement): An Announcement that should be linked with the event. groups (GroupQuerySet): Groups that the event is visible to. attending (UserQuerySet): Users that are attending the event. show_attending (bool): Whether users can mark if they are attending or not attending. show_on_dashboard (bool): Whether the event will be shown on the dashboard. approved (bool): Whether the event has been approved and will be displayed. approved_by (User): The user who approved the event. rejected (bool): Whether the event was rejected and shouldn't be shown in the list of events that need to be approved. rejected_by (User): The user who rejected the event. public (bool): Whether the event is public and can be shown on the login page. category (str): The category of the event, used for ordering on the login page. open_to (bool): Whether this event is open to parents, students, or both, shown on the login page. """ objects = EventManager() title = models.CharField(max_length=100) description = models.TextField(max_length=10000) links = models.ManyToManyField("Link", blank=True) added = models.DateTimeField(auto_now=True) updated = models.DateTimeField(auto_now_add=True) time = models.DateTimeField(blank=True, null=True) location = models.CharField(max_length=100) user = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, on_delete=set_historical_user) scheduled_activity = models.ForeignKey(EighthScheduledActivity, null=True, blank=True, on_delete=models.CASCADE) announcement = models.ForeignKey(Announcement, null=True, blank=True, related_name="event", on_delete=models.CASCADE) groups = models.ManyToManyField(DjangoGroup, blank=True) attending = models.ManyToManyField(settings.AUTH_USER_MODEL, blank=True, related_name="attending") show_attending = models.BooleanField(default=True) show_on_dashboard = models.BooleanField(default=True) approved = models.BooleanField(default=False) rejected = models.BooleanField(default=False) approved_by = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, related_name="approved_event", on_delete=set_historical_user) rejected_by = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, related_name="rejected_event", on_delete=set_historical_user) public = models.BooleanField(default=True, verbose_name="Show on Login Page") category = models.CharField(max_length=6, choices=(("school", "In School"), ("sports", "Sports")), default="school") open_to = models.CharField(max_length=8, choices=(("everyone", "Everyone"), ("students", "Students"), ("parents", "Parents")), default="everyone")
[docs] def show_fuzzy_date(self): """Checks whether the event is in the next or previous 2 weeks. Returns: Whether to display the fuzzy date. """ date = self.time if date <= timezone.now(): diff = timezone.now() - date if diff.days >= 14: return False else: diff = date - timezone.now() if diff.days >= 14: return False return True
[docs] def created_hook(self, request): """Run when an event is created.""" if not request.user.has_admin_permission("events"): # Send approval email event_approval_request(request, self)
@property def is_this_year(self): """Return whether the event was created after the start of the school year.""" return is_current_year(self.added) @property def dashboard_type(self): """Return what type of object it is""" return "event" @property def pinned(self): """TODO: implement event pinning""" return False @property def user_map(self): """Return or create an EventUserMap""" try: return self._user_map # pylint: disable=no-member; Defined via a related_name in EventUserMap except EventUserMap.DoesNotExist: return EventUserMap.objects.create(event=self) @property def happened(self): """Return whether an event has happened.""" return self.time < timezone.now() def __str__(self): if not self.approved: return "UNAPPROVED - {} - {}".format(self.title, self.time) else: return "{} - {}".format(self.title, self.time) class Meta: ordering = ["time"]
[docs]class TJStarUUIDMap(models.Model): """A mapping between a user and UUID for the tjSTAR widget. Attributes: user: The associated user. uuid: The tjSTAR UUUID """ user = models.ForeignKey(settings.AUTH_USER_MODEL, on_delete=models.CASCADE) uuid = models.CharField(max_length=40)
[docs]def clear_event_cache(sender, **kwargs): # pylint: disable=unused-argument """Clears any cached event data.""" cache.delete("sports_school_events")
post_save.connect(clear_event_cache, sender=Event) post_delete.connect(clear_event_cache, sender=Event)