jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-19 11:17:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix gtfs with 2023.3 (sqlachemy update) (#89175)

Browse Source
This commit is contained in:
Arjan 2023-03-13 11:57:49 +01:00 committed by Paulus Schoutsen
parent 990ecbba72
commit 6ebd493c4d
1 changed files with 13 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
homeassistant/components/gtfs/sensor.py
View File

@ -342,12 +342,14 @@ def get_next_departure(
origin_stop_time.departure_time origin_stop_time.departure_time
LIMIT :limit LIMIT :limit
""" """
result = schedule.engine.execute( result = schedule.engine.connect().execute(
text(sql_query), text(sql_query),
origin_station_id=start_station_id, {
end_station_id=end_station_id, "origin_station_id": start_station_id,
today=now_date, "end_station_id": end_station_id,
limit=limit, "today": now_date,
"limit": limit,
},
) )
# Create lookup timetable for today and possibly tomorrow, taking into # Create lookup timetable for today and possibly tomorrow, taking into
@ -357,7 +359,8 @@ def get_next_departure(
yesterday_start = today_start = tomorrow_start = None yesterday_start = today_start = tomorrow_start = None
yesterday_last = today_last = "" yesterday_last = today_last = ""
for row in result: for row_cursor in result:
row = row_cursor._asdict()
if row["yesterday"] == 1 and yesterday_date >= row["start_date"]: if row["yesterday"] == 1 and yesterday_date >= row["start_date"]:
extras = {"day": "yesterday", "first": None, "last": False} extras = {"day": "yesterday", "first": None, "last": False}
if yesterday_start is None: if yesterday_start is None:
@ -800,7 +803,10 @@ class GTFSDepartureSensor(SensorEntity):
@staticmethod @staticmethod
def dict_for_table(resource: Any) -> dict: def dict_for_table(resource: Any) -> dict:
"""Return a dictionary for the SQLAlchemy resource given.""" """Return a dictionary for the SQLAlchemy resource given."""
return {col: getattr(resource, col) for col in resource.__table__.columns} _dict = {}
for column in resource.__table__.columns:
_dict[column.name] = str(getattr(resource, column.name))
return _dict
def append_keys(self, resource: dict, prefix: str | None = None) -> None: def append_keys(self, resource: dict, prefix: str | None = None) -> None:
"""Properly format key val pairs to append to attributes.""" """Properly format key val pairs to append to attributes."""