feat: ePub builder with chapters, images, TOC, cover

Made-with: Cursor

tests/test_epub_builder.py

@@ -0,0 +1,102 @@
+import json
+import os
+from datetime import datetime, date
+from PIL import Image as PILImage
+
+from src.models import Article, Image
+from src.epub_builder import build_epub
+
+
+def _create_test_image(path):
+    os.makedirs(os.path.dirname(path), exist_ok=True)
+    img = PILImage.new("RGB", (800, 450), color="green")
+    img.save(path, format="JPEG")
+
+
+def test_build_epub_creates_file(app, db, tmp_path):
+    with app.app_context():
+        img_path = str(tmp_path / "images" / "abc123.jpg")
+        _create_test_image(img_path)
+
+        a1 = Article(
+            guid="g1", title="First Article", author="Author A",
+            pub_date=datetime(2026, 4, 6, 10, 0),
+            categories=json.dumps(["Government"]),
+            link="http://example.com/1",
+            content_html=f'<p>Content one.</p><img src="{img_path}" />',
+        )
+        a2 = Article(
+            guid="g2", title="Second Article", author="Author B",
+            pub_date=datetime(2026, 4, 7, 10, 0),
+            categories=json.dumps(["Culture Calendar"]),
+            link="http://example.com/2",
+            content_html="<p>Content two.</p>",
+        )
+        db.session.add_all([a1, a2])
+        db.session.flush()
+
+        img_record = Image(
+            article_id=a1.id, original_url="https://example.com/photo.jpg",
+            local_path=img_path, width=800, height=450,
+        )
+        db.session.add(img_record)
+        db.session.commit()
+
+        cover_img = PILImage.new("RGB", (800, 480), color="white")
+        cover_path = str(tmp_path / "cover.jpg")
+        cover_img.save(cover_path, format="JPEG")
+
+        output_dir = str(tmp_path / "issues")
+        epub_path = build_epub(
+            week_start=date(2026, 4, 6),
+            week_end=date(2026, 4, 12),
+            article_ids=[a1.id, a2.id],
+            cover_path=cover_path,
+            output_dir=output_dir,
+        )
+
+        assert os.path.exists(epub_path)
+        assert epub_path.endswith(".epub")
+        assert os.path.getsize(epub_path) > 0
+
+
+def test_build_epub_respects_article_order(app, db, tmp_path):
+    with app.app_context():
+        a1 = Article(
+            guid="g1", title="Later Article", author="A",
+            pub_date=datetime(2026, 4, 8, 10, 0),
+            categories="[]", link="http://a", content_html="<p>Later</p>",
+        )
+        a2 = Article(
+            guid="g2", title="Earlier Article", author="B",
+            pub_date=datetime(2026, 4, 6, 10, 0),
+            categories="[]", link="http://b", content_html="<p>Earlier</p>",
+        )
+        db.session.add_all([a1, a2])
+        db.session.commit()
+
+        cover_path = str(tmp_path / "cover.jpg")
+        PILImage.new("RGB", (800, 480)).save(cover_path, format="JPEG")
+
+        epub_path = build_epub(
+            week_start=date(2026, 4, 6),
+            week_end=date(2026, 4, 12),
+            article_ids=[a1.id, a2.id],
+            cover_path=cover_path,
+            output_dir=str(tmp_path / "issues"),
+        )
+
+        from ebooklib import epub as epublib
+        book = epublib.read_epub(epub_path)
+        spine_items = [book.get_item_with_id(item_id)
+                       for item_id, _ in book.spine if item_id != "nav"]
+        titles = []
+        for item in spine_items:
+            if item and b"<h1>" in item.get_content():
+                content = item.get_content().decode("utf-8")
+                start = content.index("<h1>") + 4
+                end = content.index("</h1>")
+                titles.append(content[start:end])
+
+        assert titles[0] == "Earlier Article"
+        assert titles[1] == "Later Article"