Skip to content

Commit 0358f51

Browse files
Roc25gred
andauthored
feat: add IMAGE_RASTER_TYPES (Fix #550) (#551)
* fix resolution info * Fix for Raw and Vector Image types * Small refactor * Create IMAGE_RASTER_TYPES and remove is_image_ext_raster * Change if statment only for raster * Rename _IMAGE_SET to _IMAGE_RASTER_SET --------- Co-authored-by: gred <[email protected]>
1 parent e02eb39 commit 0358f51

File tree

2 files changed

+9
-10
lines changed

2 files changed

+9
-10
lines changed

tagstudio/src/core/media_types.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -207,7 +207,7 @@ class MediaCategories:
207207
".rw2",
208208
}
209209
_IMAGE_VECTOR_SET: set[str] = {".svg"}
210-
_IMAGE_SET: set[str] = {
210+
_IMAGE_RASTER_SET: set[str] = {
211211
".apng",
212212
".avif",
213213
".bmp",
@@ -392,9 +392,14 @@ class MediaCategories:
392392
extensions=_IMAGE_VECTOR_SET,
393393
is_iana=False,
394394
)
395+
IMAGE_RASTER_TYPES: MediaCategory = MediaCategory(
396+
media_type=MediaType.IMAGE,
397+
extensions=_IMAGE_RASTER_SET,
398+
is_iana=False,
399+
)
395400
IMAGE_TYPES: MediaCategory = MediaCategory(
396401
media_type=MediaType.IMAGE,
397-
extensions=_IMAGE_SET | _IMAGE_RAW_SET | _IMAGE_VECTOR_SET,
402+
extensions=_IMAGE_RASTER_SET | _IMAGE_RAW_SET | _IMAGE_VECTOR_SET,
398403
is_iana=True,
399404
)
400405
INSTALLER_TYPES: MediaCategory = MediaCategory(

tagstudio/src/qt/widgets/preview_panel.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -625,13 +625,7 @@ def update_widgets(self) -> bool:
625625
self.preview_gif.show()
626626

627627
image = None
628-
if (
629-
MediaCategories.is_ext_in_category(ext, MediaCategories.IMAGE_TYPES)
630-
and MediaCategories.is_ext_in_category(ext, MediaCategories.IMAGE_RAW_TYPES)
631-
and MediaCategories.is_ext_in_category(
632-
ext, MediaCategories.IMAGE_VECTOR_TYPES
633-
)
634-
):
628+
if MediaCategories.is_ext_in_category(ext, MediaCategories.IMAGE_RASTER_TYPES):
635629
image = Image.open(str(filepath))
636630
elif MediaCategories.is_ext_in_category(ext, MediaCategories.IMAGE_RAW_TYPES):
637631
try:
@@ -668,7 +662,7 @@ def update_widgets(self) -> bool:
668662
# Stats for specific file types are displayed here.
669663
if image and (
670664
MediaCategories.is_ext_in_category(
671-
ext, MediaCategories.IMAGE_TYPES, mime_fallback=True
665+
ext, MediaCategories.IMAGE_RASTER_TYPES, mime_fallback=True
672666
)
673667
or MediaCategories.is_ext_in_category(
674668
ext, MediaCategories.VIDEO_TYPES, mime_fallback=True

0 commit comments

Comments
 (0)