fix(api): Iterate over parts dict in PDF Template

src/api/src/create_book/exceptions.py

@@ -8,4 +8,5 @@ class StoryNotFoundError(WattpadError):
     ...
 
 
-class PartNotFoundError(StoryNotFoundError): ...
+class PartNotFoundError(StoryNotFoundError):
+    ...

src/api/src/create_book/generators/pdf.py

@@ -111,9 +111,9 @@ class PDFGenerator(AbstractGenerator):
                     if not img_data:
                         continue
 
-                    img_tag["src"] = (
+                    img_tag[
-                        f"data:image/jpg;base64,{b64encode(img_data).decode()}"
+                        "src"
-                    )
+                    ] = f"data:image/jpg;base64,{b64encode(img_data).decode()}"
 
             data[part["id"]] = tree.prettify()
 

src/api/src/create_book/generators/pdf/book.html

@@ -46,13 +46,14 @@ id="copyright-license-image">
         <section id="contents" class="toc">
             <h1>Table of Contents</h1>
             <ul>
-                {% for part_id in part_ids %}
+                {% for part_id in parts %}
                     <li><a href="#{{part_id}}"></a></li>
                 {% endfor %}
             </ul>
             </section>
-        {% for part_html in part_htmls %}
+            {% for part_id in parts %}
-            {{part_html | safe}}
+
+            {{parts[part_id] | safe}}
         {% endfor %}
     </div>
 

src/api/src/create_book/parser.py

@@ -15,7 +15,8 @@ def clean_tree(title: str, id: int, body: str) -> BeautifulSoup:
         f"""
     <h1 class="chapter-name" id={id}>{title}</h1>
     <section class="chapter-body"></section>
-"""
+""",
+        parser="html.parser",  # head/body tags aren't generated
     )
 
     insert_at = cast(Tag, new_soup.find("section"))