def add_line(self, name, code, stationList):
line = Line(name, code)
self.lineList.append(line)
prevStation = None
nextStation = None
stationObjectList = []
routeObjectList = []
for i, v in enumerate(stationList):
name = v['station']
if name in self.stationMap:
station = self.stationMap[name]
else:
station = Station(name, 0, 0)
self.stationMap[name] = station
route = Route(line, prevStation, nextStation)
prevStation = station
stationObjectList.append(station)
routeObjectList.append(route)
length = len(stationObjectList)
for i, route in enumerate(routeObjectList):
if i != length - 1:
route.nextStation = stationObjectList[i + 1]
stationObjectList[i].add_routes(route)
