Simplify qbittorrent sensor class init (#90411)

catch LoginException directly in QBittorrentSensor init

Since the `exception` arg in QBittorrentSensor `__init__` is always
LoginException, we catch LoginException directly in `__init__` instead
of passing LoginException as an argument.
pull/82560/head^2
Chris Xiao 2023-03-28 10:52:16 -04:00 committed by GitHub
parent f081fa8feb
commit 048d30904e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 5 deletions

View File

@ -90,8 +90,7 @@ def setup_platform(
name = config.get(CONF_NAME)
entities = [
QBittorrentSensor(description, client, name, LoginRequired)
for description in SENSOR_TYPES
QBittorrentSensor(description, client, name) for description in SENSOR_TYPES
]
add_entities(entities, True)
@ -111,12 +110,10 @@ class QBittorrentSensor(SensorEntity):
description: SensorEntityDescription,
qbittorrent_client,
client_name,
exception,
) -> None:
"""Initialize the qBittorrent sensor."""
self.entity_description = description
self.client = qbittorrent_client
self._exception = exception
self._attr_name = f"{client_name} {description.name}"
self._attr_available = False
@ -130,7 +127,7 @@ class QBittorrentSensor(SensorEntity):
_LOGGER.error("Connection lost")
self._attr_available = False
return
except self._exception:
except LoginRequired:
_LOGGER.error("Invalid authentication")
return