This repository has been archived on 2024-11-02. You can view files and clone it, but cannot push or open issues or pull requests.
diagram-parser/src/formatter.py
2024-02-21 20:22:58 +00:00

24 lines
976 B
Python

import owlboard_connector
def humanYaml(pis_list):
additional_pis = ''
manual_review = ''
for pis in pis_list:
if len(pis['services']) == 1:
print(f"Only one valid service for {pis['pis']}")
crs = []
for stop in pis['services'][0]['stops']:
crs.append(owlboard_connector.convert_tiploc_to_crs(stop))
additional_pis += f' - code: "{pis["pis"]}"\n'
additional_pis += f' stops: [{",".join(crs)}]\n'
elif len(pis['services']) > 1:
print(f"More than one possible service for {pis['pis']}")
manual_review += f' - code: "{pis["pis"]}"\n'
for service in pis["services"]:
crs = []
for stop in service['stops']:
crs.append(owlboard_connector.convert_tiploc_to_crs(stop))
manual_review += f'OR stops:[{",".join(crs)}]\n'
print(additional_pis)
print(manual_review)