Skip to content

Commit 75ebdbb

Browse files
committed
Implement try-except for unregistering node
1 parent 89dc0b3 commit 75ebdbb

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

plugwise_usb/network/__init__.py

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -159,9 +159,12 @@ async def clear_cache(self) -> None:
159159

160160
async def unregister_node(self, mac: str) -> None:
161161
"""Unregister node from current Plugwise network."""
162-
await self._register.unregister_node(mac)
163-
await self._nodes[mac].unload()
164-
self._nodes.pop(mac)
162+
try:
163+
await self._register.unregister_node(mac)
164+
await self._nodes[mac].unload()
165+
self._nodes.pop(mac)
166+
except KeyError as exc:
167+
raise MessageError("Mac not registered, already deleted?")
165168

166169
# region - Handle stick connect/disconnect events
167170
def _subscribe_to_protocol_events(self) -> None:

0 commit comments

Comments
 (0)