add KeyError (#17)

home-assistant-ecosystem · Jan 3, 2024 · e3f4b6d · e3f4b6d
1 parent f336532
commit e3f4b6d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/CHANGES.txt b/CHANGES.txt
@@ -1,6 +1,11 @@
 Changes
 =======
 
+20231213 - 0.4.1
+----------------
+
+- Add KeyError exception (thanks @miaucl)
+
 20231120 - 0.4.0
 ----------------
 

diff --git a/opendata_transport/__init__.py b/opendata_transport/__init__.py
@@ -89,7 +89,7 @@ async def async_get_data(self):
  try:
  for station in data["stations"]:
  self.locations.append(self.get_station(station))
- except (TypeError, IndexError):
+ except (KeyError, TypeError, IndexError):
  raise exceptions.OpendataTransportError()
 
 
@@ -174,7 +174,7 @@ async def __async_get_data(self, station):
  try:
  for journey in data["stationboard"]:
  self.journeys.append(self.get_journey(journey))
- except (TypeError, IndexError):
+ except (KeyError, TypeError, IndexError):
  raise exceptions.OpendataTransportError()
 
  async def async_get_data(self):
@@ -309,5 +309,5 @@ async def async_get_data(self):
  for connection in data["connections"]:
  self.connections[index] = self.get_connection(connection)
  index = index + 1
- except (TypeError, IndexError):
+ except (KeyError, TypeError, IndexError):
  raise exceptions.OpendataTransportError()