refactor: move Movie class and related functionality from movie.py to picture.py

Author: ya7010

src/tppt/pptx/shape/movie.py

@@ -1,5 +1,7 @@
-from typing import IO, Literal, Self
+from typing import IO, Literal, NotRequired, Self, assert_never
 
+from pptx.opc.constants import CONTENT_TYPE
+from pptx.shapes.picture import Movie as PptxMovie
 from pptx.shapes.picture import Picture as PptxPicture
 
 from tppt.types import FilePath
@@ -38,3 +40,80 @@ def to_pptx(self) -> PptxPicture:
     def from_pptx(cls, pptx_obj: PptxPicture) -> Self:
         """Create from pptx shape."""
         return cls(pptx_obj)
+
+
+MOVIE_MIME_TYPE = Literal[
+    "video/x-ms-asf",
+    "video/avi",
+    "video/quicktime",
+    "video/mp4",
+    "video/mpeg",
+    "video/msvideo",
+    "video/x-ms-wmv",
+    "video/x-msvideo",
+]
+
+
+def to_pptx_movie_mime_type(mime_type: MOVIE_MIME_TYPE) -> str:
+    """Convert movie mime type to pptx movie mime type."""
+    match mime_type:
+        case "video/x-ms-asf":
+            return CONTENT_TYPE.ASF
+        case "video/avi":
+            return CONTENT_TYPE.AVI
+        case "video/quicktime":
+            return CONTENT_TYPE.MOV
+        case "video/mp4":
+            return CONTENT_TYPE.MP4
+        case "video/mpeg":
+            return CONTENT_TYPE.MPG
+        case "video/msvideo":
+            return CONTENT_TYPE.MS_VIDEO
+        case "video/x-ms-wmv":
+            return CONTENT_TYPE.WMV
+        case "video/x-msvideo":
+            return CONTENT_TYPE.X_MS_VIDEO
+        case _:
+            assert_never(mime_type)
+
+
+class MovieProps(RangeProps):
+    """Movie properties."""
+
+    poster_frame_image: NotRequired[FilePath | IO[bytes]]
+
+    mime_type: MOVIE_MIME_TYPE
+    """
+    Mime type of the movie.
+
+    NOTE: The Movie shape is an experimental feature of python-pptx, and specifying the mime_type is recommended.
+    """
+
+
+class MovieData(MovieProps):
+    """Movie data."""
+
+    type: Literal["movie"]
+
+    movie_file: FilePath | IO[bytes]
+
+
+class Movie(BaseShape[PptxMovie]):
+    """Movie data class."""
+
+    def __init__(
+        self,
+        pptx_obj: PptxMovie,
+        data: MovieData | None = None,
+        /,
+    ) -> None:
+        self._pptx = pptx_obj
+
+    def to_pptx(self) -> PptxMovie:
+        """Convert to pptx shape."""
+        return self._pptx
+
+    @classmethod
+    def from_pptx(cls, pptx_obj: PptxMovie) -> Self:
+        """Create from pptx shape."""
+        return cls(pptx_obj)

src/tppt/pptx/shape/picture.py

@@ -14,7 +14,7 @@
 
 from pptx.slide import Slide as PptxSlide
 
-from tppt.pptx.shape.movie import (
+from tppt.pptx.shape.picture import (
     Movie,
     MovieData,
     MovieProps,