jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-28 07:37:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy issues in zha store (#74032)

Browse Source
This commit is contained in:
epenet 2022-06-28 08:52:20 +02:00 committed by GitHub
parent cc8170fcfe
commit 87b46a699a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 7 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
homeassistant/components/zha/core/store.py
View File

@ -5,7 +5,7 @@ from collections import OrderedDict
from collections.abc import MutableMapping from collections.abc import MutableMapping
import datetime import datetime
import time import time
from typing import TYPE_CHECKING, cast from typing import TYPE_CHECKING, Any, cast
import attr import attr
@ -45,10 +45,11 @@ class ZhaStorage:
@callback @callback
def async_create_device(self, device: ZHADevice) -> ZhaDeviceEntry: def async_create_device(self, device: ZHADevice) -> ZhaDeviceEntry:
"""Create a new ZhaDeviceEntry.""" """Create a new ZhaDeviceEntry."""
ieee_str: str = str(device.ieee)
device_entry: ZhaDeviceEntry = ZhaDeviceEntry( device_entry: ZhaDeviceEntry = ZhaDeviceEntry(
name=device.name, ieee=str(device.ieee), last_seen=device.last_seen name=device.name, ieee=ieee_str, last_seen=device.last_seen
) )
self.devices[device_entry.ieee] = device_entry self.devices[ieee_str] = device_entry
self.async_schedule_save() self.async_schedule_save()
return device_entry return device_entry
@ -81,8 +82,8 @@ class ZhaStorage:
ieee_str: str = str(device.ieee) ieee_str: str = str(device.ieee)
old = self.devices[ieee_str] old = self.devices[ieee_str]
if old is not None and device.last_seen is None: if device.last_seen is None:
return return old
changes = {} changes = {}
changes["last_seen"] = device.last_seen changes["last_seen"] = device.last_seen
@ -93,7 +94,7 @@ class ZhaStorage:
async def async_load(self) -> None: async def async_load(self) -> None:
"""Load the registry of zha device entries.""" """Load the registry of zha device entries."""
data = await self._store.async_load() data = cast(dict[str, Any], await self._store.async_load())
devices: OrderedDict[str, ZhaDeviceEntry] = OrderedDict() devices: OrderedDict[str, ZhaDeviceEntry] = OrderedDict()

3
mypy.ini
View File

@ -3001,6 +3001,3 @@ ignore_errors = true
[mypy-homeassistant.components.zha.core.registries] [mypy-homeassistant.components.zha.core.registries]
ignore_errors = true ignore_errors = true
[mypy-homeassistant.components.zha.core.store]
ignore_errors = true

1
script/hassfest/mypy_config.py
View File

@ -146,7 +146,6 @@ IGNORED_MODULES: Final[list[str]] = [
"homeassistant.components.xiaomi_miio.switch", "homeassistant.components.xiaomi_miio.switch",
"homeassistant.components.zha.core.discovery", "homeassistant.components.zha.core.discovery",
"homeassistant.components.zha.core.registries", "homeassistant.components.zha.core.registries",
"homeassistant.components.zha.core.store",
] ]
# Component modules which should set no_implicit_reexport = true. # Component modules which should set no_implicit_reexport = true.