diff --git a/renamer/extractor.py b/renamer/extractor.py index f5643da..7a2f204 100644 --- a/renamer/extractor.py +++ b/renamer/extractor.py @@ -64,9 +64,6 @@ class MediaExtractor: 'extension': [ ('FileInfo', lambda: self.fileinfo_extractor.extract_extension()) ], - 'tracks': [ - ('MediaInfo', lambda: self.mediainfo_extractor.extract_tracks()) - ] } # Conditions for when a value is considered valid diff --git a/renamer/extractors/mediainfo_extractor.py b/renamer/extractors/mediainfo_extractor.py index 3d066be..5125906 100644 --- a/renamer/extractors/mediainfo_extractor.py +++ b/renamer/extractors/mediainfo_extractor.py @@ -111,7 +111,7 @@ class MediaInfoExtractor: def extract_audio_tracks(self) -> list[dict]: """Extract audio track data""" tracks = [] - for a in self.audio_tracks[:3]: # Up to 3 audios + for a in self.audio_tracks[:10]: # Up to 10 audios track_data = { 'codec': getattr(a, 'format', None) or getattr(a, 'codec', None) or 'unknown', 'channels': getattr(a, 'channel_s', None), @@ -124,18 +124,10 @@ class MediaInfoExtractor: def extract_subtitle_tracks(self) -> list[dict]: """Extract subtitle track data""" tracks = [] - for s in self.sub_tracks[:3]: # Up to 3 subs + for s in self.sub_tracks[:10]: # Up to 10 subs track_data = { 'language': getattr(s, 'language', None) or 'und', 'format': getattr(s, 'format', None) or getattr(s, 'codec', None) or 'unknown', } tracks.append(track_data) - return tracks - - def extract_tracks(self) -> dict: - """Extract media track information as data""" - return { - 'video_tracks': self.extract_video_tracks(), - 'audio_tracks': self.extract_audio_tracks(), - 'subtitle_tracks': self.extract_subtitle_tracks(), - } \ No newline at end of file + return tracks \ No newline at end of file