feat: add OpenDocument thumbnail support

Co-Authored-By: Josh Beatty <[email protected]>
TagStudioDev · Oct 13, 2024 · 91cf3ad · 91cf3ad
1 parent 5b85462
commit 91cf3ad
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/tagstudio/src/core/media_types.py b/tagstudio/src/core/media_types.py
@@ -31,6 +31,7 @@ class MediaType(str, Enum):
     INSTALLER: str = "installer"
     MATERIAL: str = "material"
     MODEL: str = "model"
+    OPEN_DOCUMENT: str = "open_document"
     PACKAGE: str = "package"
     PDF: str = "pdf"
     PLAINTEXT: str = "plaintext"
@@ -214,6 +215,18 @@ class MediaCategories:
     _INSTALLER_SET: set[str] = {".appx", ".msi", ".msix"}
     _MATERIAL_SET: set[str] = {".mtl"}
     _MODEL_SET: set[str] = {".3ds", ".fbx", ".obj", ".stl"}
+    _OPEN_DOCUMENT_SET: set[str] = {
+        ".fodg",
+        ".fodp",
+        ".fods",
+        ".fodt",
+        ".mscz",
+        ".odf",
+        ".odg",
+        ".odp",
+        ".ods",
+        ".odt",
+    }
     _PACKAGE_SET: set[str] = {
         ".aab",
         ".akp",
@@ -387,6 +400,11 @@ class MediaCategories:
         extensions=_MODEL_SET,
         is_iana=True,
     )
+    OPEN_DOCUMENT_TYPES: MediaCategory = MediaCategory(
+        media_type=MediaType.OPEN_DOCUMENT,
+        extensions=_OPEN_DOCUMENT_SET,
+        is_iana=False,
+    )
     PACKAGE_TYPES: MediaCategory = MediaCategory(
         media_type=MediaType.PACKAGE,
         extensions=_PACKAGE_SET,
@@ -456,6 +474,7 @@ class MediaCategories:
         INSTALLER_TYPES,
         MATERIAL_TYPES,
         MODEL_TYPES,
+        OPEN_DOCUMENT_TYPES,
         PACKAGE_TYPES,
         PDF_TYPES,
         PLAINTEXT_TYPES,

diff --git a/tagstudio/src/qt/widgets/thumb_renderer.py b/tagstudio/src/qt/widgets/thumb_renderer.py
@@ -5,6 +5,7 @@
 
 import math
 import struct
+import zipfile
 from copy import deepcopy
 from io import BytesIO
 from pathlib import Path
@@ -603,6 +604,27 @@ def _source_engine(self, filepath: Path) -> Image.Image:
                 logger.error("Couldn't render thumbnail", filepath=filepath, error=e)
         return im
 
+    def _open_doc_thumb(self, filepath: Path) -> Image.Image:
+        """Extract and render a thumbnail for an OpenDocument file.
+
+        Args:
+            filepath (Path): The path of the file.
+        """
+        file_path_within_zip = "Thumbnails/thumbnail.png"
+        im: Image.Image = None
+        with zipfile.ZipFile(filepath, "r") as zip_file:
+            # Check if the file exists in the zip
+            if file_path_within_zip in zip_file.namelist():
+                # Read the specific file into memory
+                file_data = zip_file.read(file_path_within_zip)
+                thumb_im = Image.open(BytesIO(file_data))
+                if thumb_im:
+                    im = Image.new("RGB", thumb_im.size, color="#1e1e1e")
+                    im.paste(thumb_im)
+            else:
+                logger.error("Couldn't render thumbnail", filepath=filepath)
+        return im
+
     def _font_short_thumb(self, filepath: Path, size: int) -> Image.Image:
         """Render a small font preview ("Aa") thumbnail from a font file.
 
@@ -936,6 +958,11 @@ def render(
                     ext, MediaCategories.VIDEO_TYPES, mime_fallback=True
                 ):
                     image = self._video_thumb(_filepath)
+                # OpenDocument/OpenOffice ======================================
+                elif MediaCategories.is_ext_in_category(
+                    ext, MediaCategories.OPEN_DOCUMENT_TYPES, mime_fallback=True
+                ):
+                    image = self._open_doc_thumb(_filepath)
                 # Plain Text ===================================================
                 if MediaCategories.is_ext_in_category(
                     ext, MediaCategories.PLAINTEXT_TYPES, mime_fallback=True