Poll Nissan servers for battery updates

homeassistant/components/nissan_leaf/__init__.py

@@ -46,7 +46,7 @@
 RESTRICTED_BATTERY = 2
 RESTRICTED_INTERVAL = timedelta(hours=12)
 
-MAX_RESPONSE_ATTEMPTS = 10
+MAX_RESPONSE_ATTEMPTS = 3
 
 PYCARWINGS2_SLEEP = 30
 
@@ -195,6 +195,14 @@ def setup_leaf(car_config):
     return True
 
 
+def _extract_start_date(battery_info):
+    """Extract the server date from the battery response."""
+    try:
+        return battery_info.answer["BatteryStatusRecords"]["OperationDateAndTime"]
+    except KeyError:
+        return None
+
+
 class LeafDataStore:
     """Nissan Leaf Data Store."""
 
@@ -326,20 +334,25 @@ async def async_refresh_data(self, now):
         self.request_in_progress = False
         async_dispatcher_send(self.hass, SIGNAL_UPDATE_LEAF)
 
-    @staticmethod
-    def _extract_start_date(battery_info):
-        """Extract the server date from the battery response."""
-        try:
-            return battery_info.answer["BatteryStatusRecords"]["OperationDateAndTime"]
-        except KeyError:
-            return None
-
     async def async_get_battery(self):
         """Request battery update from Nissan servers."""
 
         try:
             # Request battery update from the car
             _LOGGER.debug("Requesting battery update, %s", self.leaf.vin)
+            start_date = None
+            try:
+                start_server_info = await self.hass.async_add_executor_job(
+                    self.leaf.get_latest_battery_status
+                )
+            except TypeError:  # pycarwings2 can fail if Nissan returns nothing
+                _LOGGER.debug("Battery status check returned nothing")
+            else:
+                if not start_server_info:
+                    _LOGGER.debug("Battery status check failed")
+                else:
+                    start_date = _extract_start_date(start_server_info)
+            await asyncio.sleep(1)  # Critical sleep
             request = await self.hass.async_add_executor_job(self.leaf.request_update)
             if not request:
                 _LOGGER.error("Battery update request failed")
@@ -367,7 +380,19 @@ async def async_get_battery(self):
                     server_info = await self.hass.async_add_executor_job(
                         self.leaf.get_latest_battery_status
                     )
-                    return server_info
+                    if not start_date or (
+                        server_info and start_date != _extract_start_date(server_info)
+                    ):
+                        return server_info
+                    # get_status_from_update returned {"resultFlag": "1"}
+                    # but the data didn't change, make a fresh request.
+                    await asyncio.sleep(1)  # Critical sleep
+                    request = await self.hass.async_add_executor_job(
+                        self.leaf.request_update
+                    )
+                    if not request:
+                        _LOGGER.error("Battery update request failed")
+                        return None
 
             _LOGGER.debug(
                 "%s attempts exceeded return latest data from server",
@@ -382,7 +407,7 @@ async def async_get_battery(self):
         except CarwingsError:
             _LOGGER.error("An error occurred getting battery status")
             return None
-        except KeyError:
+        except (KeyError, TypeError):
             _LOGGER.error("An error occurred parsing response from server")
             return None