+# This file is part of Librarian, licensed under GNU Affero GPLv3 or later.
+# Copyright © Fundacja Wolne Lektury. See NOTICE for more information.
+#
+from collections import defaultdict
+import os
+from urllib.request import urlopen
from lxml import etree
+from librarian.html import add_table_of_contents, add_table_of_themes, add_image_sizes
from librarian import OutputFile
-class HtmlBuilder:
+class TreeBuilder:
+ @property
+ def cursor(self):
+ return self.current_cursors[-1]
+
+ def enter_fragment(self, fragment):
+ cursor = self.cursors.get(fragment, self.cursor)
+ self.current_cursors.append(cursor)
+
+ def exit_fragment(self):
+ self.current_cursors.pop()
+
+ def create_fragment(self, name, element):
+ assert name not in self.cursors
+ self.cursors[name] = element
+
+ def forget_fragment(self, name):
+ del self.cursors[name]
+
+ def start_element(self, tag, attrib=None):
+ self.current_cursors.append(etree.SubElement(
+ self.cursor,
+ tag,
+ **(attrib or {})
+ ))
+
+ def end_element(self):
+ self.current_cursors.pop()
+
+ def push_text(self, text):
+ cursor = self.cursor
+ if len(cursor):
+ cursor[-1].tail = (cursor[-1].tail or '') + text
+ else:
+ cursor.text = (cursor.text or '') + text
+
+ def simple_element(self, tag, text='', attrib=None):
+ self.start_element(tag, attrib)
+ self.push_text(text)
+ self.end_element()
+
+
+class HtmlBuilder(TreeBuilder):
+ build_method_fn = 'html_build'
file_extension = "html"
- identifier = "html"
+ with_themes = True
+ with_toc = True
+ with_footnotes = True
+ with_nota_red = True
+ with_ids = True
+ with_numbering = True
+ no_externalities = False
+ orphans = True
+
+ root_tag = 'div'
+ root_attrib = {'id': 'book-text'}
+
+ def __init__(self, gallery_path=None, gallery_url=None, base_url=None):
+ self._base_url = base_url
+ self.gallery_path = gallery_path
+ self.gallery_url = gallery_url
- def __init__(self, image_location='https://wolnelektury.pl/media/book/pictures/marcos-historia-kolorow/'):
- self.image_location = image_location
+ self.tree = text = etree.Element(self.root_tag, **self.root_attrib)
+ self.header = etree.Element('h1')
- #self.tree = etree.Element('html')
- #body = etree.SubElement(self.tree, 'body')
- #text = etree.SubElement(body, 'div', **{'id': 'book-text'})
- self.tree = text = etree.Element('div', **{'id': 'book-text'})
- toc = etree.SubElement(text, 'div', id='toc')
- themes = etree.SubElement(text, 'div', id='themes')
- h1 = etree.SubElement(text, 'h1')
+ self.footnotes = etree.Element('div', id='footnotes')
+
+ self.nota_red = etree.Element('div', id='nota_red')
self.cursors = {
None: text,
- 'toc': toc,
- 'themes': themes,
- 'header': h1,
+ 'header': self.header,
+ 'footnotes': self.footnotes,
+ 'nota_red': self.nota_red,
}
- self.current_cursors = [None]
+ self.current_cursors = [text]
- def enter_fragment(self, fragment):
- self.current_cursors.append(fragment)
+ @property
+ def base_url(self):
+ if self._base_url is not None:
+ return self._base_url
+ else:
+ return 'https://wolnelektury.pl/media/book/pictures/{}/'.format(self.document.meta.url.slug)
- def exit_fragment(self):
- self.current_cursors.pop()
-
- def build(self, document):
- document.tree.getroot().html_build(self)
+ def build(self, document, element=None, **kwargs):
+ self.document = document
+ self.document.assign_ids()
+ self.prepare_images()
- head = etree.Element('head')
- self.tree.insert(0, head)
- etree.SubElement(
- head,
- 'link',
- href="https://static.wolnelektury.pl/css/compressed/book_text.b15153e56c0a.css",
- rel="stylesheet",
- type="text/css",
- )
-
+ if element is None:
+ element = document.tree.getroot()
+
+ element.html_build(self)
+ self.postprocess(document)
+ return self.output()
+
+ def prepare_images(self):
+ # Temporarily use the legacy method, before transitioning to external generators.
+ if self.gallery_path is None:
+ return
+ try:
+ os.makedirs(self.gallery_path)
+ except OSError:
+ pass
+ add_image_sizes(self.document.tree, self.gallery_path, self.gallery_url, self.base_url)
+
+ def output(self):
+ if not len(self.tree):
+ return None
return OutputFile.from_bytes(
etree.tostring(
self.tree,
)
)
- def start_element(self, tag, attrib):
- self.cursors[self.current_cursors[-1]] = etree.SubElement(
- self.cursors[self.current_cursors[-1]],
- tag,
- **attrib
- )
- print(self.cursors)
+ def postprocess(self, document):
+ _ = document.tree.getroot().gettext
- def end_element(self):
- self.cursors[self.current_cursors[-1]] = self.cursors[self.current_cursors[-1]].getparent()
+ if document.meta.translators:
+ self.enter_fragment('header')
+ self.start_element('span', {'class': 'translator'})
+ self.push_text(_("translated by") + " ")
+ self.push_text(
+ ", ".join(
+ translator.readable()
+ for translator in document.meta.translators
+ )
+ )
+ self.exit_fragment()
- def push_text(self, text):
- cursor = self.cursors[self.current_cursors[-1]]
- if len(cursor):
- cursor.tail = (cursor[-1].tail or '') + text
+ if len(self.header):
+ self.tree.insert(0, self.header)
+
+ if self.with_nota_red and len(self.nota_red):
+ self.tree.append(self.nota_red)
+ if self.with_themes:
+ add_table_of_themes(self.tree)
+ if self.with_toc:
+ add_table_of_contents(self.tree)
+
+ if len(self.footnotes):
+ fnheader = etree.Element("h3")
+ fnheader.text = _("Footnotes")
+ self.footnotes.insert(0, fnheader)
+ self.tree.append(self.footnotes)
+
+ def add_visible_number(self, element):
+ assert '_id' in element.attrib, etree.tostring(element)
+ self.start_element('a', {
+ 'href': f'#{element.attrib["_id"]}',
+ 'class': 'wl-num',
+ })
+ self.push_text(element.attrib['_visible_numbering'])
+ self.end_element()
+
+
+class StandaloneHtmlBuilder(HtmlBuilder):
+ css_url = "https://static.wolnelektury.pl/css/compressed/book_text.css"
+
+ def postprocess(self, document):
+ super(StandaloneHtmlBuilder, self).postprocess(document)
+
+ tree = etree.Element('html')
+ body = etree.SubElement(tree, 'body')
+ body.append(self.tree)
+ self.tree = tree
+
+ head = etree.Element('head')
+ tree.insert(0, head)
+
+ etree.SubElement(head, 'meta', charset='utf-8')
+ etree.SubElement(head, 'title').text = document.meta.title
+
+ etree.SubElement(
+ head,
+ 'meta',
+ name="viewport",
+ content="width=device-width, initial-scale=1, maximum-scale=1"
+ )
+
+ if self.no_externalities:
+ etree.SubElement(
+ head, 'style',
+ ).text = urlopen(self.css_url).read().decode('utf-8')
else:
- cursor.text = (cursor.text or '') + text
+ etree.SubElement(
+ head,
+ 'link',
+ href=self.css_url,
+ rel="stylesheet",
+ type="text/css",
+ )
+
+ etree.SubElement(
+ body, 'script',
+ src="https://ajax.googleapis.com/ajax/libs/jquery/1/jquery.min.js"
+ )
+
+ etree.SubElement(
+ body,
+ "script",
+ src="http://malsup.github.io/min/jquery.cycle2.min.js"
+ )
+
+
+class SnippetHtmlBuilder(HtmlBuilder):
+ with_themes = False
+ with_toc = False
+ with_footnotes = False
+ with_nota_red = False
+ with_ids = False
+ with_numbering = False
+
+
+class AbstraktHtmlBuilder(HtmlBuilder):
+ with_themes = False
+ with_toc = False
+ with_footnotes = False
+ with_nota_red = False
+ with_ids = False
+ with_numbering = False
+
+ root_tag = 'blockquote'
+ root_attrib = {}
+
+ def build(self, document, element=None, **kwargs):
+ if element is None:
+ element = document.tree.find('//abstrakt')
+ if element is None:
+ return OutputFile.from_bytes(b'')
+ element.attrib['_force'] = '1'
+ return super().build(document, element, **kwargs)
+
+
+class DaisyHtmlBuilder(StandaloneHtmlBuilder):
+ file_extension = 'xhtml'
+ with_themes = False
+ with_toc = False
+ with_footnotes = False
+ with_nota_red = False
+ with_deep_identifiers = False
+ no_externalities = True
+ with_numbering = False
+
+ def output(self):
+ tree = etree.ElementTree(self.tree)
+ tree.docinfo.public_id = '-//W3C//DTD XHTML 1.0 Transitional//EN'
+ tree.docinfo.system_url = 'http://www.w3.org/TR/xhtml1/DTD/xhtml1-transitional.dtd'
+ return OutputFile.from_bytes(
+ etree.tostring(
+ tree,
+ encoding='utf-8',
+ pretty_print=True,
+ xml_declaration=True
+ )
+ )
+
fnp / librarian.git / blobdiff