Add executor calls

hardbyte · Jun 9, 2024 · af967c0 · af967c0
1 parent 15ef13b
commit af967c0
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/custom_components/evnex/__init__.py b/custom_components/evnex/__init__.py
@@ -101,7 +101,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     password = entry.data[CONF_PASSWORD]
 
     # Load tokens from storage
-    evnex_auth_tokens = retrieve_evnex_auth_tokens(hass, entry)
+    evnex_auth_tokens = await hass.async_add_executor_job(retrieve_evnex_auth_tokens, hass, entry)
     evnex_auth_tokens = {} if evnex_auth_tokens is None else evnex_auth_tokens
 
     httpx_client = get_async_client(hass)
@@ -147,7 +147,8 @@ async def async_update_data(is_retry: bool = False):
 
             account: EvnexUserDetail = await evnex_client.get_user_detail()
 
-            persist_evnex_auth_tokens(
+            await hass.async_add_executor_job(
+                persist_evnex_auth_tokens,
                 hass,
                 entry,
                 evnex_client.id_token,
@@ -226,7 +227,8 @@ async def async_update_data(is_retry: bool = False):
             if not is_retry:
                 _LOGGER.debug("Refreshing auth and trying again")
                 await hass.async_add_executor_job(evnex_client.authenticate)
-                persist_evnex_auth_tokens(
+                await hass.async_add_executor_job(
+                    persist_evnex_auth_tokens,
                     hass,
                     entry,
                     evnex_client.id_token,