feat: Increase audio and subtitle track extraction limits to 10

This commit is contained in:
sHa
2025-12-26 12:07:33 +00:00
parent 6d377c9871
commit 8f68624529
2 changed files with 3 additions and 14 deletions

View File

@@ -64,9 +64,6 @@ class MediaExtractor:
'extension': [ 'extension': [
('FileInfo', lambda: self.fileinfo_extractor.extract_extension()) ('FileInfo', lambda: self.fileinfo_extractor.extract_extension())
], ],
'tracks': [
('MediaInfo', lambda: self.mediainfo_extractor.extract_tracks())
]
} }
# Conditions for when a value is considered valid # Conditions for when a value is considered valid

View File

@@ -111,7 +111,7 @@ class MediaInfoExtractor:
def extract_audio_tracks(self) -> list[dict]: def extract_audio_tracks(self) -> list[dict]:
"""Extract audio track data""" """Extract audio track data"""
tracks = [] 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 = { track_data = {
'codec': getattr(a, 'format', None) or getattr(a, 'codec', None) or 'unknown', 'codec': getattr(a, 'format', None) or getattr(a, 'codec', None) or 'unknown',
'channels': getattr(a, 'channel_s', None), 'channels': getattr(a, 'channel_s', None),
@@ -124,18 +124,10 @@ class MediaInfoExtractor:
def extract_subtitle_tracks(self) -> list[dict]: def extract_subtitle_tracks(self) -> list[dict]:
"""Extract subtitle track data""" """Extract subtitle track data"""
tracks = [] 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 = { track_data = {
'language': getattr(s, 'language', None) or 'und', 'language': getattr(s, 'language', None) or 'und',
'format': getattr(s, 'format', None) or getattr(s, 'codec', None) or 'unknown', 'format': getattr(s, 'format', None) or getattr(s, 'codec', None) or 'unknown',
} }
tracks.append(track_data) tracks.append(track_data)
return tracks 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(),
}