src/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     if 'last_login' in (kwargs.get('update_fields') or {}):
  34         # Quick hack - this change seems to result from logging user in so just ignore it.
  35         return
  36     books = set()
  37     for c in instance.chunk_set.all():
  38         books.add(c.book)
  39         c.touch()
  40     for b in books:
  41         b.touch()
  42 models.signals.post_save.connect(user_changed, sender=User)
  43
  44
  45 def publish_listener(sender, *args, **kwargs):
  46     if isinstance(sender, BookPublishRecord):
  47         sender.book.touch()
  48         for c in sender.book:
  49             c.touch()
  50     elif isinstance(sender, ImagePublishRecord):
  51         sender.image.touch()
  52 post_publish.connect(publish_listener)
  53
  54
  55 def chunk_publishable_listener(sender, *args, **kwargs):
  56     sender.tree.touch()
  57     if isinstance(sender.tree, Chunk):
  58         sender.tree.book.touch()
  59 post_publishable.connect(chunk_publishable_listener)
  60
  61 def publishable_listener(sender, *args, **kwargs):
  62     sender.tree.touch()
  63 post_publishable.connect(publishable_listener, sender=Image)
  64
  65
  66 def listener_create(sender, instance, created, **kwargs):
  67     if created:
  68         instance.chunk_set.create(number=1, slug='1')
  69 models.signals.post_save.connect(listener_create, sender=Book)
  70