ouyangbo/hermes-agent
1
0
Fork 0
Code Issues Pull Requests Actions 10 Packages Projects Releases Wiki Activity

refactor(homeassistant): shared _close for ws/session teardown

Browse Source
This commit is contained in:
Teknium
2026-09-02 22:21:45 -07:00
parent 843adc5520
commit a6f1d3cd0a
1 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
plugins/platforms/homeassistant/adapter.py
+8 -6
View File
@@ -157,12 +157,15 @@ class HomeAssistantAdapter(BasePlatformAdapter):
return False
return True
@staticmethod
async def _close(obj) -> None:
if obj and not obj.closed:
await obj.close()
async def _cleanup_ws(self) -> None:
if self._ws and not self._ws.closed:
await self._ws.close()
await self._close(self._ws)
self._ws = None
if self._session and not self._session.closed:
await self._session.close()
await self._close(self._session)
self._session = None
async def disconnect(self) -> None:
@@ -175,8 +178,7 @@ class HomeAssistantAdapter(BasePlatformAdapter):
pass
self._listen_task = None
await self._cleanup_ws()
if self._rest_session and not self._rest_session.closed:
await self._rest_session.close()
await self._close(self._rest_session)
self._rest_session = None
logger.info("[%s] Disconnected", self.name)