Add first alerts.

[redakcja.git] / src / alerts / models.py

diff --git a/src/alerts/models.py b/src/alerts/models.py
new file mode 100644 (file)
index 0000000..268089a
--- /dev/null
+++ b/src/alerts/models.py
@@ -0,0 +1,32 @@
+from django.db import models
+from django.dispatch import receiver
+from dvcs.models import post_commit
+from .rules import rules, rules_by_tag
+
+
+class Alert(models.Model):
+    book = models.ForeignKey('documents.Book', models.CASCADE)
+    tag = models.CharField(max_length=32)
+
+    @property
+    def rule(self):
+        return rules_by_tag[self.tag]
+
+    @classmethod
+    def validate_book(cls, book):
+        cls.objects.filter(book=book).delete()
+        try:
+            wlbook = book.wldocument(publishable=False, librarian2=True)
+        except:
+            cls.objects.create(book=book, tag='parse')
+            return
+
+        for rule in rules:
+            if rule.check_meta(wlbook.meta):
+                print(rule.tag, book)
+                cls.objects.create(book=book, tag=rule.tag)
+
+
+@receiver(post_commit)
+def validate_post_commit(sender, **kwargs):
+    Alert.validate_book(sender.tree.book)