diff --git a/main.py b/main.py index 956fa00..340867c 100644 --- a/main.py +++ b/main.py @@ -1,4 +1,5 @@ import enum +import re from typing import Any, DefaultDict, List, Optional import httpx @@ -48,7 +49,11 @@ async def stop( passages = DefaultDict(list) for passage in res.json().get("values"): if passage.get("id") == stop_id: - passages[passage.get("ligne")].append(passage.get("delaipassage")) + ligne = passage.get("ligne") + ligne = re.sub( + "[A-Z]$", "", ligne + ) # Remove letter suffix to group by commercial line name + passages[ligne].append(passage.get("delaipassage")) passages_list = [] for ligne, delais in passages.items(): passages_list.append(Passage(ligne=ligne, delais=delais))