epub-generator 0.1.2__tar.gz → 0.1.4__tar.gz

{epub_generator-0.1.2 → epub_generator-0.1.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: epub-generator
-Version: 0.1.2
+Version: 0.1.4
 Summary: A simple Python EPUB 3.0 generator with a single API call
 License: MIT
 Keywords: epub,epub3,ebook,generator,publishing
@@ -48,7 +48,7 @@ pip install epub-generator
 ### Generate Your First Book in 5 Minutes
 
 ```python
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 # Prepare book data
 epub_data = EpubData(
@@ -61,9 +61,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Chapter 1"]),
-                    Text(kind=TextKind.BODY, content=["This is the first paragraph."]),
-                    Text(kind=TextKind.BODY, content=["This is the second paragraph."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Chapter 1"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the first paragraph."]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the second paragraph."]),
                 ]
             ),
         ),
@@ -80,7 +80,7 @@ That's it! You now have a valid EPUB 3.0 ebook file.
 
 - **Minimal API**: Just one function call `generate_epub()`
 - **EPUB 3.0**: Generates standards-compliant EPUB 3.0 format
-- **Rich Content**: Supports text, images, tables, math formulas (block-level and inline), footnotes
+- **Rich Content**: Supports text, images, tables, math formulas (block-level and inline), footnotes, custom HTML tags
 - **Flexible Structure**: Nested chapters, prefaces, cover images
 - **Math Support**: LaTeX to MathML/SVG conversion with inline formula support
 - **Type Safe**: Full type annotations included
@@ -92,7 +92,7 @@ That's it! You now have a valid EPUB 3.0 ebook file.
 ```python
 from datetime import datetime, timezone
 from pathlib import Path
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     meta=BookMeta(
@@ -111,8 +111,8 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Chapter 1"]),
-                    Text(kind=TextKind.BODY, content=["Main content..."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Chapter 1"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Main content..."]),
                 ]
             ),
         ),
@@ -125,7 +125,7 @@ generate_epub(epub_data, "book_with_cover.epub")
 ### Nested Chapter Structure
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     chapters=[
@@ -136,7 +136,7 @@ epub_data = EpubData(
                     title="Chapter 1.1",
                     get_chapter=lambda: Chapter(
                         elements=[
-                            Text(kind=TextKind.BODY, content=["Content 1.1..."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["Content 1.1..."]),
                         ]
                     ),
                 ),
@@ -144,7 +144,7 @@ epub_data = EpubData(
                     title="Chapter 1.2",
                     get_chapter=lambda: Chapter(
                         elements=[
-                            Text(kind=TextKind.BODY, content=["Content 1.2..."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["Content 1.2..."]),
                         ]
                     ),
                 ),
@@ -160,7 +160,7 @@ generate_epub(epub_data, "book_with_nested_chapters.epub")
 
 ```python
 from pathlib import Path
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Image
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Image
 
 epub_data = EpubData(
     chapters=[
@@ -168,7 +168,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Here's an image:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Here's an image:"]),
                     Image(
                         path=Path("image.png"),  # Image path
                         alt_text="Image description",
@@ -185,7 +185,7 @@ generate_epub(epub_data, "book_with_images.epub")
 ### Add Footnotes
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Mark, Footnote
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Mark, Footnote
 
 epub_data = EpubData(
     chapters=[
@@ -193,8 +193,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(
+                    TextBlock(
                         kind=TextKind.BODY,
+                        level=0,
                         content=[
                             "This is text with a footnote",
                             Mark(id=1),  # Footnote marker
@@ -206,7 +207,7 @@ epub_data = EpubData(
                     Footnote(
                         id=1,
                         contents=[
-                            Text(kind=TextKind.BODY, content=["This is the footnote content."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["This is the footnote content."]),
                         ],
                     ),
                 ],
@@ -221,7 +222,7 @@ generate_epub(epub_data, "book_with_footnotes.epub")
 ### Add Tables
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Table
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Table
 
 epub_data = EpubData(
     chapters=[
@@ -229,7 +230,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Here's a table:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Here's a table:"]),
                     Table(
                         html_content="""
                         <table>
@@ -253,7 +254,7 @@ generate_epub(epub_data, "book_with_tables.epub")
 Block-level formulas:
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Formula, LaTeXRender
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Formula, LaTeXRender
 
 epub_data = EpubData(
     chapters=[
@@ -261,7 +262,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Pythagorean theorem:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Pythagorean theorem:"]),
                     Formula(latex_expression="x^2 + y^2 = z^2"),  # Block-level formula
                 ]
             ),
@@ -282,8 +283,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(
+                    TextBlock(
                         kind=TextKind.BODY,
+                        level=0,
                         content=[
                             "The Pythagorean theorem ",
                             Formula(latex_expression="a^2 + b^2 = c^2"),  # Inline formula
@@ -301,10 +303,46 @@ epub_data = EpubData(
 generate_epub(epub_data, "book_with_inline_math.epub", latex_render=LaTeXRender.MATHML)
 ```
 
+### Add Custom HTML Tags
+
+You can embed custom HTML tags within text content:
+
+```python
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, HTMLTag
+
+epub_data = EpubData(
+    chapters=[
+        TocItem(
+            title="Chapter 1",
+            get_chapter=lambda: Chapter(elements=[TextBlock(
+                kind=TextKind.BODY,
+                level=0,
+                content=[
+                    "This is normal text with ",
+                    HTMLTag(
+                        name="span",
+                        attributes=[("class", "highlight"), ("style", "color: red;")],
+                        content=["highlighted content"],
+                    ),
+                    " and more text with ",
+                    HTMLTag(
+                        name="strong",
+                        content=["bold text"],
+                    ),
+                    ".",
+                ],
+            )]),
+        ),
+    ],
+)
+
+generate_epub(epub_data, "book_with_html_tags.epub")
+```
+
 ### Add Prefaces
 
 ```python
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     meta=BookMeta(title="Book with Prefaces"),
@@ -313,8 +351,8 @@ epub_data = EpubData(
             title="Preface",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Preface"]),
-                    Text(kind=TextKind.BODY, content=["This is the preface content..."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Preface"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the preface content..."]),
                 ]
             ),
         ),
@@ -324,7 +362,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Main content..."]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Main content..."]),
                 ]
             ),
         ),
@@ -417,12 +455,13 @@ class Chapter:
 
 `ContentBlock` is a union of:
 
-- **`Text`**: Text paragraph
+- **`TextBlock`**: Text paragraph
   ```python
   @dataclass
-  class Text:
-      kind: TextKind                            # BODY | HEADLINE | QUOTE
-      content: list[str | Mark | Formula]       # Text with optional marks and inline formulas
+  class TextBlock:
+      kind: TextKind                                # BODY | HEADLINE | QUOTE
+      level: int                                    # Heading level (0→h1, 1→h2, max h6; only for HEADLINE)
+      content: list[str | Mark | Formula | HTMLTag] # Text with optional marks, inline formulas, and HTML tags
   ```
 
 - **`Image`**: Image reference
@@ -447,6 +486,15 @@ class Chapter:
       latex_expression: str                     # LaTeX expression
   ```
 
+- **`HTMLTag`**: HTML tag
+  ```python
+  @dataclass
+  class HTMLTag:
+      name: str                                 # Tag name (e.g., "span", "div")
+      attributes: list[tuple[str, str]] = []    # List of (attribute, value) pairs
+      content: list[str | Mark | Formula | HTMLTag] = [] # Inner HTML content
+  ```
+
 #### `Footnote`
 
 Footnote/citation.

{epub_generator-0.1.2 → epub_generator-0.1.4}/README.md

@@ -21,7 +21,7 @@ pip install epub-generator
 ### Generate Your First Book in 5 Minutes
 
 ```python
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 # Prepare book data
 epub_data = EpubData(
@@ -34,9 +34,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Chapter 1"]),
-                    Text(kind=TextKind.BODY, content=["This is the first paragraph."]),
-                    Text(kind=TextKind.BODY, content=["This is the second paragraph."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Chapter 1"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the first paragraph."]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the second paragraph."]),
                 ]
             ),
         ),
@@ -53,7 +53,7 @@ That's it! You now have a valid EPUB 3.0 ebook file.
 
 - **Minimal API**: Just one function call `generate_epub()`
 - **EPUB 3.0**: Generates standards-compliant EPUB 3.0 format
-- **Rich Content**: Supports text, images, tables, math formulas (block-level and inline), footnotes
+- **Rich Content**: Supports text, images, tables, math formulas (block-level and inline), footnotes, custom HTML tags
 - **Flexible Structure**: Nested chapters, prefaces, cover images
 - **Math Support**: LaTeX to MathML/SVG conversion with inline formula support
 - **Type Safe**: Full type annotations included
@@ -65,7 +65,7 @@ That's it! You now have a valid EPUB 3.0 ebook file.
 ```python
 from datetime import datetime, timezone
 from pathlib import Path
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     meta=BookMeta(
@@ -84,8 +84,8 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Chapter 1"]),
-                    Text(kind=TextKind.BODY, content=["Main content..."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Chapter 1"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Main content..."]),
                 ]
             ),
         ),
@@ -98,7 +98,7 @@ generate_epub(epub_data, "book_with_cover.epub")
 ### Nested Chapter Structure
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     chapters=[
@@ -109,7 +109,7 @@ epub_data = EpubData(
                     title="Chapter 1.1",
                     get_chapter=lambda: Chapter(
                         elements=[
-                            Text(kind=TextKind.BODY, content=["Content 1.1..."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["Content 1.1..."]),
                         ]
                     ),
                 ),
@@ -117,7 +117,7 @@ epub_data = EpubData(
                     title="Chapter 1.2",
                     get_chapter=lambda: Chapter(
                         elements=[
-                            Text(kind=TextKind.BODY, content=["Content 1.2..."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["Content 1.2..."]),
                         ]
                     ),
                 ),
@@ -133,7 +133,7 @@ generate_epub(epub_data, "book_with_nested_chapters.epub")
 
 ```python
 from pathlib import Path
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Image
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Image
 
 epub_data = EpubData(
     chapters=[
@@ -141,7 +141,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Here's an image:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Here's an image:"]),
                     Image(
                         path=Path("image.png"),  # Image path
                         alt_text="Image description",
@@ -158,7 +158,7 @@ generate_epub(epub_data, "book_with_images.epub")
 ### Add Footnotes
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Mark, Footnote
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Mark, Footnote
 
 epub_data = EpubData(
     chapters=[
@@ -166,8 +166,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(
+                    TextBlock(
                         kind=TextKind.BODY,
+                        level=0,
                         content=[
                             "This is text with a footnote",
                             Mark(id=1),  # Footnote marker
@@ -179,7 +180,7 @@ epub_data = EpubData(
                     Footnote(
                         id=1,
                         contents=[
-                            Text(kind=TextKind.BODY, content=["This is the footnote content."]),
+                            TextBlock(kind=TextKind.BODY, level=0, content=["This is the footnote content."]),
                         ],
                     ),
                 ],
@@ -194,7 +195,7 @@ generate_epub(epub_data, "book_with_footnotes.epub")
 ### Add Tables
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Table
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Table
 
 epub_data = EpubData(
     chapters=[
@@ -202,7 +203,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Here's a table:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Here's a table:"]),
                     Table(
                         html_content="""
                         <table>
@@ -226,7 +227,7 @@ generate_epub(epub_data, "book_with_tables.epub")
 Block-level formulas:
 
 ```python
-from epub_generator import generate_epub, EpubData, TocItem, Chapter, Text, TextKind, Formula, LaTeXRender
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, Formula, LaTeXRender
 
 epub_data = EpubData(
     chapters=[
@@ -234,7 +235,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Pythagorean theorem:"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Pythagorean theorem:"]),
                     Formula(latex_expression="x^2 + y^2 = z^2"),  # Block-level formula
                 ]
             ),
@@ -255,8 +256,9 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(
+                    TextBlock(
                         kind=TextKind.BODY,
+                        level=0,
                         content=[
                             "The Pythagorean theorem ",
                             Formula(latex_expression="a^2 + b^2 = c^2"),  # Inline formula
@@ -274,10 +276,46 @@ epub_data = EpubData(
 generate_epub(epub_data, "book_with_inline_math.epub", latex_render=LaTeXRender.MATHML)
 ```
 
+### Add Custom HTML Tags
+
+You can embed custom HTML tags within text content:
+
+```python
+from epub_generator import generate_epub, EpubData, TocItem, Chapter, TextBlock, TextKind, HTMLTag
+
+epub_data = EpubData(
+    chapters=[
+        TocItem(
+            title="Chapter 1",
+            get_chapter=lambda: Chapter(elements=[TextBlock(
+                kind=TextKind.BODY,
+                level=0,
+                content=[
+                    "This is normal text with ",
+                    HTMLTag(
+                        name="span",
+                        attributes=[("class", "highlight"), ("style", "color: red;")],
+                        content=["highlighted content"],
+                    ),
+                    " and more text with ",
+                    HTMLTag(
+                        name="strong",
+                        content=["bold text"],
+                    ),
+                    ".",
+                ],
+            )]),
+        ),
+    ],
+)
+
+generate_epub(epub_data, "book_with_html_tags.epub")
+```
+
 ### Add Prefaces
 
 ```python
-from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, Text, TextKind
+from epub_generator import generate_epub, EpubData, BookMeta, TocItem, Chapter, TextBlock, TextKind
 
 epub_data = EpubData(
     meta=BookMeta(title="Book with Prefaces"),
@@ -286,8 +324,8 @@ epub_data = EpubData(
             title="Preface",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.HEADLINE, content=["Preface"]),
-                    Text(kind=TextKind.BODY, content=["This is the preface content..."]),
+                    TextBlock(kind=TextKind.HEADLINE, level=0, content=["Preface"]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["This is the preface content..."]),
                 ]
             ),
         ),
@@ -297,7 +335,7 @@ epub_data = EpubData(
             title="Chapter 1",
             get_chapter=lambda: Chapter(
                 elements=[
-                    Text(kind=TextKind.BODY, content=["Main content..."]),
+                    TextBlock(kind=TextKind.BODY, level=0, content=["Main content..."]),
                 ]
             ),
         ),
@@ -390,12 +428,13 @@ class Chapter:
 
 `ContentBlock` is a union of:
 
-- **`Text`**: Text paragraph
+- **`TextBlock`**: Text paragraph
   ```python
   @dataclass
-  class Text:
-      kind: TextKind                            # BODY | HEADLINE | QUOTE
-      content: list[str | Mark | Formula]       # Text with optional marks and inline formulas
+  class TextBlock:
+      kind: TextKind                                # BODY | HEADLINE | QUOTE
+      level: int                                    # Heading level (0→h1, 1→h2, max h6; only for HEADLINE)
+      content: list[str | Mark | Formula | HTMLTag] # Text with optional marks, inline formulas, and HTML tags
   ```
 
 - **`Image`**: Image reference
@@ -420,6 +459,15 @@ class Chapter:
       latex_expression: str                     # LaTeX expression
   ```
 
+- **`HTMLTag`**: HTML tag
+  ```python
+  @dataclass
+  class HTMLTag:
+      name: str                                 # Tag name (e.g., "span", "div")
+      attributes: list[tuple[str, str]] = []    # List of (attribute, value) pairs
+      content: list[str | Mark | Formula | HTMLTag] = [] # Inner HTML content
+  ```
+
 #### `Footnote`
 
 Footnote/citation.

{epub_generator-0.1.2 → epub_generator-0.1.4}/epub_generator/__init__.py

@@ -8,10 +8,11 @@ from .types import (
     EpubData,
     Footnote,
     Formula,
+    HTMLTag,
     Image,
     Mark,
     Table,
-    Text,
+    TextBlock,
     TextKind,
     TocItem,
 )
@@ -29,10 +30,11 @@ __all__ = [
     "Chapter",
     "ChapterGetter",
     "ContentBlock",
-    "Text",
+    "TextBlock",
     "TextKind",
     "Table",
     "Formula",
+    "HTMLTag",
     "Image",
     "Footnote",
     "Mark",

{epub_generator-0.1.2 → epub_generator-0.1.4}/epub_generator/generation/gen_chapter.py

@@ -7,15 +7,18 @@ from ..types import (
     Chapter,
     ContentBlock,
     Formula,
+    HTMLTag,
     Image,
     Mark,
     Table,
-    Text,
+    TextBlock,
     TextKind,
 )
 from .gen_asset import process_formula, process_image, process_table
 from .xml_utils import serialize_element, set_epub_type
 
+_MAX_HEADING_LEVEL = 6 # HTML standard defines heading levels from h1 to h6
+
 
 def generate_chapter(
     context: Context,
@@ -88,9 +91,10 @@ def _render_footnotes(
 
 
 def _render_content_block(context: Context, block: ContentBlock) -> Element | None:
-    if isinstance(block, Text):
+    if isinstance(block, TextBlock):
         if block.kind == TextKind.HEADLINE:
-            container = Element("h1")
+            heading_level = min(block.level + 1, _MAX_HEADING_LEVEL)
+            container = Element(f"h{heading_level}")
         elif block.kind == TextKind.QUOTE:
             container = Element("p")
         elif block.kind == TextKind.BODY:
@@ -98,8 +102,11 @@ def _render_content_block(context: Context, block: ContentBlock) -> Element | No
         else:
             raise ValueError(f"Unknown TextKind: {block.kind}")
 
-        _render_text_content(context, container, block.content)
-
+        _render_text_content(
+            context=context,
+            parent=container,
+            content=block.content,
+        )
         if block.kind == TextKind.QUOTE:
             blockquote = Element("blockquote")
             blockquote.append(container)
@@ -120,7 +127,7 @@ def _render_content_block(context: Context, block: ContentBlock) -> Element | No
         return None
 
 
-def _render_text_content(context: Context, parent: Element, content: list[str | Mark | Formula]) -> None:
+def _render_text_content(context: Context, parent: Element, content: list[str | Mark | Formula | HTMLTag]) -> None:
     """Render text content with inline citation marks."""
     current_element = parent
     for item in content:
@@ -136,6 +143,18 @@ def _render_text_content(context: Context, parent: Element, content: list[str |
                 else:
                     current_element.tail += item
 
+        elif isinstance(item, HTMLTag):
+            tag_element = Element(item.name)
+            for attr, value in item.attributes:
+                tag_element.set(attr, value)
+            _render_text_content(
+                context=context,
+                parent=tag_element,
+                content=item.content,
+            )
+            parent.append(tag_element)
+            current_element = tag_element
+
         elif isinstance(item, Formula):
             formula_element = process_formula(
                 context=context,

{epub_generator-0.1.2 → epub_generator-0.1.4}/epub_generator/generation/gen_epub.py

@@ -6,9 +6,10 @@ from uuid import uuid4
 from zipfile import ZipFile
 
 from ..context import Context, Template
+from ..html_tag import search_content
 from ..i18n import I18N
 from ..options import LaTeXRender, TableRender
-from ..types import EpubData, Formula, Text
+from ..types import Chapter, EpubData, Formula, TextBlock
 from .gen_chapter import generate_chapter
 from .gen_nav import gen_nav
 from .gen_toc import NavPoint, gen_toc
@@ -135,21 +136,21 @@ def _write_chapters_from_data(
             assert_not_aborted()
 
 
-def _chapter_has_formula(chapter) -> bool:
+def _chapter_has_formula(chapter: Chapter) -> bool:
     """Check if chapter contains any formulas (block-level or inline)."""
     for element in chapter.elements:
         if isinstance(element, Formula):
             return True
-        if isinstance(element, Text):
-            for item in element.content:
+        if isinstance(element, TextBlock):
+            for item in search_content(element.content):
                 if isinstance(item, Formula):
                     return True
     for footnote in chapter.footnotes:
         for content_block in footnote.contents:
             if isinstance(content_block, Formula):
                 return True
-            if isinstance(content_block, Text):
-                for item in content_block.content:
+            if isinstance(content_block, TextBlock):
+                for item in search_content(content_block.content):
                     if isinstance(item, Formula):
                         return True
     return False

epub_generator-0.1.4/epub_generator/html_tag.py

@@ -0,0 +1,11 @@
+from typing import Generator
+
+from .types import Formula, HTMLTag, Mark
+
+
+def search_content(content: list[str | Mark | Formula | HTMLTag]) -> Generator[str | Mark | Formula, None, None]:
+    for child in content:
+        if isinstance(child, HTMLTag):
+            yield from search_content(child.content)
+        else:
+            yield child

{epub_generator-0.1.2 → epub_generator-0.1.4}/epub_generator/types.py

@@ -107,10 +107,12 @@ class Image:
     """Alt text (defaults to "image")"""
 
 @dataclass
-class Text:
+class TextBlock:
     kind: TextKind
     """Kind of text block."""
-    content: list[str | Mark | Formula]
+    level: int
+    """Heading level starting from 0 (only for HEADLINE: level 0 → h1, level 1 → h2, max h6; ignored for BODY and QUOTE)."""
+    content: list["str | Mark | Formula | HTMLTag"]
     """Text content with optional citation marks."""
 
 @dataclass
@@ -126,7 +128,7 @@ class Footnote:
     """Content blocks"""
 
 
-ContentBlock = Text | Table | Formula | Image
+ContentBlock = TextBlock | Table | Formula | Image
 """Union of all content blocks that appear in main chapter content."""
 
 @dataclass
@@ -138,4 +140,17 @@ class Chapter:
     footnotes: list[Footnote] = field(default_factory=list)
     """Footnotes"""
 
-ChapterGetter = Callable[[], Chapter]
+ChapterGetter = Callable[[], Chapter]
+
+@dataclass
+class HTMLTag:
+    """Generic HTML tag representation."""
+
+    name: str
+    """Tag name"""
+
+    attributes: list[tuple[str, str]] = field(default_factory=list)
+    """List of (attribute, value) pairs"""
+
+    content: list["str | Mark | Formula | HTMLTag"] = field(default_factory=list)
+    """Inner HTML content"""

{epub_generator-0.1.2 → epub_generator-0.1.4}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "epub-generator"
-version = "0.1.2"
+version = "0.1.4"
 description = "A simple Python EPUB 3.0 generator with a single API call"
 authors = ["Tao Zeyu <i@taozeyu.com>"]
 license = "MIT"