videoslicer-horizontal.py

#!/usr/bin/env python
# uses ffmpeg crop filter
# slices a video in 3 eqal vertical scenes
# e.g. a video with 5780x1080 will make 3x 1920x1080 scenes
# _______________________________
# |         |         |         |
# | scene 1 | scene 2 | scene 3 |
# |         |         |         |
# |_________|_________|_________|

import os
from pathlib import Path
import subprocess

# define users home directory
home = str(Path.home())

# video input files (current directory)
video_input_directory = Path.cwd()

# video output directory
video_output = os.path.join(home,'Desktop', 'sliced_videos')
Path(video_output).mkdir(parents=True, exist_ok=True)

# video container that script searches for
#mimetype = '.mkv'
mimetype = ['.mp4','.MP4','.MTS','mkv']

# nameing of the file should be "date" + space + "time"
# eg:   2022-05-24 15-46-07.mkv  
def video_slicer(root, file):
    videoin =  os.path.join(root, file)
    
    # create folders depending on date and time in filename
    video_day, video_time = Path(file).stem.split(' ')
    print(video_day, video_time)
    Path(video_output, video_day, video_time).mkdir(parents=True, exist_ok=True)
    
    videoout1 =  os.path.join(video_output, video_day, video_time, video_day+'_'+video_time+'_scene1'+'.mkv')
    videoout2 =  os.path.join(video_output, video_day, video_time, video_day+'_'+video_time+'_scene2'+'.mkv')
    videoout3 =  os.path.join(video_output, video_day, video_time, video_day+'_'+video_time+'_scene3'+'.mkv')

    # slice the videos with ffmpeg
    subprocess.call(['ffmpeg', '-i', videoin, '-filter:v', 'crop=iw/3:ih:0:0',    '-c:a', 'copy', videoout1, '-y' ])
    subprocess.call(['ffmpeg', '-i', videoin, '-filter:v', 'crop=iw/3:ih:iw/3:0', '-c:a', 'copy', videoout2, '-y'])
    subprocess.call(['ffmpeg', '-i', videoin, '-filter:v', 'crop=iw/3:ih:(iw/3)*2:0', '-c:a', 'copy' ,videoout3, '-y' ])

def main():
    for root, dirs, files in os.walk( video_input_directory ):
        for file in files:
            if file.endswith( mimetype ):
                video_slicer(root,file)

#def main():
#    for root, dirs, files in os.walk( video_input_directory ):
#        for file in files:
#            name, extension = os.path.splitext(file)
#            if extension in mimetype:
#                video_slicer(root,file)

if __name__ == '__main__':
    main()