apps/catalogue/models/document.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of MIL/PEER, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from __future__ import unicode_literals
   7
   8 from datetime import date
   9 from django.conf import settings
  10 from django.core.exceptions import ObjectDoesNotExist, MultipleObjectsReturned
  11 from django.db import models
  12 from django.template.loader import render_to_string
  13 from django.utils.encoding import force_unicode
  14 from django.utils.translation import ugettext_lazy as _
  15 from dvcs.models import Ref
  16 from organizations.models import Organization
  17 from catalogue.constants import STAGES
  18 from .tag import Tag, Category
  19
  20
  21 def metadata_from_text(text):
  22     from lxml import etree
  23     metadata = {}
  24     text = text.replace(u'\ufeff', '')
  25     # This is bad. The editor shouldn't spew unknown HTML entities.
  26     text = text.replace(u'&nbsp;', u'\u00a0')
  27
  28     try:
  29         t = etree.fromstring(text)
  30     except:
  31         return {'title': '<<Resource invalid>>'}
  32     header = t.find('.//header')
  33     if header is None:
  34         header = etree.fromstring(text).find('.//{http://nowoczesnapolska.org.pl/sst#}header')
  35     metadata['title'] = getattr(header, 'text', ' ') or ' '
  36     # print 'meta', d['title']
  37
  38     m = t.find('metadata')
  39     if m is None:
  40         m = t.find('{http://nowoczesnapolska.org.pl/sst#}metadata')
  41     if m is not None:
  42         c = m.find('{http://purl.org/dc/elements/1.1/}relation.coverimage.url')
  43         if c is not None:
  44             metadata['cover_url'] = c.text
  45         for category in Category.objects.all():
  46             c = m.find('{http://purl.org/dc/elements/1.1/}' + category.dc_tag)
  47             if c is not None:
  48                 if category.multiple:
  49                     if category.dc_tag not in metadata:
  50                         metadata[category.dc_tag] = []
  51                     metadata[category.dc_tag].append(c.text)
  52                 else:
  53                     metadata[category.dc_tag] = c.text
  54     return metadata
  55
  56
  57 class Document(Ref):
  58     """ An editable chunk of text."""
  59
  60     owner_user = models.ForeignKey(settings.AUTH_USER_MODEL, null=True)
  61     owner_organization = models.ForeignKey(Organization, null=True)
  62     stage = models.CharField(_('stage'), max_length=128, blank=True, default=STAGES[0][0])
  63     assigned_to = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, related_name='assignments')
  64     deleted = models.BooleanField(default=False)
  65     tags = models.ManyToManyField(Tag, blank=True)
  66     # we need to know if it were ever published (for notifications)
  67     published = models.BooleanField(default=False)
  68
  69     # Where to cache searchable stuff from metadata?
  70     # Probably in some kind of search index.
  71
  72     class Meta:
  73         verbose_name = _('document')
  74         verbose_name_plural = _('documents')
  75
  76     def short_html(self):
  77         return render_to_string('catalogue/book_list/book.html', {'book': self})
  78
  79     def meta(self):
  80         return metadata_from_text(self.materialize())
  81
  82     def can_edit(self, user):
  83         if user.is_superuser:
  84             return True
  85         if self.owner_user:
  86             return self.owner_user == user
  87         else:
  88             return self.owner_organization.is_member(user)
  89
  90     def set_stage(self, stage):
  91         self.stage = stage
  92         plan = self.get_plan()
  93         if plan is not None:
  94             self.assigned_to = plan.user
  95         else:
  96             self.assigned_to = None
  97         self.save()
  98
  99     def stage_name(self):
 100         return force_unicode(dict(STAGES)[self.stage]) if self.stage else None
 101
 102     def get_plan(self):
 103         try:
 104             plan = self.plan_set.get(stage=self.stage)
 105         except (ObjectDoesNotExist, MultipleObjectsReturned):
 106             return None
 107         return plan
 108
 109     def is_overdue(self):
 110         plan = self.get_plan()
 111         return plan is not None and plan.deadline and plan.deadline < date.today()
 112
 113     def commit(self, *args, **kwargs):
 114         super(Document, self).commit(*args, **kwargs)
 115         m = self.meta()
 116         for category in Category.objects.all():
 117             values = m.get(category.dc_tag)
 118             if not category.multiple:
 119                 values = [values]
 120             if not values:
 121                 values = []
 122             tags = category.tag_set.filter(dc_value__in=values)
 123             category.set_tags_for(self, tags)