from  moviepy.editor import *
import os,traceback

def isdigit(c):
    return c>='0' and c<='9'

def getPattern(patternString):
    """
    :param patternString: 含有模式串的字符串，模式串仿C语言整型输出格式，类似于：%d、%02d等
    :return: 返回将patternString拆分成三段的一个列表，三个元素分别是：模式串前面的字符串、模式串（含百分号、数字和d)、模式串后面的字符串，如果没有找到模式串则返回None
    """
    patternIndex = patternString.index("%")
    if patternIndex<0:
        print("模式字符串中未发现模式匹配符：%")
        return None
    matched = False
    for patternEnd in range(patternIndex+1,len(patternString)):
        if patternString[patternEnd]in'dD':
            matched = True
            break
        else:
            if  not isdigit(patternString[patternEnd]):
                print("模式字符串中的模式存在非数字")
                return None
    if matched: return [patternString[:patternIndex],patternString[patternIndex:patternEnd+1],patternString[patternEnd+1:]]
    else:
        print("模式字符串中的模式不是字母d结尾")
        return None

def composePPTImgsAndSounds(imgFilePattern,soundFilePattern,startSeq=1,endSeq=None):
    """
    :param imgFilePattern: 图像文件模式，要求图像文件中必须含文件序号，序号采用%d、%02d/%03d的方式来识别
    :param soundFilePattern:音频文件格式，要求文件中必须含文件序号，序号采用%d、%02d/%03d的方式来识别
    :param startSeq:文件序列号其实值，一般为0或1
    :param endSeq:文件序列号结束值，可以不填
    :return:生成的视频
    """
    if endSeq is None:endSeq=9999999  #当对应序号找不到文件时结束
    patterns = getPattern(imgFilePattern)
    if patterns is None:
        print('图像文件名不符合模式格式要求')
        return None
    else:
        imgPrefix,imgPattern,imgPostfix = patterns
    patterns = getPattern(soundFilePattern)
    if patterns is None:
        print('音频文件名不符合模式格式要求')
        return None
    else:
        soundPrefix, soundPattern, soundPostfix = patterns
    clips = []
    try:
        for seq in range(startSeq,endSeq):
            imgFName = f"{imgPrefix}{seq:{imgPattern[1:]}}{imgPostfix}"
            soundFName = f"{soundPrefix}{seq:{soundPattern[1:]}}{soundPostfix}"

            if os.path.exists(soundFName) and os.path.exists(imgFName):
                soundClip = AudioFileClip(soundFName)
                imgClip = ImageClip(imgFName).set_duration(soundClip.duration).set_fps(2).set_audio(soundClip)
                clips.append(imgClip)
            else:
                if seq:
                    print(f"音视和图像文件加载结束，最后一个文件的序号为{seq}")
                    break
                else:
                    print(f"图像文件{imgClip}或音频文件{soundClip}不存在")
                    return None

    except Exception as e:
        print(f"加载音频和图像文件失败：\n{e}\n{traceback.format_exc()}")
        return None

    print("输出视频文件....")
    clip = concatenate_videoclips(clips,'compose')
    clip.write_videofile(r"C:\temp\ppt.mp4",threads=8)
    print("视频文件已经生成")
    return clip


composePPTImgsAndSounds(r'F:\bak\pr\大H\十百千\视频制作\5G在线计费动态授权机制介绍3.9\幻灯片%d.PNG',r'C:\temp\录音文件\p%d.m4a')