nahuel
/
dublinbus-display
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Correct arrival and departure dates (h:mm:ss -> hh:mm:ss)

This commit is contained in:
Nahuel Lofeudo 2023-04-23 07:14:47 +01:00
parent a95e2a29a0
commit 4a7a09fce7
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
gtfs_client.py
View File

@ -90,6 +90,9 @@ class GTFSClient():
df = pd.concat([chunk[chunk["stop_id"].isin(wanted_stop_ids)] for chunk in iter_csv]) df = pd.concat([chunk[chunk["stop_id"].isin(wanted_stop_ids)] for chunk in iter_csv])
if not df.empty: if not df.empty:
# Fix arrival and departure times so that comparisons work the way they are expected to
df["arrival_time"] = df.apply(lambda row: row["arrival_time"] if len(row["arrival_time"]) == 8 else "0"+row["arrival_time"], axis=1)
df["departure_time"] = df.apply(lambda row: row["departure_time"] if len(row["departure_time"]) == 8 else "0"+row["departure_time"], axis=1)
feed_dict["stop_times"] = gk.cn.clean_column_names(df) feed_dict["stop_times"] = gk.cn.clean_column_names(df)
feed_dict["dist_units"] = dist_units feed_dict["dist_units"] = dist_units