From 9df8d3d36076d991f1f9f9b445f102cd7338adb9 Mon Sep 17 00:00:00 2001 From: jxxghp Date: Fri, 22 Sep 2023 11:20:12 +0800 Subject: [PATCH] fix bug --- app/api/endpoints/dashboard.py | 17 ++++++++++------- 1 file changed, 10 insertions(+), 7 deletions(-) diff --git a/app/api/endpoints/dashboard.py b/app/api/endpoints/dashboard.py index fa1bda7a..1e9d451b 100644 --- a/app/api/endpoints/dashboard.py +++ b/app/api/endpoints/dashboard.py @@ -64,13 +64,16 @@ def downloader(db: Session = Depends(get_db), """ transfer_info = DashboardChain(db).downloader_info() free_space = SystemUtils.free_space(Path(settings.DOWNLOAD_PATH)) - return schemas.DownloaderInfo( - download_speed=transfer_info.download_speed, - upload_speed=transfer_info.upload_speed, - download_size=transfer_info.download_size, - upload_size=transfer_info.upload_size, - free_space=free_space - ) + if transfer_info: + return schemas.DownloaderInfo( + download_speed=transfer_info.download_speed, + upload_speed=transfer_info.upload_speed, + download_size=transfer_info.download_size, + upload_size=transfer_info.upload_size, + free_space=free_space + ) + else: + return schemas.DownloaderInfo() @router.get("/schedule", summary="后台服务", response_model=List[schemas.ScheduleInfo])