108 lines
3.7 KiB
Python
Executable File
108 lines
3.7 KiB
Python
Executable File
#!/usr/bin/python3
|
|
import argparse
|
|
import random
|
|
import os
|
|
import pathlib
|
|
import moviepy.editor
|
|
from moviepy.video.io.ffmpeg_tools import ffmpeg_extract_subclip
|
|
from pymediainfo import MediaInfo
|
|
import threading
|
|
|
|
|
|
class Args:
|
|
parser = ArgumentParser = argparse.ArgumentParser(
|
|
description='Extracting mediainfo, sample and screens from random timecodes from video')
|
|
parser.add_argument('-s', action='store_true', dest='screens', help="create screenshots")
|
|
parser.add_argument('-c', action='store_true', dest='sample', help="extract sample")
|
|
parser.add_argument('-m', action='store_true', dest='mediainfo', help='create mediainfo')
|
|
parser.add_argument('-n', action='store', dest="count", default=10,
|
|
help="number of screenshots, default 10", type=int)
|
|
parser.add_argument('-i', action='store', dest='input', help="input file", required=True)
|
|
args = parser.parse_args()
|
|
args.input = "D:/ReLive/Escape from Tarkov/2021.02.18-22.15.mp4"
|
|
args.screens = True
|
|
|
|
|
|
def sample(fold, exname, max_time):
|
|
"""
|
|
Create sample with duration 2m if file longer then 6m
|
|
"""
|
|
|
|
print("sample")
|
|
sam = fold + "/sample/" + exname
|
|
if max_time <= 360:
|
|
ffmpeg_extract_subclip(Args.args.input, max_time / 3, max_time * 2 / 3, targetname=sam)
|
|
else:
|
|
ffmpeg_extract_subclip(Args.args.input, max_time / 2 - 60, max_time / 2 + 60, targetname=sam)
|
|
|
|
|
|
def info(mfile, media_info):
|
|
print("Creating MediaInfo")
|
|
f = open(mfile, "w")
|
|
f.write(str(media_info))
|
|
f.close()
|
|
print("Created MediaInfo")
|
|
|
|
|
|
def screens(fold, mintimescr, maxtimescr, screen_create, num_screen):
|
|
time = random.randint(mintimescr, maxtimescr)
|
|
imgname = fold + "/" + str(time) + ".png"
|
|
print(f"{num_screen + 1} time {time}s")
|
|
|
|
screen_create.save_frame(imgname, t=time)
|
|
screen_create.reader.close()
|
|
screen_create.audio.reader.close_proc()
|
|
|
|
|
|
class Head:
|
|
"""
|
|
Creates mediainfo, screenshots and sample from file
|
|
"""
|
|
|
|
def body():
|
|
bname: str
|
|
|
|
bname = moviepy.editor.os.path.basename(Args.args.input)
|
|
fname = moviepy.editor.os.path.splitext(bname)[0]
|
|
exname = moviepy.editor.os.path.splitext(bname)[1]
|
|
print("Filename is %s" % bname)
|
|
fold = str(pathlib.Path.home()) + "/samples/" + fname
|
|
|
|
if os.path.isdir(fold):
|
|
print("Directory %s is exist" % fold)
|
|
else:
|
|
try:
|
|
moviepy.editor.os.makedirs(fold)
|
|
except OSError:
|
|
print("Creation of the directory %s failed" % fold)
|
|
else:
|
|
print("Successfully created the directory %s " % fold)
|
|
print("\nFiles will be located at %s" % fold)
|
|
|
|
media_info = MediaInfo.parse(Args.args.input, output="", parse_speed=2)
|
|
mi = MediaInfo.parse(Args.args.input)
|
|
mfile = fold + "/mediainfo.txt"
|
|
duration = mi.tracks[0]
|
|
max_time = duration.duration / 1000
|
|
|
|
if Args.args.sample:
|
|
sample(fold, exname, max_time)
|
|
if Args.args.screens:
|
|
print("Creating %s screenshots" % Args.args.count)
|
|
max_time = round(max_time)
|
|
print("Rounded duration %ss" % max_time)
|
|
screen_create = moviepy.editor.VideoFileClip(Args.args.input)
|
|
mintimescr = round(max_time * 0.05)
|
|
maxtimescr = round(max_time * 0.95)
|
|
for i in range(Args.args.count):
|
|
num_screen = i
|
|
thread = threading.Thread(name="screenshot",
|
|
target=screens(fold, mintimescr, maxtimescr, screen_create, num_screen))
|
|
thread.start()
|
|
if Args.args.mediainfo:
|
|
info(mfile, media_info)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
Head.body()
|