@@ -76,27 +76,29 @@ def get_upload_speed(coordinator: QBittorrentDataCoordinator) -> int:
7676
7777
7878def get_download_speed_limit (coordinator : QBittorrentDataCoordinator ) -> int :
79- """Get current download speed."""
79+ """Get current download speed limit ."""
8080 server_state = cast (Mapping , coordinator .data .get ("server_state" ))
8181 return cast (int , server_state .get ("dl_rate_limit" ))
8282
8383
8484def get_upload_speed_limit (coordinator : QBittorrentDataCoordinator ) -> int :
85- """Get current upload speed."""
85+ """Get current upload speed limit ."""
8686 server_state = cast (Mapping [str , Any ], coordinator .data .get ("server_state" ))
8787 return cast (int , server_state .get ("up_rate_limit" ))
8888
8989
90- def get_alltime_download (coordinator : QBittorrentDataCoordinator ) -> int :
91- """Get current download speed ."""
90+ def get_alltime_download (coordinator : QBittorrentDataCoordinator ) -> int | None :
91+ """Get all-time download volume ."""
9292 server_state = cast (Mapping , coordinator .data .get ("server_state" ))
93- return cast (int , server_state .get ("alltime_dl" ))
93+ value = cast (int , server_state .get ("alltime_dl" ))
94+ return value or None
9495
9596
96- def get_alltime_upload (coordinator : QBittorrentDataCoordinator ) -> int :
97- """Get current download speed ."""
97+ def get_alltime_upload (coordinator : QBittorrentDataCoordinator ) -> int | None :
98+ """Get all-time upload volume ."""
9899 server_state = cast (Mapping , coordinator .data .get ("server_state" ))
99- return cast (int , server_state .get ("alltime_ul" ))
100+ value = cast (int , server_state .get ("alltime_ul" ))
101+ return value or None
100102
101103
102104def get_global_ratio (coordinator : QBittorrentDataCoordinator ) -> float :
0 commit comments