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.db import models
  11 from django.template.loader import render_to_string
  12 from django.utils.translation import ugettext_lazy as _
  13 from dvcs.models import Ref
  14 from organizations.models import Organization
  15 from catalogue.constants import STAGES
  16
  17
  18 class Document(Ref):
  19     """ An editable chunk of text."""
  20
  21     owner_user = models.ForeignKey(settings.AUTH_USER_MODEL, null=True)
  22     owner_organization = models.ForeignKey(Organization, null=True)
  23     stage = models.CharField(_('stage'), max_length=128, blank=True, default=STAGES[0])
  24     assigned_to = models.ForeignKey(settings.AUTH_USER_MODEL, null=True, related_name='assignments')
  25     deleted = models.BooleanField(default=False)
  26
  27     # Where to cache searchable stuff from metadata?
  28     # Probably in some kind of search index.
  29
  30     class Meta:
  31         verbose_name = _('document')
  32         verbose_name_plural = _('documents')
  33
  34     def short_html(self):
  35         return render_to_string('catalogue/book_list/book.html', {'book': self})
  36
  37     def meta(self):
  38         from lxml import etree
  39         # Wrong! should be metadata
  40         d = {}
  41
  42         data = self.materialize()
  43         data = data.replace(u'\ufeff', '')
  44         # This is bad. The editor shouldn't spew unknown HTML entities.
  45         data = data.replace(u'&nbsp;', u'\u00a0')
  46
  47         try:
  48             t = etree.fromstring(data)
  49         except:
  50             return {'title': '<<Resource invalid>>'}
  51         header = t.find('.//header')
  52         if header is None:
  53             header = etree.fromstring(data).find('.//{http://nowoczesnapolska.org.pl/sst#}header')
  54         d['title'] = getattr(header, 'text', ' ') or ' '
  55         #print 'meta', d['title']
  56
  57         m = t.find('metadata')
  58         if m is None:
  59             m = t.find('{http://nowoczesnapolska.org.pl/sst#}metadata')
  60         if m is not None:
  61             c = m.find('{http://purl.org/dc/elements/1.1/}relation.coverimage.url')
  62             if c is not None:
  63                 d['cover_url'] = c.text
  64             c = m.find('{http://purl.org/dc/elements/1.1/}audience')
  65             if c is not None:
  66                 d['audience'] = c.text
  67
  68         return d
  69
  70     def can_edit(self, user):
  71         if self.owner_user:
  72             return self.owner_user == user
  73         else:
  74             return self.owner_organization.is_member(user)
  75
  76     def set_stage(self, stage):
  77         self.stage = stage
  78         plan = self.get_plan()
  79         if plan is not None:
  80             self.assigned_to = plan.user
  81         else:
  82             self.assigned_to = None
  83         self.save()
  84
  85     def get_plan(self):
  86         try:
  87             plan = self.plan_set.get(stage=self.stage)
  88         except:
  89             return None
  90         return plan
  91
  92     def is_overdue(self):
  93         plan = self.get_plan()
  94         return plan is not None and plan.deadline and plan.deadline < date.today()