diff --git a/homeassistant/components/qbittorrent/sensor.py b/homeassistant/components/qbittorrent/sensor.py index efdab3122f54..afad29a5b731 100644 --- a/homeassistant/components/qbittorrent/sensor.py +++ b/homeassistant/components/qbittorrent/sensor.py @@ -76,27 +76,29 @@ def get_upload_speed(coordinator: QBittorrentDataCoordinator) -> int: def get_download_speed_limit(coordinator: QBittorrentDataCoordinator) -> int: - """Get current download speed.""" + """Get current download speed limit.""" server_state = cast(Mapping, coordinator.data.get("server_state")) return cast(int, server_state.get("dl_rate_limit")) def get_upload_speed_limit(coordinator: QBittorrentDataCoordinator) -> int: - """Get current upload speed.""" + """Get current upload speed limit.""" server_state = cast(Mapping[str, Any], coordinator.data.get("server_state")) return cast(int, server_state.get("up_rate_limit")) -def get_alltime_download(coordinator: QBittorrentDataCoordinator) -> int: - """Get current download speed.""" +def get_alltime_download(coordinator: QBittorrentDataCoordinator) -> int | None: + """Get all-time download volume.""" server_state = cast(Mapping, coordinator.data.get("server_state")) - return cast(int, server_state.get("alltime_dl")) + value = cast(int, server_state.get("alltime_dl")) + return value or None -def get_alltime_upload(coordinator: QBittorrentDataCoordinator) -> int: - """Get current download speed.""" +def get_alltime_upload(coordinator: QBittorrentDataCoordinator) -> int | None: + """Get all-time upload volume.""" server_state = cast(Mapping, coordinator.data.get("server_state")) - return cast(int, server_state.get("alltime_ul")) + value = cast(int, server_state.get("alltime_ul")) + return value or None def get_global_ratio(coordinator: QBittorrentDataCoordinator) -> float: