svthalia/concrexit

from django.conf import settings
from django.db.models.signals import post_save
from django.utils import timezone

from members.models import Member
from utils.models.signals import suspendingreceiver

from ..models import Category, NewAlbumMessage


@suspendingreceiver(
    post_save,
    sender="photos.Album",
    dispatch_uid="schedule_new_album_pushnotification",
)
def schedule_new_album_pushnotification(sender, instance, **kwargs):
    """Create, update or delete a scheduled message for the saved album if necessary."""
    message = getattr(instance, "new_album_notification", None)

    if instance.hidden or instance.is_processing:
        # Remove existing not-sent notification if there is one.
        if message is not None and not message.sent:
            instance.new_album_notification = None
            message.delete()
    elif message is None or not message.sent:
        # Update existing notification or create new one.

        if message is None:
            message = NewAlbumMessage(album=instance)

        message.title = "New album uploaded"
        message.body = f"A new photo album '{instance.title}' has just been uploaded"
        message.url = f"{settings.BASE_URL}{instance.get_absolute_url()}"
        message.time = timezone.now() + timezone.timedelta(hours=1)
        message.category = Category.objects.get(key=Category.PHOTO)
        message.save()

        message.users.set(Member.current_members.all())