Replace dict with DeviceInfo (#49950)

* Replace dict with DeviceInfo

* Clean up

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
Milan Meulemans 2021-05-02 00:37:19 +02:00 committed by GitHub
parent ddd7e79ee9
commit 29d72714f3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
70 changed files with 159 additions and 138 deletions

View file

@ -1,5 +1,6 @@
"""Base class for UniFi clients."""
from homeassistant.helpers.device_registry import CONNECTION_NETWORK_MAC
from homeassistant.helpers.entity import DeviceInfo
from .unifi_entity_base import UniFiBase
@ -44,7 +45,7 @@ class UniFiClient(UniFiBase):
return self.controller.available
@property
def device_info(self) -> dict:
def device_info(self) -> DeviceInfo:
"""Return a client description for device registry."""
return {
"connections": {(CONNECTION_NETWORK_MAC, self.client.mac)},