Missing toc items.

[librarian.git] / src / librarian / epub.py

diff --git a/src/librarian/epub.py b/src/librarian/epub.py
index c286903..fc5ee16 100644 (file)
--- a/src/librarian/epub.py
+++ b/src/librarian/epub.py
@@ -327,6 +327,18 @@ def transform_chunk(chunk_xml, chunk_no, annotations, empty=False,
                 )
             )
         elif element.tag in ('naglowek_podrozdzial', 'naglowek_scena'):
+            if not toc:
+                toc.append(
+                    (
+                        epub.Link(
+                            "part%d.xhtml" % chunk_no,
+                            " ",
+                            "part%d" % chunk_no
+                        ),
+                        []
+                    )
+                )
+
             subnumber = len(toc[-1][1])
             toc[-1][1].append(
                 epub.Link(
@@ -641,7 +653,7 @@ def transform(wldoc, verbose=False, style=None,
     output.toc = toc[0][1]
 
     if len(toc) < 2:
-        toc.append(
+        output.toc.append(
             epub.Link(
                 "part1.xhtml",
                 "Początek utworu",
@@ -651,7 +663,7 @@ def transform(wldoc, verbose=False, style=None,
 
     # Last modifications in container files and EPUB creation
     if len(annotations) > 0:
-        toc.append(
+        output.toc.append(
             epub.Link(
                 "annotations.xhtml",
                 "Przypisy",
@@ -675,7 +687,7 @@ def transform(wldoc, verbose=False, style=None,
         output.add_item(item)
         spine.append(item)
 
-    toc.append(
+    output.toc.append(
         epub.Link(
             "support.xhtml",
             "Wesprzyj Wolne Lektury",
@@ -694,7 +706,7 @@ def transform(wldoc, verbose=False, style=None,
     output.add_item(item)
     spine.append(item)
 
-    toc.append(
+    output.toc.append(
         epub.Link(
             "last.xhtml",
             "Strona redakcyjna",