librarian/document.py

   1 # -*- coding: utf-8 -*-
   2 #
   3 # This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
   4 # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information.
   5 #
   6 from StringIO import StringIO
   7 from lxml import etree
   8 from . import SSTNS
   9 from .core import Section
  10 from .parser import SSTParser
  11
  12
  13 class Document(object):
  14     # Do I use meta_context?
  15     def __init__(self, edoc, meta_context=None):
  16         self.edoc = edoc
  17
  18         root_elem = edoc.getroot()
  19         if meta_context is not None:
  20             root_elem.meta_context = meta_context
  21
  22         if not isinstance(root_elem, Section):
  23             if root_elem.tag != SSTNS('section'):
  24                 if root_elem.tag == 'section':
  25                     for element in root_elem.iter():
  26                         if element.tag in ('section', 'header', 'div', 'span', 'aside', 'metadata'):
  27                             element.tag = str(SSTNS(element.tag))
  28
  29                     parser = SSTParser()
  30                     tree = etree.parse(StringIO(etree.tostring(root_elem)), parser)
  31                     tree.xinclude()
  32                     self.edoc = tree
  33                 else:
  34                     raise ValueError("Invalid root element. Found '%s', should be '%s'" % (
  35                         root_elem.tag, SSTNS('section')))
  36             else:
  37                 raise ValueError("Invalid class of root element. Use librarian.parser.SSTParser.")
  38         # print etree.tostring(self.edoc.getroot())
  39
  40     @classmethod
  41     def from_string(cls, xml, *args, **kwargs):
  42         return cls.from_file(StringIO(xml), *args, **kwargs)
  43
  44     @classmethod
  45     def from_file(cls, xmlfile, *args, **kwargs):
  46         # first, prepare for parsing
  47         if isinstance(xmlfile, basestring):
  48             file = open(xmlfile, 'rb')
  49             try:
  50                 data = file.read()
  51             finally:
  52                 file.close()
  53         else:
  54             data = xmlfile.read()
  55
  56         if not isinstance(data, unicode):
  57             data = data.decode('utf-8')
  58
  59         data = data.replace(u'\ufeff', '')
  60         # This is bad. The editor shouldn't spew unknown HTML entities.
  61         data = data.replace(u'&nbsp;', u'\u00a0')
  62
  63         parser = SSTParser()
  64         tree = etree.parse(StringIO(data.encode('utf-8')), parser)
  65         tree.xinclude()
  66         return cls(tree, *args, **kwargs)
  67
  68     @property
  69     def meta(self):
  70         """ Document's metadata is root's metadata. """
  71         return self.edoc.getroot().meta