ASJServer/CeleryTask/TimeAlbumTask.py

import logging
import os
import subprocess
from celery import shared_task
from Ansjer.config import LOGGER
import django

# 设置 Django 的环境变量
os.environ.setdefault("DJANGO_SETTINGS_MODULE", 'Ansjer.cn_config.test_settings')  # 确保替换为实际的 settings 模块路径
django.setup()

@shared_task
def start_xvfb():
    # 启动 Xvfb
    subprocess.Popen(['Xvfb', ':99', '-screen', '0', '1024x768x16'])
    os.environ['DISPLAY'] = ':99'

@shared_task
def generate_video(image_files, output_path):
    LOGGER.info('start开始视频生成任务')
    start_xvfb.delay()  # 确保 Xvfb 已经启动
    try:
        video_files = []
        music_path = "static/temp/music.mp3"
        # 定义特效列表
        transitions = [
            "BowTieHorizontal.glsl",
            "burn.glsl",
            "cube.glsl",
            "pinwheel.glsl",
            "windowslice.glsl",
            "Radial.glsl",
            "rotateTransition.glsl",
            "wind.glsl",
            "squeeze.glsl"
        ]

        # 生成相邻图片的视频
        for i in range(len(image_files) - 1):
            output_video = os.path.join(output_path, f'{i + 1}_{i + 2}.mp4')
            video_files.append(output_video)

            cmd = [
                'ffmpeg', '-loop', '1', '-i', image_files[i],
                '-loop', '1', '-i', image_files[i + 1],
                '-filter_complex',
                f'gltransition=duration=1:offset=0.7:source=static/gl-transitions/{transitions[i]}',
                '-t', '2', output_video
            ]

            subprocess.run(cmd, check=True)

        # 生成拼接列表文件
        with open(os.path.join(output_path, 'filelist.txt'), 'w') as f:
            for video in video_files:
                f.write(f"file '{video}'\n")

        # 拼接视频
        final_output = os.path.join(output_path, 'final_output.mp4')
        subprocess.run(['ffmpeg', '-f', 'concat', '-safe', '0', '-i', 'filelist.txt', '-c', 'copy', final_output],
                       check=True)

        # 添加背景音乐
        final_output_with_music = os.path.join(output_path, 'final_output_with_music.mp4')
        subprocess.run(['ffmpeg', '-i', final_output, '-i', music_path, '-c:v', 'copy', '-c:a', 'aac', '-b:a', '192k',
                        '-shortest', final_output_with_music], check=True)

        # 清理临时文件
        for video in video_files:
            os.remove(video)
        os.remove('static/temp/filelist.txt')
    except Exception as e:
        LOGGER.info(f'视频合成失败: {e}')