X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/8c0d2b20aee2edeb9bd8639ee7fb1887359e9797..3afe4876ce4d6ff431eadff13a2fe24a3028f386:/lib/librarian/dcparser.py?ds=sidebyside diff --git a/lib/librarian/dcparser.py b/lib/librarian/dcparser.py index 9281325c6..557509c95 100644 --- a/lib/librarian/dcparser.py +++ b/lib/librarian/dcparser.py @@ -45,7 +45,7 @@ def str_to_unicode(value, previous): def str_to_unicode_list(value, previous): if previous is None: previous = [] - previous.append(str_to_unicode(value)) + previous.append(str_to_unicode(value, None)) return previous @@ -76,7 +76,7 @@ def str_to_date(value, previous): # ========== class ParseError(Exception): def __init__(self, message): - super(self, Exception).__init__(message) + super(ParseError, self).__init__(message) class XMLNamespace(object): @@ -117,6 +117,8 @@ class BookInfo(object): DC('source.URL') : ('source_url', str_to_unicode), DC('identifier.url') : ('url', str_to_unicode), DC('relation.hasPart') : ('parts', str_to_unicode_list), + DC('rights.license') : ('license', str_to_unicode), + DC('rights') : ('license_description', str_to_unicode), } @classmethod @@ -134,6 +136,8 @@ class BookInfo(object): raise ParseError(e) description = tree.find('//' + book_info.RDF('Description')) + book_info.wiki_url = description.get(cls.RDF('about'), None) + if description is None: raise ParseError('no Description tag found in document') @@ -157,6 +161,9 @@ class BookInfo(object): root = etree.Element(self.RDF('RDF')) description = etree.SubElement(root, self.RDF('Description')) + if self.wiki_url: + description.set(self.RDF('about'), self.wiki_url) + for tag, (attribute, converter) in self.mapping.iteritems(): if hasattr(self, attribute): e = etree.Element(tag) @@ -165,6 +172,17 @@ class BookInfo(object): return unicode(etree.tostring(root, 'utf-8'), 'utf-8') + def to_dict(self): + etree._namespace_map[str(self.RDF)] = 'rdf' + etree._namespace_map[str(self.DC)] = 'dc' + + result = {'about': self.wiki_url} + for tag, (attribute, converter) in self.mapping.iteritems(): + if hasattr(self, attribute): + result[attribute] = unicode(getattr(self, attribute)) + + return result + def parse(file_name): return BookInfo.from_file(file_name)