+# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
+#
+from lxml import etree
+from librarian import FB2NS, XLINKNS, OutputFile
+from .html import TreeBuilder
+
+
+class FB2Builder(TreeBuilder):
+ file_extension = 'fb2'
+ build_method_fn = 'fb2_build'
+ orphans = False
+
+ def __init__(self, base_url=None):
+ self.tree = etree.Element(
+ FB2NS('FictionBook'),
+ nsmap={
+ None: FB2NS.uri,
+ 'l': XLINKNS.uri,
+ }
+ )
+ description = etree.SubElement(self.tree, 'description')
+ self.body = etree.SubElement(self.tree, 'body')
+ self.header = etree.SubElement(self.body, 'title')
+ self.epigraph = etree.SubElement(self.body, 'epigraph')
+ self.text = etree.SubElement(self.body, 'section')
+
+ self.footnotes = etree.Element(FB2NS('body'), name="notes")
+ self.sections = []
+
+ self.cursors = {
+ None: self.text,
+ 'meta': description,
+ 'header': self.header,
+ 'epigraph': self.epigraph,
+ 'footnotes': self.footnotes,
+ #'nota_red': self.nota_red,
+ }
+ self.current_cursors = [self.text]
+ self.add_epigraph()
+
+ def start_section(self, precedence):
+ while self.sections and self.sections[-1] >= precedence:
+ self.end_element()
+ self.sections.pop()
+ self.start_element('section')
+ self.sections.append(precedence)
+
+ def add_epigraph(self):
+ self.enter_fragment('epigraph')
+ self.start_element(FB2NS('p'))
+ self.push_text('Utwór opracowany został w\xa0ramach projektu ')
+ self.start_element(FB2NS('a'), {XLINKNS('href'): 'https://wolnelektury.pl/'})
+ self.push_text('Wolne Lektury')
+ self.end_element()
+ self.push_text(' przez ')
+ self.start_element(FB2NS('a'), {XLINKNS('href'): 'https://fundacja.wolnelektury.pl/'})
+ self.push_text('fundację Wolne Lektury')
+ self.end_element()
+ self.push_text('.')
+ self.end_element()
+ self.exit_fragment()
+
+ def add_meta(self, doc):
+ self.enter_fragment('meta')
+
+ self.start_element('title-info')
+
+ self.start_element('genre')
+ self.push_text('literature')
+ self.end_element()
+ for author in doc.meta.authors:
+ self.start_element('author')
+ self.simple_element('first-name', ' '.join(author.first_names))
+ self.simple_element('last-name', author.last_name)
+ self.end_element()
+ self.simple_element('book-title', doc.meta.title)
+ if doc.meta.released_to_public_domain_at:
+ self.simple_element('date', doc.meta.released_to_public_domain_at)
+ self.simple_element('lang', doc.meta.language)
+
+ self.end_element()
+
+ self.start_element('document-info')
+ # contributor.editor
+ # contributor.technical_editor
+ self.simple_element('program-used', 'Wolne Lektury Librarian')
+ self.simple_element('date', doc.meta.created_at)
+ self.simple_element('id', str(doc.meta.url))
+ self.simple_element('version', '0')
+
+ self.end_element()
+ self.start_element('publish-info')
+ self.simple_element('publisher', '; '.join(doc.meta.publisher))
+ self.end_element()
+ self.exit_fragment()
+
+ def build(self, doc, mp3=None):
+ self.add_meta(doc)
+ doc.tree.getroot().fb2_build(self)
+ return self.output()
+
+ def output(self):
+ return OutputFile.from_bytes(
+ etree.tostring(
+ self.tree,
+ encoding='utf-8',
+ pretty_print=True,
+ xml_declaration=True,
+ )
+ )
+
+
+
+
+'''
+import os.path
+from copy import deepcopy
+from lxml import etree
+
+from librarian import functions, OutputFile
+from .epub import replace_by_verse
+
+
+functions.reg_substitute_entities()
+functions.reg_person_name()
+
+
+def sectionify(tree):
+ """Finds section headers and adds a tree of _section tags."""
+ sections = [
+ 'naglowek_czesc',
+ 'naglowek_akt', 'naglowek_rozdzial', 'naglowek_scena',
+ 'naglowek_podrozdzial']
+ section_level = dict((v, k) for (k, v) in enumerate(sections))
+
+ # We can assume there are just subelements an no text at section level.
+ for level, section_name in reversed(list(enumerate(sections))):
+ for header in tree.findall('//' + section_name):
+ section = header.makeelement("_section")
+ header.addprevious(section)
+ section.append(header)
+ sibling = section.getnext()
+ while (sibling is not None and
+ section_level.get(sibling.tag, 1000) > level):
+ section.append(sibling)
+ sibling = section.getnext()
+
+
+def transform(wldoc, verbose=False,
+ cover=None, flags=None):
+ document = deepcopy(wldoc)
+ del wldoc
+
+ if flags:
+ for flag in flags:
+ document.edoc.getroot().set(flag, 'yes')
+
+ document.clean_ed_note()
+ document.clean_ed_note('abstrakt')
+
+ style_filename = os.path.join(os.path.dirname(__file__), 'fb2/fb2.xslt')
+ style = etree.parse(style_filename)
+
+ replace_by_verse(document.edoc)
+ sectionify(document.edoc)
+
+ result = document.transform(style)
+
+ return OutputFile.from_bytes(str(result).encode('utf-8'))
+
+# vim:et
+'''
fnp / librarian.git / commitdiff