Experimental class-based conversion now working for TXT and HTML.

[librarian.git] / src / librarian / builders / html.py

diff --git a/src/librarian/builders/html.py b/src/librarian/builders/html.py
index 8015c6a..40d7777 100644 (file)
--- a/src/librarian/builders/html.py
+++ b/src/librarian/builders/html.py
@@ -1,4 +1,8 @@
+# coding: utf-8
+from __future__ import unicode_literals
+
 from lxml import etree
+from librarian.html import add_anchors, add_table_of_contents, add_table_of_themes
 from librarian import OutputFile
 
 
@@ -9,41 +13,52 @@ class HtmlBuilder:
     def __init__(self, image_location='https://wolnelektury.pl/media/book/pictures/marcos-historia-kolorow/'):
         self.image_location = image_location
 
-        #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.header = etree.SubElement(text, 'h1')
+
+        self.footnotes = etree.Element('div', id='footnotes')
+        self.footnote_counter = 0
+
+        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]
 
+    @property
+    def cursor(self):
+        return self.cursors[self.current_cursors[-1]]
+
+    @cursor.setter
+    def cursor(self, value):
+        self.cursors[self.current_cursors[-1]] = value
+
     def enter_fragment(self, fragment):
         self.current_cursors.append(fragment)
 
     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 preprocess(self, document):
+        document._compat_assign_ordered_ids()
+        document._compat_assign_section_ids()
+
     def build(self, document):
+        self.preprocess(document)
         document.tree.getroot().html_build(self)
+        self.postprocess(document)
 
-        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",
-        )
-        
         return OutputFile.from_bytes(
             etree.tostring(
                 self.tree,
@@ -53,20 +68,92 @@ class HtmlBuilder:
             )
         )
 
-    def start_element(self, tag, attrib):
-        self.cursors[self.current_cursors[-1]] = etree.SubElement(
-            self.cursors[self.current_cursors[-1]],
+    def postprocess(self, document):
+        _ = document.tree.getroot().master.gettext
+
+        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()
+
+        add_anchors(self.tree)
+        if len(self.nota_red):
+            self.tree.append(self.nota_red)
+        add_table_of_themes(self.tree)
+        add_table_of_contents(self.tree)
+
+        if self.footnote_counter:
+            fnheader = etree.Element("h3")
+            fnheader.text = _("Footnotes")
+            self.footnotes.insert(0, fnheader)
+            self.tree.append(self.footnotes)
+
+    def start_element(self, tag, attrib=None):
+        self.cursor = etree.SubElement(
+            self.cursor,
             tag,
-            **attrib
+            **(attrib or {})
         )
-        print(self.cursors)
 
     def end_element(self):
-        self.cursors[self.current_cursors[-1]] = self.cursors[self.current_cursors[-1]].getparent()
+        self.cursor = self.cursor.getparent()
 
     def push_text(self, text):
-        cursor = self.cursors[self.current_cursors[-1]]
+        if text == 'Między nami nic nie było':
+            print(self.cursors)
+            print(self.current_cursors)
+        cursor = self.cursor
         if len(cursor):
-            cursor.tail = (cursor[-1].tail or '') + text
+            cursor[-1].tail = (cursor[-1].tail or '') + text
         else:
             cursor.text = (cursor.text or '') + text
+
+
+class StandaloneHtmlBuilder(HtmlBuilder):
+    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"
+        )
+
+        etree.SubElement(
+            head,
+            'link',
+            href="https://static.wolnelektury.pl/css/compressed/book_text.css",
+            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"
+        )