瀏覽代碼

fix video

tomoya 2 年之前
父節點
當前提交
85596e5f09
共有 2 個文件被更改,包括 3 次插入5 次删除
  1. 1 3
      backend/app/app/api/api_v1/endpoints/videos.py
  2. 2 2
      backend/app/app/worker.py

+ 1 - 3
backend/app/app/api/api_v1/endpoints/videos.py

@@ -67,9 +67,7 @@ def upload_plot(
     finally:
         upload_file.file.close()
 
-    celery_app.send_task("app.worker.test_celery", kwargs={"video_id": video.id, 
-                                                           "zip_filename": video.stored_file_name+".zip", 
-                                                           "user_id": current_user.id})
+    celery_app.send_task("app.worker.test_celery", args=[video.id, video.stored_file_name, current_user.id])
     return video
 
 @router.get("/{id}")

+ 2 - 2
backend/app/app/worker.py

@@ -21,14 +21,14 @@ def test_celery(word: str) -> str:
 
 
 @celery_app.task(acks_late=True)
-def make_video(video_id:int, zip_filename:str, user_id:int) -> str:
+def make_video(args:list) -> str:
+    video_id, zip_filename, user_id = args
     # download 
     r = requests.get(download_to_local_url, stream=True)
     with open(str(VIDEO_STORAGE/zip_filename), 'wb') as f:
         r.raise_for_status()
         for chunk in r.iter_content(chunk_size=1024):
             f.write(chunk)
-
     # make video