letters

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

diff --git a/src/librarian/builders/epub.py b/src/librarian/builders/epub.py
index 8f21879..1e57583 100644 (file)
--- a/src/librarian/builders/epub.py
+++ b/src/librarian/builders/epub.py
@@ -4,6 +4,7 @@
 from datetime import date
 import io
 import os
+import re
 import tempfile
 from ebooklib import epub
 from lxml import etree
@@ -80,9 +81,12 @@ class EpubBuilder(Builder):
     isbn_field = 'isbn_epub'
     orphans = True
 
-    def __init__(self, *args, **kwargs):
+    def __init__(self, *args, debug=False, **kwargs):
+        self.numbering = 0
         self.chars = set()
         self.fundr = 0
+        self.debug = debug
+        self.splits = []
         super().__init__(*args, **kwargs)
     
     def build(self, document, **kwargs):
@@ -285,7 +289,7 @@ class EpubBuilder(Builder):
                 file_as=str(author),
                 uid='creator{}'.format(i)
             )
-        for translator in self.document.meta.translators:
+        for i, translator in enumerate(self.document.meta.translators):
             self.output.add_author(
                 translator.readable(),
                 file_as=str(translator),
@@ -707,3 +711,8 @@ class EpubBuilder(Builder):
             file_name=name
         )
         return name
+
+    def process_comment(self, comment):
+        m = re.match(r'TRIM:(\d+)', comment.text)
+        if m is not None:
+            self.splits.append(comment.sourceline - int(m.group(1)))