mirror of https://github.com/sgoudham/Enso-Bot.git
You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
303 lines
8.3 KiB
Python
303 lines
8.3 KiB
Python
5 years ago
|
#!/usr/local/bin/python3
|
||
|
|
# module sys
|
||
|
|
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
|
||
|
|
def ins_img(input_file, img_data, out_file):
|
||
|
|
try:
|
||
|
|
if len(img_data) <= 0:
|
||
|
|
return False
|
||
|
|
|
||
|
|
img_list = []
|
||
|
|
img_list_str = " -i "
|
||
|
|
png_complex = []
|
||
|
|
complex_png_str = ","
|
||
|
|
|
||
|
|
for img in img_data:
|
||
|
|
if len(img["x"]) == 0:
|
||
|
|
img["x"] = "0"
|
||
|
|
|
||
|
|
if len(img["y"]) == 0:
|
||
|
|
img["y"] = "0"
|
||
|
|
img_list.append(img["img"])
|
||
|
|
|
||
|
|
if len(img["str_time"]) > 0:
|
||
|
|
if len(img["end_time"]) > 0:
|
||
|
|
cmp_str = "overlay=x=%s:y=%s:enable='if(gt(t,%s),lt(t,%s))'" % (img["x"], img["y"], img["str_time"], img["end_time"])
|
||
|
|
else:
|
||
|
|
cmp_str = "overlay=x=%s:y=%s:enable='if(gt(t,%s))'" % (img["x"], img["y"], img["str_time"])
|
||
|
|
else:
|
||
|
|
cmp_str = "overlay=x=%s:y=%s" % (img["x"], img["y"])
|
||
|
|
|
||
|
|
png_complex.append(cmp_str)
|
||
|
|
|
||
|
|
img_str_list = img_list_str.join(img_list)
|
||
|
|
complex_png_str = complex_png_str.join(png_complex)
|
||
|
|
|
||
|
|
cmd = "ffmpeg -i %s -i %s -filter_complex \"%s\" -y %s" % (input_file, img_str_list, complex_png_str, out_file)
|
||
|
|
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频添加动图 gif apng
|
||
|
|
def ins_dynamic_img(input_file, img_data, out_file):
|
||
|
|
try:
|
||
|
|
if img_data["img"] == "":
|
||
|
|
return False
|
||
|
|
|
||
|
|
if img_data["x"] == "":
|
||
|
|
img_data["x"] = 0
|
||
|
|
|
||
|
|
if img_data["y"] == "":
|
||
|
|
img_data["y"] = 0
|
||
|
|
|
||
|
|
if img_data["str_time"] != "":
|
||
|
|
if img_data["end_time"] != "":
|
||
|
|
comp = "overlay=x=%s:y=%s:shortest=1:enable='if(gt(t,%s), lt(t,%s))'" % (img_data["x"], img_data["y"],
|
||
|
|
img_data["str_time"],
|
||
|
|
img_data["end_time"])
|
||
|
|
else:
|
||
|
|
comp = "overlay=x=%s:y=%s:shortest=1:enable='if(gt(t,%s)'" % (img_data["x"], img_data["y"],
|
||
|
|
img_data["str_time"])
|
||
|
|
else:
|
||
|
|
comp = "overlay=x=%s:y=%s:shortest=1"
|
||
|
|
|
||
|
|
cmd = "ffmpeg -i %s -ignore_loop 0 -i %s -filter_complex \"%s\" -y %s" % (input_file, img_data["img"], comp,
|
||
|
|
out_file)
|
||
|
|
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频静音 分离音频流
|
||
|
|
|
||
|
|
def separate_audio(input_file, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -vcodec copy -an %s" % (input_file, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频静音 使用静音帧 为视频静音
|
||
|
|
def video_ins_mute_audio(input_file, mute_mp3_file, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -filter_complex '[1:0]apad' -shortest %s" % (input_file, mute_mp3_file, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频设置分辨率 及 码率
|
||
|
|
def trans_code(input_file, width, height, rate, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -s %sx%s -b %sk -acodec copy %s" % (input_file, width, height, rate, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频添加弹幕
|
||
|
|
def ins_barrage(input_file, barrage, out_file):
|
||
|
|
try:
|
||
|
|
if len(barrage) == 0:
|
||
|
|
return False
|
||
|
|
|
||
|
|
bag = []
|
||
|
|
bag_str = ", "
|
||
|
|
vf_str = ""
|
||
|
|
|
||
|
|
for val in barrage:
|
||
|
|
if val["fontsize"] == "":
|
||
|
|
val["fontsize"] = 40
|
||
|
|
|
||
|
|
if val["fontcolor"] == "":
|
||
|
|
val["fontcolor"] = "white"
|
||
|
|
|
||
|
|
if val["y"] == "":
|
||
|
|
val["y"] = "100"
|
||
|
|
|
||
|
|
if val["str_time"] == "":
|
||
|
|
val["str_time"] = 0
|
||
|
|
else:
|
||
|
|
val["str_time"] = int(val["str_time"])
|
||
|
|
|
||
|
|
if val["speet"] == "":
|
||
|
|
val["speet"] = 150
|
||
|
|
else:
|
||
|
|
val["speet"] = int(val["speet"])
|
||
|
|
|
||
|
|
txt = "drawtext=text='%s':fontcolor=%s:fontsize=%s:fontfile=%s:y=%s:x=w-(t-%d)*%d:enable='gte(t,%d)'" % (
|
||
|
|
val["context"],
|
||
|
|
val["fontcolor"],
|
||
|
|
val["fontsize"],
|
||
|
|
val["fontfile"],
|
||
|
|
val["y"],
|
||
|
|
val["str_time"],
|
||
|
|
val["speet"],
|
||
|
|
val["str_time"]
|
||
|
|
)
|
||
|
|
bag.append(txt)
|
||
|
|
|
||
|
|
vf_str = bag_str.join(bag)
|
||
|
|
|
||
|
|
cmd = "ffmpeg -y -i %s -vf \"%s\" %s" % (input_file, vf_str, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 调整视频速率 speed 小于 1 减速，大于 1 加速 1 等速
|
||
|
|
def playback_speed(input_file, speed, out_file):
|
||
|
|
try:
|
||
|
|
if speed == "":
|
||
|
|
speed = "1"
|
||
|
|
cmd = "ffmpeg -y -i %s -filter_complex \"setpts=PTS/%s\" %s" % (input_file, speed, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频倒放 ( 视频 + 音频 )
|
||
|
|
def a_v_reverse(input_file, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -vf vf reverse -af areverse %s " % (input_file, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频倒放 (视频)
|
||
|
|
def v_reverse(input_file, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -vf vf reverse %s " % (input_file, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频截取 截取 duration 时长的视频 从 str_second 开始截取
|
||
|
|
def v_intercept(input_file, str_second, duration, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -i %s -ss %s -t %s -f mp4 %s" % (input_file, str_second, duration, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频合并 严格模式 文件协议合并
|
||
|
|
def strict_v_merge(input_file, out_file):
|
||
|
|
try:
|
||
|
|
cmd = "ffmpeg -y -f concat -safe 0 -i %s -acodec copy %s" % (input_file, out_file)
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频合并 有损模式 input_file_list = ["1.mp4", "2.ts", "3.flv"]
|
||
|
|
def damage_v_merge(input_file_list, out_file):
|
||
|
|
try:
|
||
|
|
if len(input_file_list) < 2:
|
||
|
|
return False
|
||
|
|
|
||
|
|
video = []
|
||
|
|
video_n = len(input_file_list)
|
||
|
|
video_str = " -i "
|
||
|
|
|
||
|
|
comp_list = []
|
||
|
|
comp_str = " "
|
||
|
|
i = 0
|
||
|
|
for val in input_file_list:
|
||
|
|
video.append(val)
|
||
|
|
v_str = "[%s:a][%s:v]" % (i, i)
|
||
|
|
comp_list.append(v_str)
|
||
|
|
|
||
|
|
i += 1
|
||
|
|
|
||
|
|
video_list = video_str.join(video)
|
||
|
|
com_list_str = comp_str.join(comp_list)
|
||
|
|
|
||
|
|
cmd = "ffmpeg -y -i %s -filter_complex \"%s concat=n=%d:v=1:a=1\" -vcodec h264_nvenc %s" % (
|
||
|
|
video_list,
|
||
|
|
com_list_str,
|
||
|
|
video_n,
|
||
|
|
out_file
|
||
|
|
)
|
||
|
|
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|
||
|
|
|
||
|
|
|
||
|
|
# 视频转 图片
|
||
|
|
def video_trans_img(input_file, out_path, img_prefix, category="png"):
|
||
|
|
try:
|
||
|
|
out_path = out_path.rstrip("/")
|
||
|
|
img = img_prefix + "_%d"
|
||
|
|
|
||
|
|
out_img = "%s/%s.%s" % (out_path, img, category)
|
||
|
|
cmd = "ffmpeg -i %s -f image2 %s" % (input_file, out_img)
|
||
|
|
|
||
|
|
res = subprocess.call(cmd, shell=True)
|
||
|
|
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
except Exception:
|
||
|
|
return False
|