use new library, remove options flow

.tox/py310/.tox-info.json

@@ -0,0 +1,6 @@
+{
+  "ToxEnv": {
+    "name": "py310",
+    "type": "VirtualEnvRunner"
+  }
+}

.tox/py311/.tox-info.json

@@ -0,0 +1,6 @@
+{
+  "ToxEnv": {
+    "name": "py311",
+    "type": "VirtualEnvRunner"
+  }
+}

custom_components/openei/__init__.py

@@ -100,9 +100,11 @@ async def _async_update_data(self) -> dict:
             _LOGGER.debug("Next update in %s seconds.", wait_seconds)
             async_call_later(self.hass, wait_seconds, self._async_refresh_data)
             try:
-                self._data = await self.hass.async_add_executor_job(
-                    self.get_sensors, self.hass, self._config
-                )
+                await self.get_sensors()
+            except openeihttp.RateLimit:
+                pass
+            except AssertionError:
+                pass
             except Exception as exception:
                 raise UpdateFailed() from exception
         return self._data
@@ -118,13 +120,15 @@ async def _async_refresh_data(self, data=None) -> None:
         _LOGGER.debug("Next update in %s seconds.", wait_seconds)
         async_call_later(self.hass, wait_seconds, self._async_refresh_data)
         try:
-            self._data = await self.hass.async_add_executor_job(
-                self.get_sensors, self.hass, self._config
-            )
+            await self.get_sensors()
+        except openeihttp.RateLimit:
+            pass           
+        except AssertionError:
+            pass        
         except Exception as exception:
             raise UpdateFailed() from exception
 
-    def get_sensors(self) -> dict:
+    async def get_sensors(self) -> dict:
         """Update sensor data."""
         api = self._config.data.get(CONF_API_KEY)
         plan = self._config.data.get(CONF_PLAN)
@@ -150,7 +154,7 @@ def get_sensors(self) -> dict:
         )
         if self._rate_limit_count == 0:
             try:
-                rate.update()
+                await rate.update()
             except openeihttp.RateLimit:
                 _LOGGER.error("API Rate limit exceded, retrying later.")
                 if self._data == {}:
@@ -180,7 +184,7 @@ def get_sensors(self) -> dict:
             data.update(_sensor)
 
         _LOGGER.debug("DEBUG: %s", data)
-        return data
+        self._data = data
 
 
 async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:

custom_components/openei/config_flow.py

@@ -68,11 +68,11 @@ async def async_step_user_3(self, user_input=None):
 
         return await self._show_config_form_3(user_input)
 
-    @staticmethod
-    @callback
-    def async_get_options_flow(config_entry):
-        """Enable option flow."""
-        return OpenEIOptionsFlowHandler(config_entry)
+    # @staticmethod
+    # @callback
+    # def async_get_options_flow(config_entry):
+    #     """Enable option flow."""
+    #     return OpenEIOptionsFlowHandler(config_entry)
 
     async def _show_config_form(self, user_input):  # pylint: disable=unused-argument
         """Show the configuration form to edit location data."""
@@ -104,76 +104,76 @@ async def _show_config_form_3(self, user_input):  # pylint: disable=unused-argum
         )
 
 
-class OpenEIOptionsFlowHandler(config_entries.OptionsFlow):
-    """Blueprint config flow options handler."""
-
-    def __init__(self, config_entry):
-        """Initialize OpenEI options flow."""
-        self.config_entry = config_entry
-        self._data = dict(config_entry.data)
-        self._errors = {}
-
-    async def async_step_init(self, user_input=None):  # pylint: disable=unused-argument
-        """Manage the options."""
-        return await self.async_step_user()
-
-    async def async_step_user(self, user_input=None):
-        """Handle a flow initialized by the user."""
-        if user_input is not None:
-            if user_input[CONF_LOCATION] == '""':
-                user_input[CONF_LOCATION] = ""
-            self._data.update(user_input)
-            return await self.async_step_user_2()
-
-        return await self._show_config_form(user_input)
-
-    async def async_step_user_2(self, user_input=None):
-        """Handle a flow initialized by the user."""
-        _LOGGER.debug("data: %s", self._data)
-        if user_input is not None:
-            self._data.update(user_input)
-            return await self.async_step_user_3()
-
-        return await self._show_config_form_2(user_input)
-
-    async def async_step_user_3(self, user_input=None):
-        """Handle a flow initialized by the user."""
-        _LOGGER.debug("data: %s", self._data)
-        if user_input is not None:
-            if user_input[CONF_MANUAL_PLAN] == '""':
-                user_input[CONF_MANUAL_PLAN] = ""
-            self._data.update(user_input)
-            return self.async_create_entry(title="", data=self._data)
-
-        return await self._show_config_form_3(user_input)
-
-    async def _show_config_form(self, user_input: Optional[Dict[str, Any]]):
-        """Show the configuration form to edit location data."""
-        return self.async_show_form(
-            step_id="user",
-            data_schema=_get_schema_step_1(user_input, self._data),
-            errors=self._errors,
-        )
-
-    async def _show_config_form_2(self, user_input: Optional[Dict[str, Any]]):
-        """Show the configuration form to edit location data."""
-        utility_list = await _get_utility_list(self.hass, self._data)
-        return self.async_show_form(
-            step_id="user_2",
-            data_schema=_get_schema_step_2(user_input, self._data, utility_list),
-            errors=self._errors,
-        )
-
-    async def _show_config_form_3(self, user_input: Optional[Dict[str, Any]]):
-        """Show the configuration form to edit location data."""
-        plan_list = await _get_plan_list(self.hass, self._data)
-        return self.async_show_form(
-            step_id="user_3",
-            data_schema=_get_schema_step_3(
-                self.hass, user_input, self._data, plan_list
-            ),
-            errors=self._errors,
-        )
+# class OpenEIOptionsFlowHandler(config_entries.OptionsFlow):
+#     """Blueprint config flow options handler."""
+
+#     def __init__(self, config_entry):
+#         """Initialize OpenEI options flow."""
+#         self.config_entry = config_entry
+#         self._data = dict(config_entry.data)
+#         self._errors = {}
+
+#     async def async_step_init(self, user_input=None):  # pylint: disable=unused-argument
+#         """Manage the options."""
+#         return await self.async_step_user()
+
+#     async def async_step_user(self, user_input=None):
+#         """Handle a flow initialized by the user."""
+#         if user_input is not None:
+#             if user_input[CONF_LOCATION] == '""':
+#                 user_input[CONF_LOCATION] = ""
+#             self._data.update(user_input)
+#             return await self.async_step_user_2()
+
+#         return await self._show_config_form(user_input)
+
+#     async def async_step_user_2(self, user_input=None):
+#         """Handle a flow initialized by the user."""
+#         _LOGGER.debug("data: %s", self._data)
+#         if user_input is not None:
+#             self._data.update(user_input)
+#             return await self.async_step_user_3()
+
+#         return await self._show_config_form_2(user_input)
+
+#     async def async_step_user_3(self, user_input=None):
+#         """Handle a flow initialized by the user."""
+#         _LOGGER.debug("data: %s", self._data)
+#         if user_input is not None:
+#             if user_input[CONF_MANUAL_PLAN] == '""':
+#                 user_input[CONF_MANUAL_PLAN] = ""
+#             self._data.update(user_input)
+#             return self.async_create_entry(title="", data=self._data)
+
+#         return await self._show_config_form_3(user_input)
+
+#     async def _show_config_form(self, user_input: Optional[Dict[str, Any]]):
+#         """Show the configuration form to edit location data."""
+#         return self.async_show_form(
+#             step_id="user",
+#             data_schema=_get_schema_step_1(user_input, self._data),
+#             errors=self._errors,
+#         )
+
+#     async def _show_config_form_2(self, user_input: Optional[Dict[str, Any]]):
+#         """Show the configuration form to edit location data."""
+#         utility_list = await _get_utility_list(self.hass, self._data)
+#         return self.async_show_form(
+#             step_id="user_2",
+#             data_schema=_get_schema_step_2(user_input, self._data, utility_list),
+#             errors=self._errors,
+#         )
+
+#     async def _show_config_form_3(self, user_input: Optional[Dict[str, Any]]):
+#         """Show the configuration form to edit location data."""
+#         plan_list = await _get_plan_list(self.hass, self._data)
+#         return self.async_show_form(
+#             step_id="user_3",
+#             data_schema=_get_schema_step_3(
+#                 self.hass, user_input, self._data, plan_list
+#             ),
+#             errors=self._errors,
+#         )
 
 
 def _get_schema_step_1(
@@ -271,7 +271,7 @@ async def _get_utility_list(hass, user_input) -> list | None:
         address = None
 
     plans = openeihttp.Rates(api=api, lat=lat, lon=lon, radius=radius, address=address)
-    plans = await hass.async_add_executor_job(_lookup_plans, plans)
+    plans = await _lookup_plans(plans)
     utilities = []
 
     for utility in plans:
@@ -296,7 +296,7 @@ async def _get_plan_list(hass, user_input) -> list | None:
         address = None
 
     plans = openeihttp.Rates(api=api, lat=lat, lon=lon, radius=radius, address=address)
-    plans = await hass.async_add_executor_job(_lookup_plans, plans)
+    plans = await _lookup_plans(plans)
     value = {}
 
     for plan in plans[utility]:
@@ -306,9 +306,9 @@ async def _get_plan_list(hass, user_input) -> list | None:
     return value
 
 
-def _lookup_plans(handler) -> list:
+async def _lookup_plans(handler) -> list:
     """Return list of utilities and plans."""
-    response = handler.lookup_plans()
+    response = await handler.lookup_plans()
     _LOGGER.debug("lookup_plans: %s", response)
     return response
 

custom_components/openei/manifest.json

@@ -7,6 +7,6 @@
   "documentation": "https://github.com/firstof9/ha-openei",
   "iot_class": "cloud_polling",
   "issue_tracker": "https://github.com/firstof9/ha-openei/issues",
-  "requirements": ["python-openei==0.1.24"],
+  "requirements": ["python-openei==0.2.0"],
   "version": "0.1.6"
 }

requirements_tests.txt

@@ -1,5 +1,5 @@
 -r requirements_dev.txt
-python-openei==0.1.24
+python-openei==0.2.0
 pytest
 pytest-cov
 pytest-homeassistant-custom-component
@@ -10,3 +10,4 @@ mypy
 pydocstyle
 isort
 pylint
+aioresponses

setup.cfg

@@ -1,5 +1,5 @@
 [mypy]
-python_version = 3.12
+python_version = 3.13
 show_error_codes = true
 ignore_errors = true
 follow_imports = silent