Minor fixes.
[wolnelektury.git] / src / catalogue / signals.py
1 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
2 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
3 #
4 from django.conf import settings
5 from django.core.cache import caches
6 from django.core.exceptions import ImproperlyConfigured
7 from django.db.models.signals import post_save, post_delete
8 from django.dispatch import receiver
9 from ssify import flush_ssi_includes
10 from newtagging.models import tags_updated
11 from picture.models import Picture, PictureArea
12 from .models import BookMedia, Book, Collection, Fragment, Tag
13
14
15 ####
16 # BookMedia
17 ####
18
19
20 @receiver([post_save, post_delete], sender=BookMedia)
21 def bookmedia_save(sender, instance, **kwargs):
22     instance.book.set_audio_length()
23     instance.book.save()
24
25
26 ####
27 # Collection
28 ####
29
30
31 @receiver(post_save, sender=Collection)
32 def collection_save(sender, instance, **kwargs):
33     caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
34     flush_ssi_includes([
35         '/katalog/%s.json' % lang
36         for lang in [lc for (lc, _ln) in settings.LANGUAGES]])
37
38
39 @receiver(post_delete, sender=Collection)
40 def collection_delete(sender, instance, **kwargs):
41     flush_ssi_includes([
42         '/katalog/%s.json' % lang
43         for lang in [lc for (lc, _ln) in settings.LANGUAGES]])
44
45 ####
46 # Book
47 ####
48
49
50 @receiver(post_save, sender=Book)
51 def book_save(sender, instance, **kwargs):
52     # Books come out anywhere.
53     caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
54     # deleting selectively is too much work
55     try:
56         caches['template_fragments'].clear()
57     except ImproperlyConfigured:
58         pass
59     instance.flush_includes()
60
61
62 @receiver(post_delete, sender=Book)
63 def book_delete(sender, instance, **kwargs):
64     caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
65     flush_ssi_includes([
66         '/katalog/%s.json' % lang
67         for lang in [lc for (lc, _ln) in settings.LANGUAGES]])
68
69     if not settings.NO_SEARCH_INDEX:
70         # remove the book from search index, when it is deleted.
71         from search.index import Index
72         idx = Index()
73         idx.remove_book(instance)
74         idx.index_tags()
75
76
77 ####
78 # Tag
79 ####
80
81
82 @receiver(Tag.after_change)
83 def tag_after_change(sender, instance, languages, **kwargs):
84     caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
85     flush_ssi_includes([
86         '/katalog/%s.json' % lang
87         for lang in [lc for (lc, _ln) in settings.LANGUAGES]])
88
89     for model in Book, Picture:
90         for model_instance in model.tagged.with_all([instance]).only('pk'):
91             model_instance.flush_includes()
92
93     if instance.category == 'author':
94         for model in Fragment, PictureArea:
95             for model_instance in model.tagged.with_all([instance]).only('pk'):
96                 model_instance.flush_includes()
97
98
99 @receiver(tags_updated)
100 def receive_tags_updated(sender, instance, affected_tags, **kwargs):
101     categories = set(tag.category for tag in affected_tags if tag.category not in ('set', 'book'))
102     if not categories:
103         return
104
105     caches[settings.CACHE_MIDDLEWARE_ALIAS].clear()
106     instance.flush_includes()
107     flush_ssi_includes([
108         '/katalog/%s.json' % lang
109         for lang in [lc for (lc, _ln) in settings.LANGUAGES]])