scripts/update_car_distances.py

   1 #!/usr/bin/python
   2 import argparse
   3 import csv
   4 import re
   5 import sqlite3
   6 from itertools import islice
   7 from typing import List
   8
   9 import isodate
  10 import jsonschema
  11
  12 from wrpylib.json_tools import order_json_keys, format_json
  13 from wrpylib.mwapi import WikiSite, page_json
  14 from wrpylib.vao import Vao
  15
  16
  17 def update_sledrun(vao: Vao, db_cities: sqlite3.Connection, site: WikiSite, title: str):
  18     sledrun_json_page = site.query_page(f'{title}/Rodelbahn.json')
  19     sledrun_json = page_json(sledrun_json_page)
  20     sledrun_json_orig = sledrun_json.copy()
  21
  22     car_parking = sledrun_json.get('car_parking')
  23     if not car_parking:
  24         return
  25
  26     parking = car_parking[0].get('position', {}).get('position')
  27     if not parking:
  28         return
  29     parking_lon = parking['longitude']
  30     parking_lat = parking['latitude']
  31
  32     sql = 'select Name, st_x(geometry) as lon, st_y(geometry) as lat from cities where ' \
  33           'st_distance(geometry, st_point(?, ?), 0) < ?;'
  34     max_dist_m = 50000
  35     print(title)
  36     for row in db_cities.execute(sql, [parking_lon, parking_lat, max_dist_m]):
  37         city_name, city_lon, city_lat = row
  38         print(f"* {city_name}")
  39         parameter = {
  40             'originCoordLat': city_lat,
  41             'originCoordLong': city_lon,
  42             'destCoordLat': parking_lat,
  43             'destCoordLong': parking_lon,
  44             'groupFilter': 'API_CAR',
  45             'totalCar': '1|evnt=0,aevnt=0,tsta=0,htsta=0,getInitEndTimes=0',
  46         }
  47         response = vao.trip(parameter).json()
  48         trip = response.get('Trip', [])
  49         if trip:
  50             leg_list = trip[0].get('LegList', {}).get('Leg', [])
  51             if leg_list:
  52                 leg = leg_list[0]
  53                 duration = isodate.parse_duration(leg['duration'])
  54                 dist_m = leg['dist']
  55                 co2_kg = trip[0]['Eco']['co2']
  56                 print(duration, dist_m, co2_kg)
  57
  58     print()
  59
  60
  61 def update_car_distances(ini_files: List[str]):
  62     db_cities = sqlite3.connect('/home/philipp/projects/winterrodeln/geodata/Ausgangspunkte.sqlite')
  63     db_cities.enable_load_extension(True)
  64     db_cities.execute("SELECT load_extension('mod_spatialite');")
  65
  66     site = WikiSite(ini_files)
  67     vao = Vao(site.config.get('vao', 'access_id'))
  68     for result in site.query(list='categorymembers', cmtitle='Kategorie:Rodelbahn', cmlimit='max'):
  69         for page in result['categorymembers']:
  70             update_sledrun(vao, db_cities, site, page['title'])
  71
  72
  73 def main():
  74     parser = argparse.ArgumentParser(description='Update car distance information in sledrun JSON files.')
  75     parser.add_argument('inifile', nargs='+', help='inifile.ini, see: https://www.winterrodeln.org/trac/wiki/ConfigIni')
  76     args = parser.parse_args()
  77     update_car_distances(args.inifile)
  78
  79
  80 if __name__ == '__main__':
  81     main()