81 lines
2.5 KiB
Python
81 lines
2.5 KiB
Python
from fastapi import FastAPI
|
|
|
|
from amarillo.models.Carpool import Region
|
|
from amarillo.plugins.grfs_export.gtfs_export import GtfsExport, GtfsFeedInfo, GtfsAgency
|
|
from amarillo.plugins.grfs_export.gtfs import GtfsRtProducer
|
|
from amarillo.utils.container import container
|
|
from amarillo.plugins.grfs_export.router import router
|
|
from amarillo.plugins.enhancer.configuration import configure_enhancer_services
|
|
from amarillo.utils.utils import assert_folder_exists
|
|
from glob import glob
|
|
import json
|
|
import schedule
|
|
import threading
|
|
import time
|
|
import logging
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
regions = {}
|
|
for region_file_name in glob('conf/region/*.json'):
|
|
with open(region_file_name) as region_file:
|
|
dict = json.load(region_file)
|
|
region = Region(**dict)
|
|
region_id = region.id
|
|
regions[region_id] = region
|
|
|
|
agencies = []
|
|
for agency_file_name in glob('conf/agency/*.json'):
|
|
with open(agency_file_name) as agency_file:
|
|
dict = json.load(agency_file)
|
|
agency = GtfsAgency(dict["id"], dict["name"], dict["url"], dict["timezone"], dict["lang"], dict["email"])
|
|
agency_id = agency.agency_id
|
|
agencies.append(agency)
|
|
|
|
def run_schedule():
|
|
while 1:
|
|
try:
|
|
schedule.run_pending()
|
|
except Exception as e:
|
|
logger.exception(e)
|
|
time.sleep(1)
|
|
|
|
def midnight():
|
|
container['stops_store'].load_stop_sources()
|
|
container['trips_store'].unflag_unrecent_updates()
|
|
container['carpools'].purge_outdated_offers()
|
|
generate_gtfs()
|
|
|
|
def generate_gtfs():
|
|
logger.info("Generate GRFS")
|
|
|
|
for region in regions.values():
|
|
# TODO make feed producer infos configurable
|
|
feed_info = GtfsFeedInfo('mfdz', 'MITFAHR|DE|ZENTRALE', 'http://www.mitfahrdezentrale.de', 'de', 1)
|
|
exporter = GtfsExport(
|
|
agencies,
|
|
feed_info,
|
|
container['trips_store'],
|
|
container['stops_store'],
|
|
region.bbox)
|
|
exporter.export(f"data/grfs/amarillo.{region.id}.gtfs.zip", "data/tmp/grfs")
|
|
|
|
def generate_gtfs_rt():
|
|
logger.info("Generate GRFS-RT")
|
|
producer = GtfsRtProducer(container['trips_store'])
|
|
for region in regions.values():
|
|
rt = producer.export_feed(time.time(), f"data/grfs/amarillo.{region.id}.gtfsrt", bbox=region.bbox)
|
|
|
|
def start_schedule():
|
|
schedule.every().day.at("00:00").do(midnight)
|
|
schedule.every(60).seconds.do(generate_gtfs_rt)
|
|
# Create all feeds once at startup
|
|
schedule.run_all()
|
|
job_thread = threading.Thread(target=run_schedule, daemon=True)
|
|
job_thread.start()
|
|
|
|
def setup(app : FastAPI):
|
|
assert_folder_exists("data/grfs")
|
|
configure_enhancer_services()
|
|
app.include_router(router)
|
|
start_schedule() |