Ver Fonte

update api video.py

tomoya há 1 ano atrás
pai
commit
005dfdff37
1 ficheiros alterados com 16 adições e 7 exclusões
  1. 16 7
      backend/app/app/api/api_v1/endpoints/videos.py

+ 16 - 7
backend/app/app/api/api_v1/endpoints/videos.py

@@ -48,11 +48,12 @@ def test(
     *,
     db: Session = Depends(deps.get_db),
     title: str,
-    anchor_id: int,
-    lang_id: int,
+    anchor: str,
+    style: str,
+    lang: str,
     current_user: models.User = Depends(deps.get_current_active_user),
 ) -> Any:
-    video_data = {"title":title, "anchor_id":anchor_id, "lang_id":lang_id}
+    video_data = {"title":title, "anchor":anchor, "lang_id":lang}
     print(video_data)
     filename = crud.video.generate_file_name(db=db, n=20)
     video_create = schemas.VideoCreate(title=title, progress_state="PENDING", stored_filename=filename)
@@ -62,20 +63,24 @@ def test(
     video_data['membership_status'] = current_user.membership_status
     video_data['available_time'] = current_user.available_time
     video_data['video_id'] = video_data['id']
+    video_data['anchor'] = anchor
+    video_data['style'] = style
+    video_data['lang'] = lang
     task = celery_app.send_task("app.worker.make_video_test",  kwargs=video_data)
     print(task)
     return "ok"
 
 @router.post("/", response_model=schemas.Video)
 def upload_plot(
+    background_tasks: BackgroundTasks,
     *,
     db: Session = Depends(deps.get_db),
+    current_user: models.User = Depends(deps.get_current_active_user),
     title: str=Form(...), 
-    anchor_id: int=Form(...),
-    lang_id: int=Form(...),
+    anchor: str=Form(...),
+    style: str=Form(...),
+    lang: str=Form(...),
     upload_file: UploadFile=File(),
-    current_user: models.User = Depends(deps.get_current_active_user),
-    background_tasks: BackgroundTasks,
 ) -> Any:
     """
     Create new video.
@@ -118,6 +123,10 @@ def upload_plot(
     video_data['membership_status'] = current_user.membership_status
     video_data['available_time'] = current_user.available_time
     video_data['video_id'] = video_data['id']
+    video_data['character'] = anchor
+    video_data['anchor'] = anchor
+    video_data['style'] = style
+    video_data['lang'] = lang
     background_tasks.add_task(wait_finish, video_data)
     return  JSONResponse(return_msg, background=background_tasks)