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.
216 lines
6.3 KiB
Python
216 lines
6.3 KiB
Python
4 years ago
|
#!/usr/bin/python2.7
|
||
|
|
# coding=utf-8
|
||
|
|
|
||
|
|
import os
|
||
|
|
|
||
|
|
import json
|
||
|
|
|
||
|
|
import subprocess
|
||
|
|
|
||
|
|
|
||
|
|
class Stream(object):
|
||
|
|
def __init__(self):
|
||
|
|
self.cmd = ""
|
||
|
|
self.out_file = ""
|
||
|
|
self.vcode_type = ""
|
||
|
|
self.input_file = ""
|
||
|
|
self.word_list_str = ""
|
||
|
|
self.subbtitle_file = ""
|
||
|
|
|
||
|
|
self.cmd = []
|
||
|
|
self.img_file = []
|
||
|
|
self.word_list = []
|
||
|
|
self.img_dynamic_list = []
|
||
|
|
|
||
|
|
self.image_list = {}
|
||
|
|
self.dynamic_list = {}
|
||
|
|
|
||
|
|
# 输入文件
|
||
|
|
def input(self, file):
|
||
|
|
self.input_file = file
|
||
|
|
|
||
|
|
# 添加图片
|
||
|
|
def img(self, img, x="0", y="0", str_time="0", end_time="0"):
|
||
|
|
if img == "":
|
||
|
|
return False
|
||
|
|
input_info = self.video_info()
|
||
|
|
|
||
|
|
if end_time == "0":
|
||
|
|
end_time = float(input_info["format"]["duration"]) + 10.0
|
||
|
|
|
||
|
|
img_data = {
|
||
|
|
"img": img,
|
||
|
|
"x": str(x),
|
||
|
|
"y": str(y),
|
||
|
|
"str_time": str(str_time),
|
||
|
|
"end_time": str(end_time)
|
||
|
|
}
|
||
|
|
|
||
|
|
self.img_file.append(img_data)
|
||
|
|
|
||
|
|
img_input = []
|
||
|
|
img_overlay = []
|
||
|
|
|
||
|
|
for val in self.img_file:
|
||
|
|
img_input.append(" -i %s" % val["img"])
|
||
|
|
img_overlay.append(" overlay=x=%s:y=%s:enable='if(gt(t,%s),lt(t,%s))" % (
|
||
|
|
val["x"],
|
||
|
|
val["y"],
|
||
|
|
val["str_time"],
|
||
|
|
val["end_time"]
|
||
|
|
)
|
||
|
|
)
|
||
|
|
|
||
|
|
img_input_str = " ".join(img_input)
|
||
|
|
img_overlay_str = ",".join(img_overlay)
|
||
|
|
|
||
|
|
self.image_list = {
|
||
|
|
"input": img_input_str,
|
||
|
|
"overlay": img_overlay_str
|
||
|
|
}
|
||
|
|
|
||
|
|
# 添加动态图片 gif apng 等
|
||
|
|
def img_dynamic(self, file, x="0", y="0", str_time="0", end_time="0"):
|
||
|
|
input_info = self.video_info()
|
||
|
|
if file == "":
|
||
|
|
return False
|
||
|
|
if end_time == "":
|
||
|
|
end_time = float(input_info["format"]["duration"]) + 10.0
|
||
|
|
|
||
|
|
apng = {
|
||
|
|
"input": " -ignore_loop 0 -i %s" % file,
|
||
|
|
"x": str(x),
|
||
|
|
"y": str(y),
|
||
|
|
"str_time": str(str_time),
|
||
|
|
"end_time": str(end_time)
|
||
|
|
}
|
||
|
|
self.img_dynamic_list.append(apng)
|
||
|
|
|
||
|
|
img_dy_input = []
|
||
|
|
img_dy_overlay = []
|
||
|
|
for val in self.img_dynamic_list:
|
||
|
|
img_dy_input.append(val["input"])
|
||
|
|
img_dy_overlay.append(" overlay=x=%s:y=%s:shortest=1:enable='if(gt(t,%s), lt(t,%s))'" % (
|
||
|
|
val["x"],
|
||
|
|
val["y"],
|
||
|
|
val["str_time"],
|
||
|
|
val["end_time"]
|
||
|
|
)
|
||
|
|
)
|
||
|
|
img_dy_input_str = " ".join(img_dy_input)
|
||
|
|
img_dy_overlay_str = ",".join(img_dy_overlay)
|
||
|
|
|
||
|
|
self.dynamic_list = {
|
||
|
|
"input": img_dy_input_str,
|
||
|
|
"overlay": img_dy_overlay_str
|
||
|
|
}
|
||
|
|
|
||
|
|
# 添加文字水印
|
||
|
|
def word_water_mark(self, c, x="0", y="0", str_time="0", end_time="0", font="", color="white"):
|
||
|
|
if font == "":
|
||
|
|
return False
|
||
|
|
input_info = self.video_info()
|
||
|
|
if c == "":
|
||
|
|
return False
|
||
|
|
if end_time == "0":
|
||
|
|
end_time = float(input_info["format"]["duration"]) + 10.0
|
||
|
|
|
||
|
|
text = " drawtext=text='%s':x=%s:y=%s:enable='if(gt(t,%s),lt(t,%s))':fontfile=%s:" \
|
||
|
|
"fontcolor=%s" % (c, str(x), str(y), str(str_time), str(end_time), str(font), str(color))
|
||
|
|
self.word_list.append(text)
|
||
|
|
|
||
|
|
self.word_list_str = ",".join(self.word_list)
|
||
|
|
|
||
|
|
# 添加字幕文件 subtitles=txt.srt
|
||
|
|
def subbtitle(self, file):
|
||
|
|
self.subbtitle_file = " subtitles=%s" % file
|
||
|
|
|
||
|
|
# 编码方式 -vcodec
|
||
|
|
def vcode(self, code):
|
||
|
|
if code == "":
|
||
|
|
return False
|
||
|
|
self.vcode_type = " -vcodec %s" % code
|
||
|
|
|
||
|
|
# 输出文件
|
||
|
|
def out(self, file):
|
||
|
|
if file == "":
|
||
|
|
return False
|
||
|
|
self.out_file = "%s" % file
|
||
|
|
|
||
|
|
# 执行脚本
|
||
|
|
def run(self):
|
||
|
|
if self.input_file == "":
|
||
|
|
return False
|
||
|
|
im = "ffmpeg -i %s" % self.input_file
|
||
|
|
ov = ""
|
||
|
|
|
||
|
|
if len(self.dynamic_list) > 0 and self.dynamic_list["input"] != "":
|
||
|
|
im = "%s %s" % (im, self.dynamic_list["input"])
|
||
|
|
|
||
|
|
if ov != "":
|
||
|
|
ov = "%s,%s" % (ov, self.dynamic_list["overlay"])
|
||
|
|
else:
|
||
|
|
ov = self.dynamic_list["overlay"]
|
||
|
|
if len(self.image_list) > 0:
|
||
|
|
im = "%s %s" % (im, self.image_list["input"])
|
||
|
|
|
||
|
|
if ov != "":
|
||
|
|
ov = "%s,%s" % (ov, self.dynamic_list["overlay"])
|
||
|
|
else:
|
||
|
|
ov = self.dynamic_list["overlay"]
|
||
|
|
|
||
|
|
# 文字水印
|
||
|
|
if self.word_list_str != "":
|
||
|
|
if ov != "":
|
||
|
|
ov = "%s,%s" % (ov, self.word_list_str)
|
||
|
|
else:
|
||
|
|
ov = self.word_list_str
|
||
|
|
|
||
|
|
# 字幕
|
||
|
|
if self.subbtitle_file != "":
|
||
|
|
if ov != "":
|
||
|
|
ov = "%s,%s" % (ov, self.subbtitle_file)
|
||
|
|
else:
|
||
|
|
ov = self.subbtitle_file
|
||
|
|
|
||
|
|
if self.vcode_type != "":
|
||
|
|
self.cmd = "%s -filter_complex \"%s\" -y %s %s" % (im, ov, self.vcode_type, self.out_file)
|
||
|
|
else:
|
||
|
|
self.cmd = "%s -filter_complex \"%s\" -y %s" % (im, ov, self.out_file)
|
||
|
|
|
||
|
|
self.do()
|
||
|
|
|
||
|
|
# 获取视频的相关时长信息
|
||
|
|
def video_info(self):
|
||
|
|
result = {}
|
||
|
|
if os.path.isfile(self.input_file) is False:
|
||
|
|
return result
|
||
|
|
|
||
|
|
cmd = ['ffprobe', '-v', 'quiet', '-print_format', 'json', '-show_format', '-show_streams', self.input_file]
|
||
|
|
returned_data = subprocess.check_output(cmd)
|
||
|
|
return json.loads(returned_data.decode('utf-8'))
|
||
|
|
|
||
|
|
# 执行命令
|
||
|
|
def do(self):
|
||
|
|
if self.cmd == "":
|
||
|
|
return False
|
||
|
|
res = subprocess.call(self.cmd, shell=True)
|
||
|
|
if res != 0:
|
||
|
|
return False
|
||
|
|
return True
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == '__main__':
|
||
|
|
stream = Stream()
|
||
|
|
stream.input("face.mp4")
|
||
|
|
stream.img("t1.png")
|
||
|
|
stream.img("t2.png", "10", y=10, str_time=5, end_time=10)
|
||
|
|
stream.img_dynamic("t1.apng", x=10, y=10, str_time=5, end_time=10)
|
||
|
|
stream.img_dynamic("t2.apng", x=10, y=10, str_time=5, end_time=9)
|
||
|
|
stream.word_water_mark("测试文字水印1", x="10", y="10", str_time="0", end_time="20", font="ttf.ttf", color="white")
|
||
|
|
stream.word_water_mark("测试文字水印2", x="10", y="10", str_time="0", end_time="20", font="ttf.ttf", color="white")
|
||
|
|
stream.subbtitle("srt.srt")
|
||
|
|
stream.out("out.mp4")
|
||
|
|
stream.run()
|
||
|
|
|