自动给视频加客户专属的封面以及片尾,加完之后自动把视频移动到已加封面的文件夹,实现自动化添加封面操作

自动给视频加客户专属的封面以及片尾,加完之后自动把视频移动到已加封面的文件夹,实现自动化添加封面操作

from moviepy.editor import *
from moviepy.video.io.VideoFileClip import VideoFileClip
import os,time,random
import shutil


#遍历文件夹,找出所有mp4,flv视频文件
def func(filepath,clip_path = []):
    files = os.listdir(filepath)#查看当前目录中的文件
    #print(files)
    for file in files:#拿到每一个文件名

        file_p = os.path.join(filepath,file)#形成文件的地址
        #print(file_p)
        file_name = file_p.split('.')

        if os.path.isdir(file_p):#如果文件是文件夹
            clip_path = func(file_p,clip_path = clip_path)
        elif file_name[-1] in ['mp4', 'flv', 'avi']:
            clip_path.append(file_p)
            # print(clip_path)
        else:
            continue
    return clip_path

def add_cover(client,index,before_path,after_path):

    n = 0

    cover = ImageClip(f"./封面/封面{client}.png").set_duration(1).set_pos('left','top')
    clip3 = VideoFileClip(random.choice(func(f'./片尾/片尾{client}',clip_path = []))).resize(cover.size)
    for clip2_path in func(filepath=before_path,clip_path = []):
        #print(clip2_path)
        clip2 = VideoFileClip(clip2_path).resize(cover.size)

        #连接视频对象
        final_clip1 =concatenate_videoclips([clip2,clip3],method='compose',transition=None,bg_color=None, ismask=False, padding = 0)
        final_clip = CompositeVideoClip([final_clip1,cover])
        #输出拼接的视频到名为out_put.mp4的文件夹里

        n += 1
        order = time.strftime('%Y%m%d', time.localtime())
        if not os.path.exists(f'out_put{client}'):
            os.mkdir(f'out_put{client}')
        final_clip.write_videofile(f"./out_put{client}/{client}{order}{n}.mp4",bitrate='1800k',threads=4)
        print(f'今日已完成{client}第{n}个视频')
        #加完封面片尾后,将视频移动到已加封面的文件夹里
        if index == -1:
            fpath,fname = os.path.split(clip2_path)
            shutil.move(clip2_path,os.path.join(after_path,fname))
            print(f'第{n}个视频已经移到已加封面文件夹')
if __name__ == '__main__':
    before_path = 'E:\\movie\\加客户封面片尾\\抖音美食视频\\未加封面'
    after_path = 'E:\\movie\\加客户封面片尾\\抖音美食视频\\\已加封面'
    clients = ['dongmen','kaicheng','tongan','yaowangtang','jinxiang','健儿散']
    # clients = ['药实惠','千金']
    for client in  clients:
        if client == clients[-1]:
            add_cover(client, index=-1,before_path=before_path,after_path=after_path)
        else:
            add_cover(client,index=0,before_path=before_path,after_path=after_path)