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

make stops sources configurable

Browse source
This commit is contained in:
Csaba 2024-02-08 13:31:21 +01:00
parent 21629aa377
commit 8b720fd8a9
1 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
amarillo/plugins/enhancer/configuration.py
View file

@ -9,21 +9,15 @@ from amarillo.plugins.enhancer.services import stops
from amarillo.plugins.enhancer.services import trips
from amarillo.plugins.enhancer.services.carpools import CarpoolService
from amarillo.plugins.enhancer.services import gtfs_generator
from amarillo.services.config import config
from amarillo.configuration import configure_services
logger = logging.getLogger(__name__)
def configure_enhancer_services():
configure_services()
def configure_container():
logger.info("Load stops...")
stop_sources = [
{"url": "https://datahub.bbnavi.de/export/rideshare_points.geojson", "vicinity": 50},
{"url": "https://data.mfdz.de/mfdz/stops/stops_zhv.csv", "vicinity": 50},
{"url": "https://data.mfdz.de/mfdz/stops/parkings_osm.csv", "vicinity": 500},
]
with open(config.stop_sources_file) as stop_sources_file:
stop_sources = json.load(stop_sources_file)
stop_store = stops.StopsStore(stop_sources)
stop_store.load_stop_sources()
@ -31,6 +25,10 @@ def configure_enhancer_services():
container['trips_store'] = trips.TripStore(stop_store)
container['carpools'] = CarpoolService(container['trips_store'])
def configure_enhancer_services():
configure_services()
configure_container()
logger.info("Restore carpools...")
for agency_id in container['agencies'].agencies: