From 39e5e517b0b6a6c0c41d8ffa6a790ac63ae227e3 Mon Sep 17 00:00:00 2001 From: "J. Nick Koston" Date: Tue, 18 Jun 2024 17:35:55 -0500 Subject: [PATCH] Small cleanups to august (#119931) --- homeassistant/components/august/camera.py | 6 ++-- homeassistant/components/august/sensor.py | 36 +++++++++++------------ 2 files changed, 21 insertions(+), 21 deletions(-) diff --git a/homeassistant/components/august/camera.py b/homeassistant/components/august/camera.py index 76ccf9fa4dd..01baf7aa51a 100644 --- a/homeassistant/components/august/camera.py +++ b/homeassistant/components/august/camera.py @@ -45,6 +45,9 @@ class AugustCamera(AugustEntityMixin, Camera): _attr_translation_key = "camera" _attr_motion_detection_enabled = True _attr_brand = DEFAULT_NAME + _image_url: str | None = None + _content_token: str | None = None + _image_content: bytes | None = None def __init__( self, data: AugustData, device: Doorbell, session: ClientSession, timeout: int @@ -53,9 +56,6 @@ class AugustCamera(AugustEntityMixin, Camera): super().__init__(data, device) self._timeout = timeout self._session = session - self._image_url = None - self._content_token = None - self._image_content = None self._attr_unique_id = f"{self._device_id:s}_camera" @property diff --git a/homeassistant/components/august/sensor.py b/homeassistant/components/august/sensor.py index 657368e019f..862117319fa 100644 --- a/homeassistant/components/august/sensor.py +++ b/homeassistant/components/august/sensor.py @@ -125,16 +125,15 @@ class AugustOperatorSensor(AugustEntityMixin, RestoreSensor): """Representation of an August lock operation sensor.""" _attr_translation_key = "operator" + _operated_remote: bool | None = None + _operated_keypad: bool | None = None + _operated_manual: bool | None = None + _operated_tag: bool | None = None + _operated_autorelock: bool | None = None def __init__(self, data: AugustData, device) -> None: """Initialize the sensor.""" super().__init__(data, device) - self._operated_remote: bool | None = None - self._operated_keypad: bool | None = None - self._operated_manual: bool | None = None - self._operated_tag: bool | None = None - self._operated_autorelock: bool | None = None - self._operated_time = None self._attr_unique_id = f"{self._device_id}_lock_operator" self._update_from_data() @@ -201,18 +200,19 @@ class AugustOperatorSensor(AugustEntityMixin, RestoreSensor): return self._attr_native_value = last_sensor_state.native_value - if ATTR_ENTITY_PICTURE in last_state.attributes: - self._attr_entity_picture = last_state.attributes[ATTR_ENTITY_PICTURE] - if ATTR_OPERATION_REMOTE in last_state.attributes: - self._operated_remote = last_state.attributes[ATTR_OPERATION_REMOTE] - if ATTR_OPERATION_KEYPAD in last_state.attributes: - self._operated_keypad = last_state.attributes[ATTR_OPERATION_KEYPAD] - if ATTR_OPERATION_MANUAL in last_state.attributes: - self._operated_manual = last_state.attributes[ATTR_OPERATION_MANUAL] - if ATTR_OPERATION_TAG in last_state.attributes: - self._operated_tag = last_state.attributes[ATTR_OPERATION_TAG] - if ATTR_OPERATION_AUTORELOCK in last_state.attributes: - self._operated_autorelock = last_state.attributes[ATTR_OPERATION_AUTORELOCK] + last_attrs = last_state.attributes + if ATTR_ENTITY_PICTURE in last_attrs: + self._attr_entity_picture = last_attrs[ATTR_ENTITY_PICTURE] + if ATTR_OPERATION_REMOTE in last_attrs: + self._operated_remote = last_attrs[ATTR_OPERATION_REMOTE] + if ATTR_OPERATION_KEYPAD in last_attrs: + self._operated_keypad = last_attrs[ATTR_OPERATION_KEYPAD] + if ATTR_OPERATION_MANUAL in last_attrs: + self._operated_manual = last_attrs[ATTR_OPERATION_MANUAL] + if ATTR_OPERATION_TAG in last_attrs: + self._operated_tag = last_attrs[ATTR_OPERATION_TAG] + if ATTR_OPERATION_AUTORELOCK in last_attrs: + self._operated_autorelock = last_attrs[ATTR_OPERATION_AUTORELOCK] class AugustBatterySensor(AugustDescriptionEntity, SensorEntity, Generic[_T]):