refactor: move type constants to new media classes

CarterPillow · Jul 24, 2024 · 53b2db9 · 53b2db9
1 parent e463635
commit 53b2db9
Show file tree

Hide file tree

Showing 6 changed files with 523 additions and 231 deletions.
diff --git a/tagstudio/src/core/constants.py b/tagstudio/src/core/constants.py
@@ -12,156 +12,6 @@
 )
 FONT_SAMPLE_SIZES: list[int] = [10, 15, 20]
 
-# TODO: Turn this whitelist into a user-configurable blacklist.
-IMAGE_TYPES: list[str] = [
-    ".png",
-    ".jpg",
-    ".jpeg",
-    ".jpg_large",
-    ".jpeg_large",
-    ".jfif",
-    ".gif",
-    ".tif",
-    ".tiff",
-    ".heic",
-    ".heif",
-    ".webp",
-    ".bmp",
-    ".svg",
-    ".avif",
-    ".apng",
-    ".jp2",
-    ".j2k",
-    ".jpg2",
-]
-RAW_IMAGE_TYPES: list[str] = [
-    ".raw",
-    ".dng",
-    ".rw2",
-    ".nef",
-    ".arw",
-    ".crw",
-    ".cr2",
-    ".cr3",
-]
-VIDEO_TYPES: list[str] = [
-    ".mp4",
-    ".webm",
-    ".mov",
-    ".hevc",
-    ".mkv",
-    ".avi",
-    ".wmv",
-    ".flv",
-    ".gifv",
-    ".m4p",
-    ".m4v",
-    ".3gp",
-]
-AUDIO_TYPES: list[str] = [
-    ".mp3",
-    ".mp4",
-    ".mpeg4",
-    ".m4a",
-    ".aac",
-    ".wav",
-    ".flac",
-    ".alac",
-    ".wma",
-    ".ogg",
-    ".aiff",
-]
-DOC_TYPES: list[str] = [
-    ".txt",
-    ".rtf",
-    ".md",
-    ".doc",
-    ".docx",
-    ".pdf",
-    ".tex",
-    ".odt",
-    ".pages",
-]
-PLAINTEXT_TYPES: list[str] = [
-    ".txt",
-    ".md",
-    ".css",
-    ".html",
-    ".xml",
-    ".json",
-    ".js",
-    ".ts",
-    ".ini",
-    ".htm",
-    ".csv",
-    ".php",
-    ".sh",
-    ".bat",
-]
-SPREADSHEET_TYPES: list[str] = [".csv", ".xls", ".xlsx", ".numbers", ".ods"]
-PRESENTATION_TYPES: list[str] = [".ppt", ".pptx", ".key", ".odp"]
-ARCHIVE_TYPES: list[str] = [
-    ".zip",
-    ".rar",
-    ".tar",
-    ".tar",
-    ".gz",
-    ".tgz",
-    ".7z",
-    ".s7z",
-]
-BLENDER_TYPES: list[str] = [
-    ".blend",
-    ".blend1",
-    ".blend2",
-    ".blend3",
-    ".blend4",
-    ".blend5",
-    ".blend6",
-    ".blend7",
-    ".blend8",
-    ".blend9",
-    ".blend10",
-    ".blend11",
-    ".blend12",
-    ".blend13",
-    ".blend14",
-    ".blend15",
-    ".blend16",
-    ".blend17",
-    ".blend18",
-    ".blend19",
-    ".blend20",
-    ".blend21",
-    ".blend22",
-    ".blend23",
-    ".blend24",
-    ".blend25",
-    ".blend26",
-    ".blend27",
-    ".blend28",
-    ".blend29",
-    ".blend30",
-    ".blend31",
-    ".blend32",
-]
-PROGRAM_TYPES: list[str] = [".exe", ".app"]
-SHORTCUT_TYPES: list[str] = [".lnk", ".desktop", ".url"]
-FONT_TYPES: list[str] = [".ttf", ".otf", ".woff", ".woff2", ".ttc"]
-
-ALL_FILE_TYPES: list[str] = (
-    IMAGE_TYPES
-    + VIDEO_TYPES
-    + AUDIO_TYPES
-    + DOC_TYPES
-    + SPREADSHEET_TYPES
-    + PRESENTATION_TYPES
-    + ARCHIVE_TYPES
-    + PROGRAM_TYPES
-    + SHORTCUT_TYPES
-    + FONT_TYPES
-)
-
 BOX_FIELDS = ["tag_box", "text_box"]
 TEXT_FIELDS = ["text_line", "text_box"]
 DATE_FIELDS = ["datetime"]