mirror of
https://github.com/c0decracker/video-splitter.git
synced 2026-01-23 02:24:16 +00:00
Ported to Python 3, fixed issue with ffmpeg command
This commit is contained in:
Vishal Patel
parent
5b888f6eb3
commit
ced4fc63a4
2 changed files with 214 additions and 28 deletions
186
ffmpeg-split-py2.py
Normal file
186
ffmpeg-split-py2.py
Normal file
|
|
@ -0,0 +1,186 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
import csv
|
||||
import subprocess
|
||||
import re
|
||||
import math
|
||||
import json
|
||||
import os
|
||||
from optparse import OptionParser
|
||||
|
||||
|
||||
length_regexp = 'Duration: (\d{2}):(\d{2}):(\d{2})\.\d+,'
|
||||
re_length = re.compile(length_regexp)
|
||||
|
||||
def split_by_manifest(filename, manifest, vcodec="copy", acodec="copy",
|
||||
extra="", **kwargs):
|
||||
""" Split video into segments based on the given manifest file.
|
||||
Arguments:
|
||||
filename (str) - Location of the video.
|
||||
manifest (str) - Location of the manifest file.
|
||||
vcodec (str) - Controls the video codec for the ffmpeg video
|
||||
output.
|
||||
acodec (str) - Controls the audio codec for the ffmpeg video
|
||||
output.
|
||||
extra (str) - Extra options for ffmpeg.
|
||||
"""
|
||||
if not os.path.exists(manifest):
|
||||
print "File does not exist: %s" % manifest
|
||||
raise SystemExit
|
||||
|
||||
with open(manifest) as manifest_file:
|
||||
manifest_type = manifest.split(".")[-1]
|
||||
if manifest_type == "json":
|
||||
config = json.load(manifest_file)
|
||||
elif manifest_type == "csv":
|
||||
config = csv.DictReader(manifest_file)
|
||||
else:
|
||||
print "Format not supported. File must be a csv or json file"
|
||||
raise SystemExit
|
||||
|
||||
split_cmd = "ffmpeg -i \"%s\" -vcodec %s -acodec %s -y %s" % (filename,
|
||||
vcodec,
|
||||
acodec,
|
||||
extra)
|
||||
split_count = 1
|
||||
split_error = []
|
||||
try:
|
||||
fileext = filename.split(".")[-1]
|
||||
except IndexError as e:
|
||||
raise IndexError("No . in filename. Error: " + str(e))
|
||||
for video_config in config:
|
||||
split_str = ""
|
||||
try:
|
||||
split_start = video_config["start_time"]
|
||||
split_length = video_config.get("end_time", None)
|
||||
if not split_length:
|
||||
split_length = video_config["length"]
|
||||
filebase = video_config["rename_to"]
|
||||
if fileext in filebase:
|
||||
filebase = ".".join(filebase.split(".")[:-1])
|
||||
|
||||
split_str += " -ss " + str(split_start) + " -t " + \
|
||||
str(split_length) + \
|
||||
" \""+ filebase + "." + fileext + \
|
||||
"\""
|
||||
print "########################################################"
|
||||
print "About to run: "+split_cmd+split_str
|
||||
print "########################################################"
|
||||
output = subprocess.Popen(split_cmd+split_str,
|
||||
shell = True, stdout =
|
||||
subprocess.PIPE).stdout.read()
|
||||
except KeyError as e:
|
||||
print "############# Incorrect format ##############"
|
||||
if manifest_type == "json":
|
||||
print "The format of each json array should be:"
|
||||
print "{start_time: <int>, length: <int>, rename_to: <string>}"
|
||||
elif manifest_type == "csv":
|
||||
print "start_time,length,rename_to should be the first line "
|
||||
print "in the csv file."
|
||||
print "#############################################"
|
||||
print e
|
||||
raise SystemExit
|
||||
|
||||
|
||||
|
||||
def split_by_seconds(filename, split_length, vcodec="copy", acodec="copy",
|
||||
extra="", **kwargs):
|
||||
if split_length and split_length <= 0:
|
||||
print "Split length can't be 0"
|
||||
raise SystemExit
|
||||
|
||||
output = subprocess.Popen("ffmpeg -i \""+filename+"\" 2>&1 | grep 'Duration'",
|
||||
shell = True,
|
||||
stdout = subprocess.PIPE
|
||||
).stdout.read()
|
||||
print output
|
||||
matches = re_length.search(output)
|
||||
if matches:
|
||||
video_length = int(matches.group(1)) * 3600 + \
|
||||
int(matches.group(2)) * 60 + \
|
||||
int(matches.group(3))
|
||||
print "Video length in seconds: "+str(video_length)
|
||||
else:
|
||||
print "Can't determine video length."
|
||||
raise SystemExit
|
||||
split_count = int(math.ceil(video_length/float(split_length)))
|
||||
if(split_count == 1):
|
||||
print "Video length is less then the target split length."
|
||||
raise SystemExit
|
||||
|
||||
split_cmd = "ffmpeg -i \"%s\" -vcodec %s -acodec %s %s" % (filename, vcodec,
|
||||
acodec, extra)
|
||||
try:
|
||||
filebase = ".".join(filename.split(".")[:-1])
|
||||
fileext = filename.split(".")[-1]
|
||||
except IndexError as e:
|
||||
raise IndexError("No . in filename. Error: " + str(e))
|
||||
for n in range(0, split_count):
|
||||
split_str = ""
|
||||
if n == 0:
|
||||
split_start = 0
|
||||
else:
|
||||
split_start = split_length * n
|
||||
|
||||
split_str += " -ss "+str(split_start)+" -t "+str(split_length) + \
|
||||
" \""+filebase + "-" + str(n) + "." + fileext + \
|
||||
"\""
|
||||
print "About to run: "+split_cmd+split_str
|
||||
output = subprocess.Popen(split_cmd+split_str, shell = True, stdout =
|
||||
subprocess.PIPE).stdout.read()
|
||||
|
||||
|
||||
def main():
|
||||
parser = OptionParser()
|
||||
|
||||
parser.add_option("-f", "--file",
|
||||
dest = "filename",
|
||||
help = "File to split, for example sample.avi",
|
||||
type = "string",
|
||||
action = "store"
|
||||
)
|
||||
parser.add_option("-s", "--split-size",
|
||||
dest = "split_length",
|
||||
help = "Split or chunk size in seconds, for example 10",
|
||||
type = "int",
|
||||
action = "store"
|
||||
)
|
||||
parser.add_option("-m", "--manifest",
|
||||
dest = "manifest",
|
||||
help = "Split video based on a json manifest file. ",
|
||||
type = "string",
|
||||
action = "store"
|
||||
)
|
||||
parser.add_option("-v", "--vcodec",
|
||||
dest = "vcodec",
|
||||
help = "Video codec to use. ",
|
||||
type = "string",
|
||||
default = "copy",
|
||||
action = "store"
|
||||
)
|
||||
parser.add_option("-a", "--acodec",
|
||||
dest = "acodec",
|
||||
help = "Audio codec to use. ",
|
||||
type = "string",
|
||||
default = "copy",
|
||||
action = "store"
|
||||
)
|
||||
parser.add_option("-e", "--extra",
|
||||
dest = "extra",
|
||||
help = "Extra options for ffmpeg, e.g. '-e -threads 8'. ",
|
||||
type = "string",
|
||||
default = "",
|
||||
action = "store"
|
||||
)
|
||||
(options, args) = parser.parse_args()
|
||||
|
||||
if options.filename and options.manifest:
|
||||
split_by_manifest(**(options.__dict__))
|
||||
elif options.filename and options.split_length:
|
||||
split_by_seconds(**(options.__dict__))
|
||||
else:
|
||||
parser.print_help()
|
||||
raise SystemExit
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
|
|
@ -26,7 +26,7 @@ def split_by_manifest(filename, manifest, vcodec="copy", acodec="copy",
|
|||
extra (str) - Extra options for ffmpeg.
|
||||
"""
|
||||
if not os.path.exists(manifest):
|
||||
print "File does not exist: %s" % manifest
|
||||
print("File does not exist: %s" % manifest)
|
||||
raise SystemExit
|
||||
|
||||
with open(manifest) as manifest_file:
|
||||
|
|
@ -36,10 +36,10 @@ def split_by_manifest(filename, manifest, vcodec="copy", acodec="copy",
|
|||
elif manifest_type == "csv":
|
||||
config = csv.DictReader(manifest_file)
|
||||
else:
|
||||
print "Format not supported. File must be a csv or json file"
|
||||
print("Format not supported. File must be a csv or json file")
|
||||
raise SystemExit
|
||||
|
||||
split_cmd = "ffmpeg -i '%s' -vcodec %s -acodec %s -y %s" % (filename,
|
||||
split_cmd = "ffmpeg -i \"%s\" -vcodec %s -acodec %s -y %s" % (filename,
|
||||
vcodec,
|
||||
acodec,
|
||||
extra)
|
||||
|
|
@ -62,24 +62,24 @@ def split_by_manifest(filename, manifest, vcodec="copy", acodec="copy",
|
|||
|
||||
split_str += " -ss " + str(split_start) + " -t " + \
|
||||
str(split_length) + \
|
||||
" '"+ filebase + "." + fileext + \
|
||||
"'"
|
||||
print "########################################################"
|
||||
print "About to run: "+split_cmd+split_str
|
||||
print "########################################################"
|
||||
output = subprocess.Popen(split_cmd+split_str,
|
||||
" \"" + filebase + "." + fileext + \
|
||||
"\""
|
||||
print("########################################################")
|
||||
print("About to run: " + split_cmd + split_str)
|
||||
print("########################################################")
|
||||
output = subprocess.Popen(split_cmd + split_str,
|
||||
shell = True, stdout =
|
||||
subprocess.PIPE).stdout.read()
|
||||
except KeyError as e:
|
||||
print "############# Incorrect format ##############"
|
||||
print("############# Incorrect format ##############")
|
||||
if manifest_type == "json":
|
||||
print "The format of each json array should be:"
|
||||
print "{start_time: <int>, length: <int>, rename_to: <string>}"
|
||||
print("The format of each json array should be:")
|
||||
print("{start_time: <int>, length: <int>, rename_to: <string>}")
|
||||
elif manifest_type == "csv":
|
||||
print "start_time,length,rename_to should be the first line "
|
||||
print "in the csv file."
|
||||
print "#############################################"
|
||||
print e
|
||||
print("start_time,length,rename_to should be the first line ")
|
||||
print("in the csv file.")
|
||||
print("#############################################")
|
||||
print(e)
|
||||
raise SystemExit
|
||||
|
||||
|
||||
|
|
@ -87,29 +87,29 @@ def split_by_manifest(filename, manifest, vcodec="copy", acodec="copy",
|
|||
def split_by_seconds(filename, split_length, vcodec="copy", acodec="copy",
|
||||
extra="", **kwargs):
|
||||
if split_length and split_length <= 0:
|
||||
print "Split length can't be 0"
|
||||
print("Split length can't be 0")
|
||||
raise SystemExit
|
||||
|
||||
output = subprocess.Popen("ffmpeg -i '"+filename+"' 2>&1 | grep 'Duration'",
|
||||
output = subprocess.Popen("ffmpeg -i \""+filename+"\" 2>&1 | grep 'Duration'",
|
||||
shell = True,
|
||||
stdout = subprocess.PIPE
|
||||
).stdout.read()
|
||||
print output
|
||||
print(output)
|
||||
matches = re_length.search(output)
|
||||
if matches:
|
||||
video_length = int(matches.group(1)) * 3600 + \
|
||||
int(matches.group(2)) * 60 + \
|
||||
int(matches.group(3))
|
||||
print "Video length in seconds: "+str(video_length)
|
||||
print("Video length in seconds: " + str(video_length))
|
||||
else:
|
||||
print "Can't determine video length."
|
||||
print("Can't determine video length.")
|
||||
raise SystemExit
|
||||
split_count = int(math.ceil(video_length/float(split_length)))
|
||||
if(split_count == 1):
|
||||
print "Video length is less then the target split length."
|
||||
print("Video length is less then the target split length.")
|
||||
raise SystemExit
|
||||
|
||||
split_cmd = "ffmpeg -i '%s' -vcodec %s -acodec %s %s" % (filename, vcodec,
|
||||
split_cmd = "ffmpeg -i \"%s\" -vcodec %s -acodec %s %s" % (filename, vcodec,
|
||||
acodec, extra)
|
||||
try:
|
||||
filebase = ".".join(filename.split(".")[:-1])
|
||||
|
|
@ -123,11 +123,11 @@ def split_by_seconds(filename, split_length, vcodec="copy", acodec="copy",
|
|||
else:
|
||||
split_start = split_length * n
|
||||
|
||||
split_str += " -ss "+str(split_start)+" -t "+str(split_length) + \
|
||||
" '"+filebase + "-" + str(n) + "." + fileext + \
|
||||
"'"
|
||||
print "About to run: "+split_cmd+split_str
|
||||
output = subprocess.Popen(split_cmd+split_str, shell = True, stdout =
|
||||
split_str += " -ss " + str(split_start) + " -t " + str(split_length) + \
|
||||
" \"" + filebase + "-" + str(n) + "." + fileext + \
|
||||
"\""
|
||||
print("About to run: " + split_cmd + split_str)
|
||||
output = subprocess.Popen(split_cmd + split_str, shell = True, stdout =
|
||||
subprocess.PIPE).stdout.read()
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue