Browse Source

时光相册-视频生成 启动 Xvfb

linhaohong 11 months ago
parent
commit
ba4230fa17
1 changed files with 6 additions and 0 deletions
  1. 6 0
      CeleryTask/TimeAlbumTask.py

+ 6 - 0
CeleryTask/TimeAlbumTask.py

@@ -9,10 +9,16 @@ import django
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", 'Ansjer.cn_config.test_settings')  # 确保替换为实际的 settings 模块路径
 os.environ.setdefault("DJANGO_SETTINGS_MODULE", 'Ansjer.cn_config.test_settings')  # 确保替换为实际的 settings 模块路径
 django.setup()
 django.setup()
 
 
+@shared_task
+def start_xvfb():
+    # 启动 Xvfb
+    subprocess.Popen(['Xvfb', ':99', '-screen', '0', '1024x768x16'])
+    os.environ['DISPLAY'] = ':99'
 
 
 @shared_task
 @shared_task
 def generate_video(image_files, output_path):
 def generate_video(image_files, output_path):
     LOGGER.info('start开始视频生成任务')
     LOGGER.info('start开始视频生成任务')
+    start_xvfb.delay()  # 确保 Xvfb 已经启动
     try:
     try:
         video_files = []
         video_files = []
         music_path = "static/temp/music.mp3"
         music_path = "static/temp/music.mp3"