时光相册-视频生成 打印日志

CeleryTask/TimeAlbumTask.py

@@ -12,53 +12,57 @@ django.setup()
 
 @shared_task
 def generate_video(image_files, output_path):
-    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"
-    ]
+    LOGGER.info('start开始视频生成任务')
+    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)
+        # 生成相邻图片的视频
+        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=gl-transitions/{transitions[i]}',
-            '-t', '2', 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=gl-transitions/{transitions[i]}',
+                '-t', '2', output_video
+            ]
 
-        subprocess.run(cmd, check=True)
+            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")
+        # 生成拼接列表文件
+        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 = 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)
+        # 添加背景音乐
+        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('filelist.txt')
+        # 清理临时文件
+        for video in video_files:
+            os.remove(video)
+        os.remove('static/temp/filelist.txt')
+    except Exception as e:
+        LOGGER.info(f'视频合成失败: {e}')