Small script to extract subtitles from an mkv/s container

mkse.py

"""
Usage: mkse.py [path_to_container]
Requirements: pymediainfo module, mkvextract in system path
"""

import subprocess
import sys
import re
from pathlib import Path
from pymediainfo import MediaInfo

file = Path(sys.argv[1])
tracks = MediaInfo.parse(file).text_tracks

codec_map = {'UTF8': 'srt', 'PGS': 'sup'}
filter_ = r'[\/\\*?|":><]'

print('\nExtracting...')
for i, track in enumerate(tracks):
    ext = track.codec_id.split('/')[1]
    subprocess.run(['mkvextract', file, 'tracks', 
                    f'''{i}:{track.language}{f"_{re.sub(filter_, '_', track.title)}" 
                    if track.title else ""}.{codec_map.get(ext, ext.lower())}'''],
                    stdout=subprocess.PIPE, stderr=subprocess.PIPE)
print('\nDone')