refactor: add support for Pydantic models in table data handling

Author: ya7010

pyproject.toml

@@ -27,6 +27,7 @@ dependencies = [
 [project.optional-dependencies]
 pandas = ["pandas>=2.0.0"]
 polars = ["polars>=0.20.0"]
+pydantic = ["pydantic>=2.0.0"]
 tool = [
   "rich>=14.0.0",
   "rich-argparse>=1.7.0",
@@ -36,6 +37,7 @@ tool = [
 dataframes = [
   "pandas",
   "polars",
+  "pydantic",
 ]
 dev = [
   "mypy>=1.15.0",

src/tppt/_features.py

@@ -15,6 +15,17 @@ class Dataclass(Protocol):
     __dataclass_fields__: ClassVar[dict]
 
 
+try:
+    import pydantic  # type: ignore[import]  # noqa: F401
+
+    USE_PYDANTIC = True
+    PydanticModel: TypeAlias = pydantic.BaseModel  # type: ignore
+
+except ImportError:
+    USE_PYDANTIC = False
+    PydanticModel: TypeAlias = _NotSupportFeature  # type: ignore
+
+
 try:
     import pandas  # type: ignore[import]  # noqa: F401
 

src/tppt/pptx/shape/table.py

@@ -22,6 +22,7 @@
     PandasDataFrame,
     PolarsDataFrame,
     PolarsLazyFrame,
+    PydanticModel,
 )
 from tppt.types._length import LiteralPoint, Points
 
@@ -35,6 +36,7 @@
 DataFrame: TypeAlias = (
     list[list[str]]
     | list[Dataclass]
+    | list[PydanticModel]
     | PandasDataFrame
     | PolarsDataFrame
     | PolarsLazyFrame
@@ -198,6 +200,15 @@ def dataframe2list(data: DataFrame) -> list[list[str]]:
                     ]
                     rows.append(row)
                 return [columns] + rows
+            # Convert list of Pydantic model instances to list of lists
+            elif isinstance(first_instance, PydanticModel):
+                columns = list(first_instance.__class__.model_fields.keys())
+                rows = []
+                for instance in data:
+                    instance = cast(Any, instance)
+                    row = [str(getattr(instance, field)) for field in columns]
+                    rows.append(row)
+                return [columns] + rows
         else:
             logger.warning("Empty data of table")
             return []

tests/test_table.py

@@ -4,9 +4,10 @@
 from dataclasses import dataclass
 
 import pytest
+from pydantic import BaseModel
 
 import tppt
-from tppt._features import USE_PANDAS, USE_POLARS, Dataclass
+from tppt._features import USE_PANDAS, USE_POLARS, Dataclass, PydanticModel
 
 
 def test_create_table_with_list_data(output: pathlib.Path) -> None:
@@ -182,3 +183,40 @@ class Employee:
         .build()
     )
     presentation.save(output / "table_dataclass_data.pptx")
+
+
+def test_create_table_with_pydantic_model(output: pathlib.Path) -> None:
+    """Test creating a table with Pydantic model."""
+
+    class Product(BaseModel):
+        """Product model for testing."""
+
+        name: str
+        category: str
+        price: int
+        stock: int
+
+    # Create product instances
+    products: list[PydanticModel] = [
+        Product(name="ノートPC", category="電子機器", price=150000, stock=10),
+        Product(name="スマートフォン", category="電子機器", price=80000, stock=20),
+        Product(name="タブレット", category="電子機器", price=50000, stock=15),
+    ]
+
+    presentation = (
+        tppt.Presentation.builder()
+        .slide(
+            lambda slide: slide.BlankLayout()
+            .builder()
+            .table(
+                products,
+                left=(100, "pt"),
+                top=(100, "pt"),
+                width=(400, "pt"),
+                height=(200, "pt"),
+                first_row_header=True,
+            )
+        )
+        .build()
+    )
+    presentation.save(output / "table_pydantic_data.pptx")