seriousm4x/ffmpeg-edl.py

## ffmpeg-edl.py
#!/usr/bin/env python3

"""
Cut Adobe Premiere .edl into single clips with ffmpeg.
Make cuts in the video track in Premiere, and export it as an edl file.

python ffmpeg-edl.py -i <edl> <edl2> -o output_folder

You can give this script multiple edl files to process.

ffmpeg and ffprobe are required.
"""

import argparse
import json
import os
import re
import subprocess

ffmpeg = "ffmpeg"
ffprobe = "ffprobe"


def tcToSec(timecode, fps):
    h, m, s, f = timecode.split(':')
    return float(h) * 3600 + float(m) * 60 + float(s) + float(f) / float(eval(fps))


def main(args):
    # loop edl files from cli
    for arg in args.i:
        edl_ext = os.path.splitext(os.path.basename(arg))[1]
        if edl_ext != ".edl":
            print("File is not an .edl")
            exit()
        with open(arg, "r", encoding="utf-8") as f:
            edl = f.read()
        edl_path = os.path.dirname(os.path.realpath(arg))
        # loop through cut blocks in edl
        for block in edl.split("\n\n"):
            clip = re.findall(r'\* FROM CLIP NAME: (.*)', block)
            if not clip:
                continue
            clip = clip[0]
            clip_name, clip_ext = os.path.splitext(os.path.basename(clip))
            # loop through lines in a block
            for line in block.split("\n"):
                if not "V  " in line:
                    continue
                # collect clip and timecode infos
                video_streams = subprocess.Popen([ffprobe, "-v", "warning", "-print_format",
                                                  "json", "-show_format", "-show_streams", clip],
                                                 stdout=subprocess.PIPE, shell=True)
                data = json.loads(video_streams.stdout.read())
                fps = data["streams"][0]["r_frame_rate"]
                splitted_line = line.split(" ")[::-1]
                inpoint = tcToSec(splitted_line[3], fps)
                outpoint = tcToSec(splitted_line[2], fps)
                duration = str(outpoint - inpoint)
                cut = line.split(" ")[0]
                # ffmpeg
                ffin = os.path.join(edl_path, clip)
                ffout = os.path.join(
                    args.o, f"{clip_name} # cut{cut}{clip_ext}")
                cmd = [ffmpeg, "-hide_banner", "-v", "warning", "-ss", str(inpoint), "-i", ffin,
                       "-t", duration, "-c", "copy", "-y", ffout]
                print("Writing", os.path.abspath(ffout))
                subprocess.check_call(cmd)


if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument('-i', help='Premiere .edl file', nargs='*')
    parser.add_argument('-o', help='Output folder', nargs="?",
                        default=os.path.dirname(os.path.abspath(__file__)))
    args = parser.parse_args()
    if not os.path.isdir(args.o):
        os.mkdir(args.o)
    main(args)
	#!/usr/bin/env python3

	"""
	Cut Adobe Premiere .edl into single clips with ffmpeg.
	Make cuts in the video track in Premiere, and export it as an edl file.

	python ffmpeg-edl.py -i <edl> <edl2> -o output_folder

	You can give this script multiple edl files to process.

	ffmpeg and ffprobe are required.
	"""

	import argparse
	import json
	import os
	import re
	import subprocess

	ffmpeg = "ffmpeg"
	ffprobe = "ffprobe"


	def tcToSec(timecode, fps):
	h, m, s, f = timecode.split(':')
	return float(h) * 3600 + float(m) * 60 + float(s) + float(f) / float(eval(fps))


	def main(args):
	# loop edl files from cli
	for arg in args.i:
	edl_ext = os.path.splitext(os.path.basename(arg))[1]
	if edl_ext != ".edl":
	print("File is not an .edl")
	exit()
	with open(arg, "r", encoding="utf-8") as f:
	edl = f.read()
	edl_path = os.path.dirname(os.path.realpath(arg))
	# loop through cut blocks in edl
	for block in edl.split("\n\n"):
	clip = re.findall(r'\* FROM CLIP NAME: (.*)', block)
	if not clip:
	continue
	clip = clip[0]
	clip_name, clip_ext = os.path.splitext(os.path.basename(clip))
	# loop through lines in a block
	for line in block.split("\n"):
	if not "V " in line:
	continue
	# collect clip and timecode infos
	video_streams = subprocess.Popen([ffprobe, "-v", "warning", "-print_format",
	"json", "-show_format", "-show_streams", clip],
	stdout=subprocess.PIPE, shell=True)
	data = json.loads(video_streams.stdout.read())
	fps = data["streams"][0]["r_frame_rate"]
	splitted_line = line.split(" ")[::-1]
	inpoint = tcToSec(splitted_line[3], fps)
	outpoint = tcToSec(splitted_line[2], fps)
	duration = str(outpoint - inpoint)
	cut = line.split(" ")[0]
	# ffmpeg
	ffin = os.path.join(edl_path, clip)
	ffout = os.path.join(
	args.o, f"{clip_name} # cut{cut}{clip_ext}")
	cmd = [ffmpeg, "-hide_banner", "-v", "warning", "-ss", str(inpoint), "-i", ffin,
	"-t", duration, "-c", "copy", "-y", ffout]
	print("Writing", os.path.abspath(ffout))
	subprocess.check_call(cmd)


	if __name__ == '__main__':
	parser = argparse.ArgumentParser()
	parser.add_argument('-i', help='Premiere .edl file', nargs='*')
	parser.add_argument('-o', help='Output folder', nargs="?",
	default=os.path.dirname(os.path.abspath(__file__)))
	args = parser.parse_args()
	if not os.path.isdir(args.o):
	os.mkdir(args.o)
	main(args)