fix: check if the temp dir exists before deleting it (#2141)

This commit is contained in:
Michael Genson 2023-02-19 18:46:17 -06:00 committed by GitHub
parent 05e2566c35
commit d639bdcfe9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -111,7 +111,9 @@ class AdminMaintenanceController(BaseAdminController):
@router.post("/clean/temp", response_model=SuccessResponse)
def clean_temp(self):
try:
shutil.rmtree(self.folders.TEMP_DIR)
if self.folders.TEMP_DIR.exists():
shutil.rmtree(self.folders.TEMP_DIR)
self.folders.TEMP_DIR.mkdir(parents=True, exist_ok=True)
except Exception as e:
raise HTTPException(status_code=500, detail=ErrorResponse.respond("Failed to clean temp")) from e