feat: Increase audio and subtitle track extraction limits to 10
This commit is contained in:
@@ -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
|
||||||
|
|||||||
@@ -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(),
|
|
||||||
}
|
|
||||||
Reference in New Issue
Block a user