amarillo/amarillo-enhancer
5
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

#25 GRFS export TODOs
All checks were successful
Amarillo/amarillo-gitea/amarillo-enhancer/pipeline/head This commit looks good
Details

Browse source
This commit is contained in:
Csaba 2024-01-18 14:52:39 +01:00
parent 6941fd9bf5
commit 78fb201db6
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
amarillo/plugins/enhancer/services/gtfs_export.py
View file

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