feat: Add HDR extraction from filename and update tests

renamer/extractor.py

@@ -38,7 +38,8 @@ class MediaExtractor:
                 ('MediaInfo', lambda: self.mediainfo_extractor.extract_aspect_ratio())
             ],
             'hdr': [
-                ('MediaInfo', lambda: self.mediainfo_extractor.extract_hdr())
+                ('MediaInfo', lambda: self.mediainfo_extractor.extract_hdr()),
+                ('Filename', lambda: self.filename_extractor.extract_hdr())
             ],
             'audio_langs': [
                 ('MediaInfo', lambda: self.mediainfo_extractor.extract_audio_langs())
@@ -46,9 +47,6 @@ class MediaExtractor:
             'meta_type': [
                 ('Metadata', lambda: self.metadata_extractor.extract_meta_type())
             ],
-            'meta_description': [
-                ('Metadata', lambda: self.metadata_extractor.extract_meta_description())
-            ],
             'file_size': [
                 ('FileInfo', lambda: self.fileinfo_extractor.extract_size())
             ],

renamer/extractors/filename_extractor.py

@@ -76,4 +76,16 @@ class FilenameExtractor:
             if re.search(r'\b' + re.escape(indicator) + r'\b', self.file_name, re.IGNORECASE):
                 return 'Unclassified'
         
-        return 'Unclassified'
+        return 'Unclassified'
+
+    def extract_hdr(self) -> str | None:
+        """Extract HDR information from filename"""
+        # Check for SDR first - indicates no HDR
+        if re.search(r'\bSDR\b', self.file_name, re.IGNORECASE):
+            return None
+        
+        # Check for HDR, but not NoHDR
+        if re.search(r'\bHDR\b', self.file_name, re.IGNORECASE) and not re.search(r'\bNoHDR\b', self.file_name, re.IGNORECASE):
+            return 'HDR'
+        
+        return None

renamer/test/filenames.txt

@@ -208,4 +208,5 @@ The long title (2008) SD 720p UKR.avi
 The long title (2008) UHD 1440p ENG.mp4
 The long title (2008) UHD 1440 ENG.mp4
 The long title (2008) 8K 4320p ENG.mp4
-
+Troll 2 (2025) WEB-DL 2160p HDR Ukr Nor [Hurtom].mkv
+Moana 2 (2024) MA WEB-DL 2160p SDR Ukr Eng [Hurtom].mkv

renamer/test/test_filename_extractor.py

@@ -66,4 +66,17 @@ def test_extract_frame_class(filename):
     assert isinstance(frame_class, str)
     # Should be one of the valid frame classes or 'Unclassified'
     valid_classes = set(FRAME_CLASSES.keys()) | {'Unclassified'}
-    assert frame_class in valid_classes
+    assert frame_class in valid_classes
+
+
+@pytest.mark.parametrize("filename", load_test_filenames())
+def test_extract_hdr(filename):
+    """Test HDR extraction from filename"""
+    file_path = Path(filename)
+    extractor = FilenameExtractor(file_path)
+    hdr = extractor.extract_hdr()
+    # Print filename and extracted HDR clearly
+    print(f"\nFilename: \033[1;36m{filename}\033[0m")
+    print(f"Extracted HDR: \033[1;32m{hdr}\033[0m")
+    # HDR should be 'HDR' or None
+    assert hdr is None or hdr == 'HDR'