Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove unnecessary use of dunder methods from entity registry (#77716)

pull/77734/head
Erik Montnemery 2022-09-02 19:02:14 +02:00 committed by GitHub
parent 2e34814d7a
commit 3a86209dec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/helpers/entity_registry.py
View File

@ -185,7 +185,7 @@ class EntityRegistryItems(UserDict[str, "RegistryEntry"]):
Maintains two additional indexes:
- id -> entry
- (domain, platform, unique_id) -> entry
- (domain, platform, unique_id) -> entity_id
"""
def __init__(self) -> None:
@ -201,14 +201,14 @@ class EntityRegistryItems(UserDict[str, "RegistryEntry"]):
del self._entry_ids[old_entry.id]
del self._index[(old_entry.domain, old_entry.platform, old_entry.unique_id)]
super().__setitem__(key, entry)
self._entry_ids.__setitem__(entry.id, entry)
self._entry_ids[entry.id] = entry
self._index[(entry.domain, entry.platform, entry.unique_id)] = entry.entity_id
def __delitem__(self, key: str) -> None:
"""Remove an item."""
entry = self[key]
self._entry_ids.__delitem__(entry.id)
self._index.__delitem__((entry.domain, entry.platform, entry.unique_id))
del self._entry_ids[entry.id]
del self._index[(entry.domain, entry.platform, entry.unique_id)]
super().__delitem__(key)
def get_entity_id(self, key: tuple[str, str, str]) -> str | None: