def load_routes(self):
routes_file = os.path.join(self.source_folder, 'routes.txt')
self.available_files['routes.txt'] = True
data = self.open(routes_file)
# Iterate over all the stops and puts them in the stops dictionary
for i in range(data.shape[0]):
r = Route()
# Required fields
r.id = data['route_id'][i]
r.short_name = data['route_short_name'][i]
r.long_name = data['route_long_name'][i]
r.type = data['route_type'][i]
# optional fields
available_fields = data.dtype.names
if 'agency_id' in available_fields:
r.agency_id = data['agency_id'][i]
if 'route_desc' in available_fields: r.desc = data['route_desc'][i]
if 'route_url' in available_fields: r.url = data['route_url'][i]
if 'route_color' in available_fields:
r.color = data['route_color'][i]
if 'route_text_color' in available_fields:
r.text_color = data['route_text_color'][i]
if 'route_sort_order' in available_fields:
r.sort_order = data['route_sort_order'][i]
self.routes[r.id] = r
del data
