Small cleanups to august (#119931)

This commit is contained in:
J. Nick Koston 2024-06-18 17:35:55 -05:00 committed by GitHub
parent 98140e0171
commit 39e5e517b0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 21 additions and 21 deletions

View File

@ -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

View File

@ -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]):