diff --git a/amarillo/plugins/enhancer/services/gtfs_export.py b/amarillo/plugins/enhancer/services/gtfs_export.py index 0a3f83b..1abd5c0 100644 --- a/amarillo/plugins/enhancer/services/gtfs_export.py +++ b/amarillo/plugins/enhancer/services/gtfs_export.py @@ -24,6 +24,7 @@ class GtfsExport: stored_stops = {} + # TODO: add lists self.drivers and self.additional_ridesharing_infos def __init__(self, agencies, feed_info, ridestore, stopstore, bbox = None): self.stops = {} self.routes = [] @@ -53,6 +54,7 @@ class GtfsExport: self._write_csvfile(gtfsfolder, 'stops.txt', self.stops.values()) self._write_csvfile(gtfsfolder, 'stop_times.txt', self.stop_times) self._write_csvfile(gtfsfolder, 'shapes.txt', self.shapes) + # TODO: write driver.txt and additional_ridesharing_info.txt self._zip_files(gtfszip_filename, gtfsfolder) def _zip_files(self, gtfszip_filename, gtfsfolder): @@ -79,6 +81,7 @@ class GtfsExport: for group in groups: self.convert_route(group) for url, trip in cloned_trips.items(): + # TODO: convert ridesharing info and driver data if self.bbox is None or trip.intersects(self.bbox): self._convert_trip(trip)