core/homeassistant/components/doorbird/entity.py

39 lines
1.4 KiB
Python

"""The DoorBird integration base entity."""
from homeassistant.helpers import device_registry as dr
from homeassistant.helpers.device_registry import DeviceInfo
from homeassistant.helpers.entity import Entity
from .const import (
DOORBIRD_INFO_KEY_BUILD_NUMBER,
DOORBIRD_INFO_KEY_DEVICE_TYPE,
DOORBIRD_INFO_KEY_FIRMWARE,
MANUFACTURER,
)
from .models import DoorBirdData
from .util import get_mac_address_from_door_station_info
class DoorBirdEntity(Entity):
"""Base class for doorbird entities."""
_attr_has_entity_name = True
def __init__(self, door_bird_data: DoorBirdData) -> None:
"""Initialize the entity."""
super().__init__()
self._door_bird_data = door_bird_data
self._door_station = door_bird_data.door_station
door_station_info = door_bird_data.door_station_info
self._mac_addr = get_mac_address_from_door_station_info(door_station_info)
firmware = door_station_info[DOORBIRD_INFO_KEY_FIRMWARE]
firmware_build = door_station_info[DOORBIRD_INFO_KEY_BUILD_NUMBER]
self._attr_device_info = DeviceInfo(
configuration_url="https://webadmin.doorbird.com/",
connections={(dr.CONNECTION_NETWORK_MAC, self._mac_addr)},
manufacturer=MANUFACTURER,
model=door_station_info[DOORBIRD_INFO_KEY_DEVICE_TYPE],
name=self._door_station.name,
sw_version=f"{firmware} {firmware_build}",
)