Improve no indexes handling

bieniu · Jul 7, 2021 · 51ea797 · 51ea797
1 parent fdbc231
commit 51ea797
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/gios/__init__.py b/gios/__init__.py
@@ -93,14 +93,17 @@ async def async_update(self) -> Dict[str, Any]:  # pylint:disable=too-many-branc
 
         indexes = await self._get_indexes()
 
+        for sensor, sensor_data in data.items():
+            with suppress(IndexError, KeyError, TypeError):
+                index_level = ATTR_INDEX_LEVEL.format(sensor.lower().replace(".", ""))
+                sensor_data[ATTR_INDEX] = indexes[index_level]["indexLevelName"].lower()
+
         with suppress(IndexError, KeyError, TypeError):
             data[ATTR_AQI.lower()] = {ATTR_NAME: ATTR_AQI}
             data[ATTR_AQI.lower()][ATTR_VALUE] = indexes["stIndexLevel"][
                 "indexLevelName"
             ].lower()
-            for sensor, sensor_data in data.items():
-                index_level = ATTR_INDEX_LEVEL.format(sensor.lower().replace(".", ""))
-                sensor_data[ATTR_INDEX] = indexes[index_level]["indexLevelName"].lower()
+
         return data
 
     async def _get_stations(self) -> List[Dict[str, Any]]: