|
@@ -80,6 +80,10 @@ async def wait_finish(dirname, filenames):
|
|
for filename in filenames:
|
|
for filename in filenames:
|
|
await publish(filename)
|
|
await publish(filename)
|
|
|
|
|
|
|
|
+ process = await asyncio.create_subprocess_exec("sshpass", "-p", "choozmo9",
|
|
|
|
+ "ssh", "-P", "5722", "-o", "StrictHostKeyChecking=no", "root@172.104.93.163", "rm", "-r", f"{str(LOCAL_ZIP_STORAGE/dirname)}")
|
|
|
|
+ await process.wait()
|
|
|
|
+
|
|
|
|
|
|
async def publish(data):
|
|
async def publish(data):
|
|
for sr_client in sr_clients.values():
|
|
for sr_client in sr_clients.values():
|
|
@@ -92,6 +96,7 @@ def get_image(
|
|
current_user: models.User = Depends(deps.get_current_active_user),
|
|
current_user: models.User = Depends(deps.get_current_active_user),
|
|
stored_file_name: str,
|
|
stored_file_name: str,
|
|
file_name: str,
|
|
file_name: str,
|
|
|
|
+ background_tasks: BackgroundTasks
|
|
) -> Any:
|
|
) -> Any:
|
|
"""
|
|
"""
|
|
Download image
|
|
Download image
|
|
@@ -99,10 +104,17 @@ def get_image(
|
|
return_files = list(BACKEND_ZIP_STORAGE.glob(stored_file_name))
|
|
return_files = list(BACKEND_ZIP_STORAGE.glob(stored_file_name))
|
|
if return_files:
|
|
if return_files:
|
|
return_file = return_files[0]
|
|
return_file = return_files[0]
|
|
- return FileResponse(path=str(return_file), filename=file_name+"_hr"+return_file.suffix)
|
|
|
|
|
|
+ background_tasks.add_task(delete_img, return_file)
|
|
|
|
+ return FileResponse(path=str(return_file), filename=file_name+"_hr"+return_file.suffix, background=background_tasks)
|
|
else:
|
|
else:
|
|
print("non")
|
|
print("non")
|
|
-
|
|
|
|
|
|
+
|
|
|
|
+async def delete_img(filename:str):
|
|
|
|
+ await asyncio.sleep(1)
|
|
|
|
+ path = Path(filename)
|
|
|
|
+ if path.exists():
|
|
|
|
+ path.unlink()
|
|
|
|
+
|
|
|
|
|
|
@router.delete("/sr")
|
|
@router.delete("/sr")
|
|
def del_image():
|
|
def del_image():
|