X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/87ff0a2df51103379aee4fe2724e90f3e690ef03..d9d5b7bc7f432058fd3bccfb096c88438e5364a9:/lib/librarian/dcparser.py diff --git a/lib/librarian/dcparser.py b/lib/librarian/dcparser.py index 871ff3a92..a4106863d 100644 --- a/lib/librarian/dcparser.py +++ b/lib/librarian/dcparser.py @@ -134,6 +134,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 +159,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 +170,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)