mirror of
https://github.com/soimort/you-get.git
synced 2025-01-24 05:55:02 +03:00
60 lines
1.4 KiB
Python
60 lines
1.4 KiB
Python
|
#!/usr/bin/env python
|
||
|
|
|
||
|
|
##################################################
|
||
|
|
# main
|
||
|
|
##################################################
|
||
|
|
|
||
|
|
def guess_output(inputs):
|
||
|
|
import os.path
|
||
|
|
inputs = map(os.path.basename, inputs)
|
||
|
|
n = min(map(len, inputs))
|
||
|
|
for i in reversed(range(1, n)):
|
||
|
|
if len(set(s[:i] for s in inputs)) == 1:
|
||
|
|
return inputs[0][:i] + '.ts'
|
||
|
|
return 'output.ts'
|
||
|
|
|
||
|
|
def concat_ts(tss, output = None):
|
||
|
|
assert tss, 'no ts file found'
|
||
|
|
import os.path
|
||
|
|
if not output:
|
||
|
|
output = guess_output(tss)
|
||
|
|
elif os.path.isdir(output):
|
||
|
|
output = os.path.join(output, guess_output(tss))
|
||
|
|
|
||
|
|
print('Merging video parts...')
|
||
|
|
ins = [open(ts, 'rb') for ts in tss]
|
||
|
|
with open(output, 'wb') as output:
|
||
|
|
for i in ins:
|
||
|
|
output.write(i.read())
|
||
|
|
|
||
|
|
return output
|
||
|
|
|
||
|
|
def usage():
|
||
|
|
print('Usage: [python3] join_ts.py --output TARGET.ts ts...')
|
||
|
|
|
||
|
|
def main():
|
||
|
|
import sys, getopt
|
||
|
|
try:
|
||
|
|
opts, args = getopt.getopt(sys.argv[1:], "ho:", ["help", "output="])
|
||
|
|
except getopt.GetoptError as err:
|
||
|
|
usage()
|
||
|
|
sys.exit(1)
|
||
|
|
output = None
|
||
|
|
for o, a in opts:
|
||
|
|
if o in ("-h", "--help"):
|
||
|
|
usage()
|
||
|
|
sys.exit()
|
||
|
|
elif o in ("-o", "--output"):
|
||
|
|
output = a
|
||
|
|
else:
|
||
|
|
usage()
|
||
|
|
sys.exit(1)
|
||
|
|
if not args:
|
||
|
|
usage()
|
||
|
|
sys.exit(1)
|
||
|
|
|
||
|
|
concat_ts(args, output)
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
main()
|