feat: add CustomSlideMaster and CustomTitleSlide for enhanced slide customization

Author: ya7010

examples/custom_slide_master.py

@@ -0,0 +1,42 @@
+from pathlib import Path
+
+import tppt
+from tppt._tppt.slide_layout import DefaultMasterSlide, DefaultTitleSlide
+
+
+def main():
+    class CustomMasterSlide(DefaultMasterSlide):
+        pass
+
+    class CustomTitleSlide(DefaultTitleSlide):
+        pass
+
+    class CustomSlideMaster(tppt.TpptSlideMaster):
+        Master = CustomMasterSlide
+        Title = CustomTitleSlide
+
+    presentation = (
+        tppt.Presentation.builder(CustomSlideMaster)
+        .slide(
+            lambda layout: layout.Master(
+                title="Custom Master Title",
+                text="Custom Master Text",
+            )
+            .builder()
+            .text(
+                "a",
+                left=(100, "pt"),
+                top=(100, "pt"),
+                width=(100, "pt"),
+                height=(100, "pt"),
+            )
+        )
+        .slide(lambda layout: layout.Title(title="Custom Title"))
+        .build()
+    )
+
+    presentation.save(Path(__file__).with_suffix(".pptx"))
+
+
+if __name__ == "__main__":
+    main()

examples/presentation_tree.py

@@ -10,7 +10,12 @@ def main() -> None:
     presentation = (
         tppt.Presentation.builder()
         .slide(
-            tppt.SlideBuilder().text(
+            lambda layout: layout.Master(
+                title="Hello, world!",
+                text="Hello, world!",
+            )
+            .builder()
+            .text(
                 "Hello, world!",
                 left=(100, "pt"),
                 top=(100, "pt"),

src/tppt/__init__.py

@@ -7,4 +7,5 @@
 from tppt._pptx.slide import SlideBuilder as SlideBuilder
 from tppt._pptx.slide_layout import SlideLayout as SlideLayout
 from tppt._pptx.slide_master import SlideMaster as SlideMaster
+from tppt._tppt.slide_layout import TpptSlideLayout as TpptSlideLayout
 from tppt._tppt.slide_master import TpptSlideMaster as TpptSlideMaster

src/tppt/_pptx/presentation.py

@@ -1,12 +1,16 @@
 """Presentation wrapper implementation."""
 
 import os
-from typing import IO, Any, Generic, Self
+from typing import IO, Any, Callable, Generic, Self, overload
 
 from pptx.presentation import Presentation as PptxPresentation
 
 from tppt._pptx.tree import ppt2dict
-from tppt._tppt.slide_master import GenericTpptSlideMaster
+from tppt._tppt.slide_layout import TpptSlideLayout
+from tppt._tppt.slide_master import (
+    DefaultSlideMaster,
+    GenericTpptSlideMaster,
+)
 from tppt.types import FilePath
 
 from .converter import PptxConvertible
@@ -38,11 +42,28 @@ def tree(self) -> dict[str, Any]:
         """Get the node tree of the presentation."""
         return ppt2dict(self._pptx)
 
+    @overload
     @classmethod
     def builder(
-        cls, slide_master: "GenericTpptSlideMaster | None" = None
-    ) -> "PresentationBuilder[GenericTpptSlideMaster]":
+        cls,
+    ) -> "PresentationBuilder[DefaultSlideMaster]": ...
+
+    @overload
+    @classmethod
+    def builder(
+        cls,
+        slide_master: "type[GenericTpptSlideMaster]",
+    ) -> "PresentationBuilder[GenericTpptSlideMaster]": ...
+
+    @classmethod
+    def builder(
+        cls,
+        slide_master=None,
+    ):
         """Get a builder for the presentation."""
+
+        if slide_master is None:
+            slide_master = DefaultSlideMaster
         return PresentationBuilder(slide_master)
 
     def save(self, file: FilePath | IO[bytes]) -> None:
@@ -64,17 +85,33 @@ def from_pptx(cls, pptx_obj: PptxPresentation) -> Self:
 class PresentationBuilder(Generic[GenericTpptSlideMaster]):
     """Builder for presentations."""
 
-    def __init__(self, slide_master: "GenericTpptSlideMaster | None" = None) -> None:
+    def __init__(
+        self,
+        slide_master: "type[GenericTpptSlideMaster]",
+    ) -> None:
         """Initialize the builder."""
         import pptx
 
         self._pptx = pptx.Presentation()
         self._slide_master = slide_master
 
-    def slide(self, slide: SlideBuilder, /) -> Self:
+    def slide(
+        self,
+        slide: SlideBuilder
+        | Callable[[type[GenericTpptSlideMaster]], SlideBuilder | TpptSlideLayout],
+        /,
+    ) -> Self:
         """Add a slide to the presentation."""
-
-        slide._build(self)
+        if isinstance(slide, SlideBuilder):
+            slide._build(self)
+        else:
+            slide_layout = slide(self._slide_master)
+            slide_builder = (
+                slide_layout.builder()
+                if isinstance(slide_layout, TpptSlideLayout)
+                else slide_layout
+            )
+            slide_builder._build(self)
 
         return self
 

src/tppt/_tppt/slide_layout.py

@@ -1,5 +1,8 @@
 import datetime
-from typing import Any, Self, overload
+from typing import TYPE_CHECKING, Any, Self, overload
+
+if TYPE_CHECKING:
+    from tppt._pptx.slide import SlideBuilder
 
 
 class TpptSlideLayout:
@@ -18,6 +21,11 @@ def __get__(self, instance: object | None, objtype: type[Any]) -> type[Self] | S
         else:
             return self
 
+    def builder(self) -> "SlideBuilder":
+        from tppt._pptx.slide import SlideBuilder
+
+        return SlideBuilder()
+
 
 class DefaultMasterSlide(TpptSlideLayout):
     """Default master slide layout."""

src/tppt/_tppt/slide_master.py

@@ -60,5 +60,4 @@ class DefaultSlideMaster(TpptSlideMaster):
 GenericTpptSlideMaster = TypeVar(
     "GenericTpptSlideMaster",
     bound=TpptSlideMaster,
-    default=DefaultSlideMaster,
 )