diff options
Diffstat (limited to 'cli_view.py')
| -rw-r--r-- | cli_view.py | 19 |
1 files changed, 19 insertions, 0 deletions
diff --git a/cli_view.py b/cli_view.py new file mode 100644 index 0000000..9c022f9 --- /dev/null +++ b/cli_view.py @@ -0,0 +1,19 @@ +import util +import ptv_client + +def format_stop_and_route_name(stop_and_route): + stop, route = stop_and_route + return 'Stop: {}, Route: {} {} {}'.format( + stop['stop_name'], ptv_client.get_route_type(route['route_type']), + route['route_number'], route['route_name']) + +def format_stop_and_route_names(stop_and_routes): + return '\n'.join(f'[{i}] {result}' for i, result in enumerate( + map(format_stop_and_route_name, stop_and_routes))) + +def format_departures(departures, direction_names): + return '\n'.join(['estimated: {}; scheduled: {}; direction: {}'.format( + util.format_time(util.parse_time(dep['estimated_departure_utc'])), + util.format_time(util.parse_time(dep['scheduled_departure_utc'])), + direction_names[dep['direction_id']]) + for dep in departures]) |