Include language id in output filename.

master
voussoir 2020-02-27 20:52:28 -08:00
parent 41e86526a7
commit a8c78acbdc
1 changed files with 9 additions and 4 deletions

View File

@ -44,10 +44,15 @@ def make_maps(input_file, prefix, search_pattern, extension_map, moveto=None):
if match is None: if match is None:
continue continue
(stream_index, codec) = (match.group(1), match.group(2)) (stream_index, language, codec) = match.groups()
if language is None:
language = ''
else:
language = language.strip('()')
language = '.' + language
extension = extension_map.get(codec, extension_map['*']) extension = extension_map.get(codec, extension_map['*'])
output_filename = input_file.replace_extension('') output_filename = input_file.replace_extension('')
output_filename = output_filename.add_extension(f'{prefix}{stream_index}.{extension}') output_filename = output_filename.add_extension(f'{prefix}{stream_index}{language}.{extension}')
print(f'{stream_index}, codec={codec}, ext=.{extension}') print(f'{stream_index}, codec={codec}, ext=.{extension}')
if moveto: if moveto:
@ -66,7 +71,7 @@ def audio_maps(input_file, moveto=None):
return make_maps( return make_maps(
input_file, input_file,
prefix='a', prefix='a',
search_pattern=r'Stream #0:(\d+)[^\s]*: Audio: (\w+)', search_pattern=r'Stream #0:(\d+)(\(\w+\))?[^\s]*: Audio: (\w+)',
extension_map=AUDIO_EXTENSIONS, extension_map=AUDIO_EXTENSIONS,
moveto=moveto, moveto=moveto,
) )
@ -75,7 +80,7 @@ def subtitle_maps(input_file, moveto=None):
return make_maps( return make_maps(
input_file, input_file,
prefix='s', prefix='s',
search_pattern=r'Stream #0:(\d+)[^\s]*: Subtitle: (\w+)', search_pattern=r'Stream #0:(\d+)(\(\w+\))?[^\s]*: Subtitle: (\w+)',
extension_map=SUBTITLE_EXTENSIONS, extension_map=SUBTITLE_EXTENSIONS,
moveto=moveto, moveto=moveto,
) )