Upgrade to required Hass version 2024.1 (#80)

.circleci/config.yml

@@ -18,7 +18,7 @@ commands:
 jobs:
   lint:
     docker:
-      - image: cimg/python:3.9
+      - image: cimg/python:3.11
     resource_class: small
     steps:
       - checkout
@@ -27,7 +27,7 @@ jobs:
       - run: find . -name "*.json" -type f -print0 | xargs -0I {} python3 -m json.tool {}
   test:
     docker:
-      - image: cimg/python:3.9
+      - image: cimg/python:3.11
     resource_class: small
     steps:
       - checkout

custom_components/purei9/coordinator.py

@@ -1,7 +1,7 @@
 """Coordinate data updates from Pure i9."""
 import logging
 from datetime import timedelta
-import async_timeout
+from asyncio import timeout
 from homeassistant.helpers.update_coordinator import (
     DataUpdateCoordinator,
     UpdateFailed,
@@ -31,7 +31,7 @@ def robot(self) -> str:
     async def _async_update_data(self):
         """Fetch data from Pure i9."""
         try:
-            async with async_timeout.timeout(10):
+            async with timeout(10):
                 return await self.hass.async_add_executor_job(
                     self.update_and_create_params
                 )

custom_components/purei9/manifest.json

@@ -8,5 +8,6 @@
     "codeowners": ["Ekman"],
     "requirements": ["purei9_unofficial==0.0.14"],
     "iot_class": "cloud_polling",
-    "config_flow": true
+    "config_flow": true,
+    "homeassistant": "2024.1.0"
 }

custom_components/purei9/vacuum.py

@@ -20,15 +20,16 @@
     STATE_IDLE
 )
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
-from homeassistant.const import CONF_PASSWORD, CONF_EMAIL
+from homeassistant.const import CONF_PASSWORD, CONF_EMAIL, CONF_COUNTRY_CODE
 from purei9_unofficial.cloudv3 import CloudRobot
 from . import purei9, const, vacuum_command, exception, utility
 
 _LOGGER = logging.getLogger(__name__)
 
 PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend({
     vol.Required(CONF_EMAIL): str,
-    vol.Required(CONF_PASSWORD): str
+    vol.Required(CONF_PASSWORD): str,
+    vol.Required(CONF_COUNTRY_CODE): str
 })
 
 async def async_setup_entry(hass, config_entry, async_add_entities):
@@ -39,6 +40,7 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
         [PureI9(coord, coord.robot, coord.data) for coord in data[const.COORDINATORS]]
     )
 
+# pylint: disable=R0904
 class PureI9(CoordinatorEntity, StateVacuumEntity):
     """The main Pure i9 vacuum entity"""
     def __init__(
@@ -149,18 +151,27 @@ async def async_start(self):
             self._params.state = STATE_CLEANING
             self.async_write_ha_state()
 
+    def start(self):
+        raise NotImplementedError
+
     async def async_return_to_base(self, **kwargs):
         """Return to the dock"""
         await self.hass.async_add_executor_job(self._robot.gohome)
         self._params.state = STATE_RETURNING
         self.async_write_ha_state()
 
+    def return_to_base(self, **kwargs):
+        raise NotImplementedError
+
     async def async_stop(self, **kwargs):
         """Stop cleaning"""
         await self.hass.async_add_executor_job(self._robot.stopclean)
         self._params.state = STATE_IDLE
         self.async_write_ha_state()
 
+    def stop(self, **kwargs):
+        raise NotImplementedError
+
     async def async_pause(self):
         """Pause cleaning"""
         # According to Home Assistant, pause should be an idempotent
@@ -171,6 +182,9 @@ async def async_pause(self):
             self._params.state = STATE_PAUSED
             self.async_write_ha_state()
 
+    def pause(self):
+        raise NotImplementedError
+
     async def async_set_fan_speed(self, fan_speed: str, **kwargs: Any):
         """Set the fan speed of the robot"""
         await self.hass.async_add_executor_job(
@@ -180,6 +194,9 @@ async def async_set_fan_speed(self, fan_speed: str, **kwargs: Any):
         self._params.fan_speed = fan_speed
         self.async_write_ha_state()
 
+    def set_fan_speed(self, fan_speed, **kwargs):
+        raise NotImplementedError
+
     async def async_send_command(
         self,
         command: str,
@@ -207,6 +224,14 @@ async def async_send_command(
         except exception.CommandException as ex:
             _LOGGER.error("Could not execute command \"%s\" due to: %s", command, ex)
 
+    def send_command(
+        self,
+        command: str,
+        params: dict[str, Any] | list[Any] | None = None,
+        **kwargs: Any,
+    ):
+        raise NotImplementedError
+
     def _handle_coordinator_update(self):
         """
         Called by Home Assistant asking the vacuum to update to the latest state.
@@ -225,3 +250,9 @@ def _handle_coordinator_update(self):
         self._params.maps = params.maps
 
         self.async_write_ha_state()
+
+    def locate(self, **kwargs):
+        raise NotImplementedError
+
+    def clean_spot(self, **kwargs):
+        raise NotImplementedError

requirements.txt

@@ -1,3 +1,3 @@
 purei9_unofficial==0.0.14
-homeassistant==2021.11.5
+homeassistant==2024.1.0
 pylint==3.3.1