skeh
/
streamaencodebot
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Compare commits

base: skeh:a7520a05b857e61f561706e426fc7475dd665db6
Branches Tags
skeh:master
..
compare: skeh:3d18d786b03380f08f644b1945eeefd580c84730
Branches Tags
skeh:master

No commits in common. "a7520a05b857e61f561706e426fc7475dd665db6" and "3d18d786b03380f08f644b1945eeefd580c84730" have entirely different histories.
a7520a05b8 ... 3d18d786b0

1 changed files with 8 additions and 15 deletions
Show Stats Expand all files Collapse all files

23
main.py
View File

@ -6,13 +6,10 @@ import os
@click.command()
@click.argument('infile')
@click.argument('outdir')
@click.option('--audio', help="Audio language or stream number")
@click.option('--hardcode', help="Hardcode target sub stream (language or stream number)")
@click.option('--overwrite/--no-overwrite', default=True)
def main(infile, outdir, hardcode=None, audio='eng', overwrite=True):
@click.option('--audio-lang')
@click.option('--hardcode')
def main(infile, outdir, hardcode=None, audio_lang='eng'):
outfile = os.path.join(outdir, os.path.splitext(os.path.split(infile)[1])[0]+'.mp4')
if os.path.exists(outfile) and not overwrite:
click.abort(0)
probe = ffmpeg.probe(infile)
audio_streams, sub_streams = [], []
@ -22,20 +19,16 @@ def main(infile, outdir, hardcode=None, audio='eng', overwrite=True):
elif stream['codec_type'] == 'subtitle':
sub_streams.append(stream)
if audio.isdigit():
audio_index = audio
elif len(audio_streams) == 1:
if len(audio_streams) == 1:
audio_index = str(audio_streams[0]['index'])
else:
audio_index = next(str(stream['index']) for stream in audio_streams if stream['tags'].get('language') == audio)
audio_index = next(str(stream['index']) for stream in audio_streams if stream['tags'].get('language') == audio_lang)
if hardcode:
if hardcode.isdigit():
sub_stream = hardcode
elif len(sub_streams) == 1:
if len(sub_streams) == 1:
sub_stream = sub_streams[0]
else:
sub_stream = next(stream for stream in sub_streams if stream['tags'].get('language') == hardcode)
sub = next(stream['index'] for stream in sub_streams if stream['tags'].get('language') == hardcode)
raw = ffmpeg.input(infile)
if hardcode:
@ -47,7 +40,7 @@ def main(infile, outdir, hardcode=None, audio='eng', overwrite=True):
else:
video = raw.video
ffmpeg.output(video, raw[audio_index], outfile, acodec='aac', vcodec='libx264', crf='18', preset='slow').run(overwrite_output=True)
ffmpeg.output(video, raw[audio_index], outfile, acodec='aac', vcodec='libx264', crf='18', preset='slow').run()
if __name__ == '__main__':
main()