Fixed importing books.
[wolnelektury.git] / lib / librarian / html.py
index ae5efa6..3dd70f5 100644 (file)
@@ -53,6 +53,8 @@ def transform(input_filename, output_filename):
     doc = etree.parse(doc_file, parser)
 
     result = doc.xslt(style)
     doc = etree.parse(doc_file, parser)
 
     result = doc.xslt(style)
+    add_anchors(result.getroot())
+    add_table_of_contents(result.getroot())
     result.write(output_filename, xml_declaration=True, pretty_print=True, encoding='utf-8')
 
 
     result.write(output_filename, xml_declaration=True, pretty_print=True, encoding='utf-8')
 
 
@@ -107,7 +109,7 @@ def extract_fragments(input_filename):
 
     for event, element in etree.iterparse(input_filename, events=('start', 'end')):
         # Process begin and end elements
 
     for event, element in etree.iterparse(input_filename, events=('start', 'end')):
         # Process begin and end elements
-        if element.tag == 'span' and element.get('class', '') in ('theme-begin', 'theme-end'):
+        if element.get('class', '') in ('theme-begin', 'theme-end'):
             if not event == 'end': continue # Process elements only once, on end event
 
             # Open new fragment
             if not event == 'end': continue # Process elements only once, on end event
 
             # Open new fragment
@@ -155,3 +157,66 @@ def extract_fragments(input_filename):
 
     return closed_fragments, open_fragments
 
 
     return closed_fragments, open_fragments
 
+
+def add_anchor(element, number):
+    anchor = etree.Element('a', href='#f%d' % number)
+    anchor.set('class', 'anchor')
+    anchor.text = str(number)
+    if element.text:
+        anchor.tail = element.text
+        element.text = u''
+    element.insert(0, anchor)
+    
+    anchor_target = etree.Element('a', name='f%d' % number)
+    element.insert(0, anchor_target)
+
+
+def any_ancestor(element, test):
+    for ancestor in element.iterancestors():
+        if test(ancestor):
+            return True
+    return False
+
+
+def add_anchors(root):
+    counter = 1
+    for element in root.iterdescendants():
+        if any_ancestor(element, lambda e: e.get('class') in ('note', 'motto', 'motto_podpis', 'dedication')
+        or e.tag == 'blockquote'):
+            continue
+        
+        if element.tag == 'p' and 'verse' in element.get('class', ''):
+            if counter == 1 or counter % 5 == 0:
+                add_anchor(element, counter)
+            counter += 1
+        elif 'paragraph' in element.get('class', ''):
+            add_anchor(element, counter)
+            counter += 1
+
+
+def add_table_of_contents(root):
+    sections = []
+
+    for element in root.iterdescendants():
+        if element.tag in ('h2', 'h3'):
+            if element.tag == 'h3' and len(sections) and sections[-1][0] == 'h2':
+                sections[-1][2].append((element.tag, ''.join(element.xpath('descendant-or-self::text()')), []))
+            else:
+                sections.append((element.tag, ''.join(element.xpath('descendant-or-self::text()')), []))
+            
+    toc = etree.Element('div')
+    toc.set('id', 'toc')
+    toc_header = etree.SubElement(toc, 'h2')
+    toc_header.text = u'Spis treÅ›ci'
+    toc_list = etree.SubElement(toc, 'ol')
+
+    for section, text, subsections in sections:
+        section_element = etree.SubElement(toc_list, 'li')
+        section_element.text = text
+        if len(subsections):
+            subsection_list = etree.SubElement(section_element, 'ol')
+            for subsection, text, _ in subsections:
+                subsection_element = etree.SubElement(subsection_list, 'li')
+                subsection_element.text = text
+    root.insert(0, toc)
+