alicia/baytransitclis
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Horrid station parsing

Browse Source
This commit is contained in:
Alicia Moore
2022-06-28 17:32:05 -07:00
parent d1c7188528
commit ad6c6b127f
1 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
bt.nim
View File

@@ -27,6 +27,30 @@ proc lineColor(color: string): ForegroundColor =
of "orange": fgMagenta of "orange": fgMagenta
else: fgDefault else: fgDefault
# try to guess which station abbreviation I meant to type
proc parseStation(station: string): string =
case station:
of "ashby": "ashb"
of "antioch": "antioch"
of "balboa": "balb"
of "berkeley", "berkly": "dtbrk"
of "civic", "cvc": "civc"
of "uptown": "19th"
of "emb", "embark": "embr"
of "fmt", "fremont": "frmt"
of "mcaurthur", "mcaurther": "mcar"
of "montgomery": "mont"
of "walnut": "wcrk"
of "hwd", "hayward": "haywd"
of "sfo": "sfia"
of "oak": "oakl"
of "orinda": "orin"
of "millbrae", "mil": "mlbr"
of "powell", "powel": "powl"
of "rockridge": "rock"
of "west": "woak"
else: station
proc formatDepartures(etd: JSONNode): string = proc formatDepartures(etd: JSONNode): string =
let name = etd["root"]["station"][0]["name"].getStr let name = etd["root"]["station"][0]["name"].getStr
let lines = etd["root"]["station"][0]["etd"] let lines = etd["root"]["station"][0]["etd"]
@@ -57,7 +81,7 @@ proc main =
if len(params) == 0: if len(params) == 0:
usage() usage()
elif len(params) == 1: elif len(params) == 1:
let station = params[0] let station = parseStation(params[0])
echo formatDepartures(getDepartures(station)) echo formatDepartures(getDepartures(station))
main() main()