X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/0739e3055a53ad77c1c7de4352750f60453e20c1..2b96b3e346a68d086568c8fc57c1df455105c062:/apps/pdcounter/models.py?ds=sidebyside

diff --git a/apps/pdcounter/models.py b/apps/pdcounter/models.py
index f4832931f..1e441ac64 100644
--- a/apps/pdcounter/models.py
+++ b/apps/pdcounter/models.py
@@ -2,11 +2,12 @@
 # This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later.
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
+from django.conf import settings
 from django.db import models
 from django.db.models import permalink
 from django.utils.translation import ugettext as _
 from datetime import datetime
-
+from django.db.models.signals import post_save, post_delete
 
 class Author(models.Model):
     name = models.CharField(_('name'), max_length=50, db_index=True)
@@ -85,3 +86,15 @@ class BookStub(models.Model):
 
     def pretty_title(self, html_links=False):
         return ', '.join((self.author, self.title))
+
+
+if not settings.NO_SEARCH_INDEX:
+    def update_index(sender, instance, **kwargs):
+        import search
+        idx = search.Index()
+        idx.index_tags(instance, remove_only=not 'created' in kwargs)
+    
+    post_delete.connect(update_index, Author)
+    post_delete.connect(update_index, BookStub)
+    post_save.connect(update_index, Author)
+    post_save.connect(update_index, BookStub)