import os
import shutil
from datetime import datetime

dir = '/data'
file = '/data/1.txt'
# 创建单个目录
os.mkdir(dir)
# 创建多层目录
os.makedirs(dir + '/123')
# 复制文件
shutil.copyfile('old', 'new')
# 复制文件夹
shutil.copytree('olddir', 'newdir')
# 重命名文件或目录
os.rename('odl', 'new')
# 移动文件或目录
shutil.move('old', 'new')
# 删除空目录
os.rmdir(dir)
# 删除空目录
os.removedirs(dir)
# 获取路径下的所有文件
os.listdir(dir)
# 删除文件
os.remove(file)
# 判断是否存在
if os.path.exists(dir):
    pass
# 判断是否文件
if os.path.isfile(file):
    pass
# 判断是否目录
if os.path.isdir(dir):
    pass
# 判断是否绝对路径
if os.path.isabs(dir):
    pass
# 获取文件属性
os.stat(file)
# 文件大小
os.stat(file).st_size
os.path.getsize(file)
# 获取文件名
os.path.basename(filePath)
# 获取路径
os.path.dirname(filePath)
# 获取绝对路径
os.path.abspath(file)
# 获取文件的创建时间
fromtimestamp = datetime.fromtimestamp(int(os.path.getctime(filePath))).date()
print(str(fromtimestamp))

# 获取文件后缀
basename = os.path.basename(path) # 获取文件名
extension = os.path.splitext(basename)[1] # 获取文件后缀
# 排序
sorted(os.listdir(path))

import threading
import time

from moviepy.editor import *

# 定义最大允许的线程数量
max_threads = 5
# 创建 Semaphore 对象
semaphore = threading.Semaphore(max_threads)


def asyncConcat(startPath, endPath, middlePath, middleAudioPath, outputPath, semaphore):
    with semaphore:
        concat(startPath, endPath, middlePath, middleAudioPath, outputPath)


def concat(startPath, endPath, middlePath, middleAudioPath, outputPath):
    try:
        start = VideoFileClip(startPath)
        end = VideoFileClip(endPath)
        middleVideo = VideoFileClip(middlePath)
        middleAudio = AudioFileClip(middleAudioPath)
        # 正文持续时间
        print('持续时间=', middleVideo.duration)
        # 音频长度剪切
        middleAudio = middleAudio.set_duration(middleVideo.duration)
        # 替换音频
        middleVideo = middleVideo.set_audio(middleAudio)
        # 分辨率
        resolution = middleVideo.size
        width, height = resolution
        print('分辨率=', resolution)
        print('分辨率：宽=', width, '高：', height)
        # 等比例放大或缩小
        start = start.resize(resolution)
        end = end.resize(resolution)
        # 上下
        # resultClip = clips_array([[start], [middle], [end]])
        # 左右
        # resultClip = clips_array([[start, middle, end]])
        # 前后
        resultClip = concatenate_videoclips([start, middleVideo, end])
        startTime = time.time()
        basename = os.path.basename(middlePath)
        basename = basename.replace('.', '-已剪辑.')
        resultClip.write_videofile(outputPath + basename, threads=50)
        print('剪辑使用时间：', time.time() - startTime)

    except Exception as e:
        print('计算视频失败：', e)


if __name__ == '__main__':
    startPath = "D:\\00-自媒体计划\\AI-start.mp4"
    endPath = "D:\\00-自媒体计划\\AI-end.mp4"
    middleDir = "f:\\Videos\\4K Tokkit\\\dbxbqm22\\"
    middlePath = "f:\\Videos\\4K Tokkit\\dbxbqm22\\7257911883674668289.mp4"
    middleAudioPath = "D:\\00-自媒体计划\\AI-背景音乐.MP3"
    outputPath = "f:\\Videos\\4K Tokkit\\test1\\"
    if not outputPath.endswith('\\'):
        outputPath = outputPath + '\\'
    count = 1
    fileCount = 0
    startTime = time.time()
    files = os.listdir(middleDir)
    # 创建线程
    threads = []
    for file in files:
        count = count + 1
        basename = os.path.basename(file)
        if '.mp4' not in basename:
            continue
        if '已剪辑' in basename:
            continue
        if os.path.exists(outputPath + basename.replace('.', '-已剪辑.')):
            print('视频已剪辑=', middleDir + basename)
            continue
        print('待剪辑视频=', middleDir + basename)
        fileCount = fileCount + 1
        # concat(startPath, endPath, middleDir + basename, middleAudioPath, outputPath)
        thread = threading.Thread(target=asyncConcat, args=(
            startPath, endPath, middleDir + basename, middleAudioPath, outputPath, semaphore))
        threads.append(thread)
        thread.start()
    # 等待所有线程执行结束
    for thread in threads:
        thread.join()
    print('共处理视频个数：', fileCount, ' 耗时：', time.time() - startTime)