|
|
@@ -103,7 +103,7 @@ punctuation = r"[.,!?;:。 、！？，；：]"
|
|
|
|
|
|
|
|
|
@router.post('/gen-zip')
|
|
|
-def generate_zip(
|
|
|
+async def generate_zip(
|
|
|
*,
|
|
|
background_tasks: BackgroundTasks,
|
|
|
current_user: models.User = Depends(deps.get_current_active_user),
|
|
|
@@ -146,7 +146,7 @@ def generate_zip(
|
|
|
return FileResponse(f'{output_dir}/{dir.name}.zip', media_type="application/zip")
|
|
|
|
|
|
@router.post('/gen-video')
|
|
|
-def generate_video(
|
|
|
+async def generate_video(
|
|
|
*,
|
|
|
background_tasks: BackgroundTasks,
|
|
|
#current_user: models.User = Depends(deps.get_current_active_user),
|
|
|
@@ -161,7 +161,7 @@ def generate_video(
|
|
|
flag=True
|
|
|
if flag:
|
|
|
background_tasks.add_task(wait_finish, model, email, texts, lang)
|
|
|
- return PlainTextResponse("OK", background=background_tasks)
|
|
|
+ return PlainTextResponse("OK")
|
|
|
else:
|
|
|
return HTTPException("No texts.")
|
|
|
|
|
|
@@ -196,13 +196,12 @@ async def wait_finish(model, email, texts, lang):
|
|
|
wb.save(excel_path)
|
|
|
output_dir = '/tmp'
|
|
|
shutil.make_archive(f'{output_dir}/{dir.name}', format='zip', root_dir=td)
|
|
|
- def remove_zip():
|
|
|
- if os.path.exists(f'{output_dir}/{dir.name}.zip'):
|
|
|
- os.remove(f'{output_dir}/{dir.name}.zip')
|
|
|
+ # def remove_zip():
|
|
|
+ # if os.path.exists(f'{output_dir}/{dir.name}.zip'):
|
|
|
+ # os.remove(f'{output_dir}/{dir.name}.zip')
|
|
|
current_user = crud.user.get(db, id=0)
|
|
|
video_create = schemas.VideoCreate(title="guest", progress_state="PENDING", stored_filename=dir.name)
|
|
|
video = crud.video.create_with_owner(db=db, obj_in=video_create, owner_id=current_user.id)
|
|
|
- return_msg = {"video_message":"accepted", "accepted":True}
|
|
|
video_data = jsonable_encoder(video)
|
|
|
video_data['membership_status'] = current_user.membership_status
|
|
|
video_data['available_time'] = current_user.available_time
|
