Python实现视频片头和片尾添加

import imageio

imageio.plugins.ffmpeg.download()
from datetime import datetime
import os
from moviepy.video.io.VideoFileClip import VideoFileClip
from moviepy.video.compositing.concatenate import concatenate_videoclips
from threading import Thread
from tkinter import Tk, Label, filedialog, Button, Entry, END, Text

class App(object):

    def __init__(self):
        self.tk = Tk()
        self.tk.geometry("540x350+10+10")
        self.tk.resizable(False, False)
        self.tk.title("视频拼接")

        # 显示片头路径文本框
        self.head_entry = Entry(self.tk)
        self.head_entry.place(x=150, y=60, width=300, height=30)

        # 显示需要添加片头的视频路径文本框
        self.source_entry = Entry(self.tk)
        self.source_entry.place(x=150, y=100, width=300, height=30)

        # 显示需要添加片尾头的视频路径文本框
        self.tail_entry = Entry(self.tk)
        self.tail_entry.place(x=150, y=140, width=300, height=30)
        # 显示需要添加片尾头的视频路径文本框
        self.save_entry = Entry(self.tk)
        self.save_entry.place(x=150, y=180, width=300, height=30)

        self.log_text = Text(self.tk)
        self.log_text.place(x=60, y=220, width=330, height=100)

    def show_log(self, info):
        self.log_text.insert(END, "{} {}\n".format(datetime.now().strftime("%H:%M:%S"), info))

    def add_author(self, name, company):
        # 添加作者名称
        L_author = Label(self.tk, text=f'作者:{name}')
        L_author.config(font='Helvetica -10 bold', fg='#030303')
        L_author.place(x=440, y=330)

        # 添加作者公司
        L_title = Label(self.tk, text=f'公司:{company}')
        L_title.config(font='Helvetica -10 bold', fg='blue')
        L_title.place(x=350, y=330)

    def head_video(self):
        """获取片头视频路径"""

        # 按钮
        s_button = Button(self.tk, text=f'选择片头', command=self.get_head_video_path)
        s_button.place(x=60, y=60)

    def source_video(self):
        # 按钮
        s_button = Button(self.tk, text=f'选择视频', command=self.get_source_video_path)
        s_button.place(x=60, y=100)

    def tail_video(self):
        # 按钮
        s_button = Button(self.tk, text=f'选择片尾', command=self.get_tail_video_path)
        s_button.place(x=60, y=140)

    def save_video(self):
        # 按钮
        s_button = Button(self.tk, text=f'保存路径', command=self.get_save_video_path)
        s_button.place(x=60, y=180)

    def get_head_video_path(self):
        # 获取到片头路径显示到文本框
        head_video_path = filedialog.askopenfilename(title="选择片头")
        self.head_entry.delete(0, END)
        self.head_entry.insert(0, head_video_path)

    def get_source_video_path(self):
        # 获取到需要添加片头的视频
        source_video_path = filedialog.askopenfilenames(title="选择视频")
        self.source_entry.delete(0, END)
        self.source_entry.insert(0, source_video_path)

    def get_tail_video_path(self):
        # 获取到需要添加片头的视频文件夹写入到文本框
        tail_video_path = filedialog.askopenfilename(title="选择片尾")
        self.tail_entry.delete(0, END)
        self.tail_entry.insert(0, tail_video_path)

    def get_save_video_path(self):
        # 保存文件的路径
        tail_video_path = filedialog.askdirectory(title="保存路径")
        self.save_entry.delete(0, END)
        self.save_entry.insert(0, tail_video_path)

    def get_all_path(self):
        """从文本框获取路径，并判断是否有传"""
        head_video_path = self.head_entry.get()
        source_video_path = [] if not self.source_entry.get() else self.source_entry.get().split(" ")
        tail_video_path = self.tail_entry.get()
        save_video_path = self.save_entry.get()
        if not head_video_path and not tail_video_path:
            self.show_log("ERROR:请选择片头或片尾")
            return
        if not source_video_path:
            self.show_log("ERROR:请选择需要加片头或者片尾的视频")
            return
        if not save_video_path:
            self.show_log("ERROR:请选择保存路径")
            return
        self.show_log("INFO:路径正确")
        return head_video_path, source_video_path, tail_video_path, save_video_path

    def concat(self):
        # 获取到路径
        head_video_path, source_video_path, tail_video_path, save_video_path = self.get_all_path()

        # 先判断片头
        if head_video_path and not head_video_path.endswith(('.mp4', '.mkv', '.avi', '.wmv', '.iso')):
            self.show_log("ERROR: 片头文件不是视频格式,错误文件%s" % head_video_path)
        if tail_video_path and not head_video_path.endswith(('.mp4', '.mkv', '.avi', '.wmv', '.iso')):
            self.show_log("ERROR: 片尾文件不是视频格式,错误文件%s" % head_video_path)
        if not os.path.exists(save_video_path):
            self.show_log("ERROR: 保存文件路径不存在")
            return
        head_video = None
        tail_video = None
        if head_video_path:
            head_video = VideoFileClip(head_video_path)  # 加载片头
        if tail_video_path:
            tail_video = VideoFileClip(tail_video_path)  # 加载片尾

        for file in source_video_path:
            th = Thread(target=self._concat,args=(file,head_video,tail_video,save_video_path))
            th.start()

    def _concat(self,file,head_video,tail_video,save_video_path):
        file_name = os.path.basename(file)
        self.show_log("INFO: %s 开始拼接" % file_name)
        video_list = []
        if not file.endswith(('.mp4', '.mkv', '.avi', '.wmv', '.iso')):
            self.show_log("ERROR: 视频格式错误,错误文件%s" % file_name)
            return
        video = VideoFileClip(file)  # 加载正片

        # 判断是否有片头或者片尾，按顺序添加
        if head_video:
            video_list.append(head_video)
        video_list.append(video)
        if tail_video:
            video_list.append(tail_video)
        final_clip = concatenate_videoclips(video_list)  # 进行视频合并
        final_clip.write_videofile(os.path.join(save_video_path, file_name))
        final_clip.close()
        self.show_log("INFO: %s 拼接完成" % file_name)

    def start(self):
        # 开始拼接
        s_button = Button(self.tk, text=f'开始', command=self.concat)
        s_button.place(x=415, y=220)

    def run(self):
        self.add_author("黄贵锋", "恒企教育")
        self.head_video()  # 片头
        self.source_video()
        self.tail_video()  # 片尾
        self.save_video()  # 保存位置
        self.start()  # 点击启动按钮
        self.tk.mainloop()

if __name__ == '__main__':
    app = App()
    app.run()