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, "{} {} ".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()