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()