apps/catalogue/models/listeners.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of FNP-Redakcja, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from django.contrib.auth.models import User
   7 from django.db import models
   8 from catalogue.models import (Book, Chunk, Image, BookPublishRecord,
   9         ImagePublishRecord)
  10 from catalogue.signals import post_publish
  11 from dvcs.signals import post_publishable
  12
  13
  14 def book_changed(sender, instance, created, **kwargs):
  15     instance.touch()
  16     for c in instance:
  17         c.touch()
  18 models.signals.post_save.connect(book_changed, sender=Book)
  19
  20
  21 def chunk_changed(sender, instance, created, **kwargs):
  22     instance.book.touch()
  23     instance.touch()
  24 models.signals.post_save.connect(chunk_changed, sender=Chunk)
  25
  26
  27 def image_changed(sender, instance, created, **kwargs):
  28     instance.touch()
  29 models.signals.post_save.connect(image_changed, sender=Image)
  30
  31
  32 def user_changed(sender, instance, *args, **kwargs):
  33     books = set()
  34     for c in instance.chunk_set.all():
  35         books.add(c.book)
  36         c.touch()
  37     for b in books:
  38         b.touch()
  39 models.signals.post_save.connect(user_changed, sender=User)
  40
  41
  42 def publish_listener(sender, *args, **kwargs):
  43     if isinstance(sender, BookPublishRecord):
  44         sender.book.touch()
  45         for c in sender.book:
  46             c.touch()
  47     elif isinstance(sender, ImagePublishRecord):
  48         sender.image.touch()
  49 post_publish.connect(publish_listener)
  50
  51
  52 def chunk_publishable_listener(sender, *args, **kwargs):
  53     sender.tree.touch()
  54     if isinstance(sender.tree, Chunk):
  55         sender.tree.book.touch()
  56 post_publishable.connect(chunk_publishable_listener)
  57
  58 def publishable_listener(sender, *args, **kwargs):
  59     sender.tree.touch()
  60 post_publishable.connect(publishable_listener, sender=Image)
  61
  62
  63 def listener_create(sender, instance, created, **kwargs):
  64     if created:
  65         instance.chunk_set.create(number=1, slug='1')
  66 models.signals.post_save.connect(listener_create, sender=Book)
  67