Add support for fabfile_local.py

[wolnelektury.git] / apps / dictionary / models.py

diff --git a/apps/dictionary/models.py b/apps/dictionary/models.py
index c0a92d1..7df3d75 100644 (file)
--- a/apps/dictionary/models.py
+++ b/apps/dictionary/models.py
@@ -3,14 +3,14 @@
 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
 #
 from django.db import models
-
-from librarian import html
+from celery.task import task
 from sortify import sortify
 
 from catalogue.models import Book
 
 
 class Note(models.Model):
+    """Represents a single annotation from a book."""
     book = models.ForeignKey(Book)
     anchor = models.CharField(max_length=64)
     html = models.TextField()
@@ -20,11 +20,16 @@ class Note(models.Model):
         ordering = ['sort_key']
 
 
-def notes_from_book(sender, **kwargs):
-    Note.objects.filter(book=sender).delete()
-    if sender.has_html_file:
-        for anchor, text_str, html_str in html.extract_annotations(sender.html_file.path):
-            Note.objects.create(book=sender, anchor=anchor,
-                               html=html_str, sort_key=sortify(text_str)[:128])
+@task(ignore_result=True)
+def build_notes(book):
+    Note.objects.filter(book=book).delete()
+    if book.html_file:
+        from librarian import html
+        for anchor, text_str, html_str in html.extract_annotations(book.html_file.path):
+            Note.objects.create(book=book, anchor=anchor,
+                               html=html_str,
+                               sort_key=sortify(text_str).strip()[:128])
 
+def notes_from_book(sender, **kwargs):
+    build_notes.delay(sender)
 Book.html_built.connect(notes_from_book)