From f5093b474a010a8059e148417f5a2484e1cff90b Mon Sep 17 00:00:00 2001 From: Otto Winter Date: Sat, 17 Mar 2018 12:27:21 +0100 Subject: [PATCH] Python 3.5 async with (#13283) --- homeassistant/helpers/entity_platform.py | 2 +- homeassistant/helpers/restore_state.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/homeassistant/helpers/entity_platform.py b/homeassistant/helpers/entity_platform.py index 712121bbdb5..501ab5057a3 100644 --- a/homeassistant/helpers/entity_platform.py +++ b/homeassistant/helpers/entity_platform.py @@ -311,7 +311,7 @@ class EntityPlatform(object): self.scan_interval) return - with (await self._process_updates): + async with self._process_updates: tasks = [] for entity in self.entities.values(): if not entity.should_poll: diff --git a/homeassistant/helpers/restore_state.py b/homeassistant/helpers/restore_state.py index aac00b07d7a..eb88a3db369 100644 --- a/homeassistant/helpers/restore_state.py +++ b/homeassistant/helpers/restore_state.py @@ -75,7 +75,7 @@ async def async_get_last_state(hass, entity_id: str): if _LOCK not in hass.data: hass.data[_LOCK] = asyncio.Lock(loop=hass.loop) - with (await hass.data[_LOCK]): + async with hass.data[_LOCK]: if DATA_RESTORE_CACHE not in hass.data: await hass.async_add_job( _load_restore_cache, hass)