From 3a86209dec8ded9f4539bc661ac42ddf458c6e06 Mon Sep 17 00:00:00 2001
From: Erik Montnemery <erik@montnemery.com>
Date: Fri, 2 Sep 2022 19:02:14 +0200
Subject: [PATCH] Remove unnecessary use of dunder methods from entity registry
(#77716)
---
homeassistant/helpers/entity_registry.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/homeassistant/helpers/entity_registry.py b/homeassistant/helpers/entity_registry.py
index cd1be43690d..701b0816d1c 100644
--- a/homeassistant/helpers/entity_registry.py
+++ b/homeassistant/helpers/entity_registry.py
@@ -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:
--
GitLab