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 jsonschema
  10
  11 from wrpylib.json_tools import order_json_keys, format_json
  12 from wrpylib.mwapi import WikiSite, page_json
  13
  14
  15 def update_sledrun(site: WikiSite, title: str):
  16     sledrun_json_page = site.query_page(f'{title}/Rodelbahn.json')
  17     sledrun_json = page_json(sledrun_json_page)
  18     sledrun_json_orig = sledrun_json.copy()
  19
  20     car_parking = sledrun_json.get('car_parking')
  21     if not car_parking:
  22         return
  23
  24     position = car_parking[0].get('position', {}).get('position')
  25     if not position:
  26         return
  27
  28     db_cities = sqlite3.connect('/home/philipp/projects/winterrodeln/geodata/Ausgangspunkte.sqlite')
  29     db_cities.enable_load_extension(True)
  30     db_cities.execute("SELECT load_extension('mod_spatialite');")
  31     sql = 'select Name, st_x(geometry) as lon, st_y(geometry) as lat from cities where ' \
  32           'st_distance(geometry, st_point(?, ?), 0) < ?;'
  33     max_dist_m = 50000
  34     print(title)
  35     for row in db_cities.execute(sql, [position['longitude'], position['latitude'], max_dist_m]):
  36         city_name, city_lon, city_lat = row
  37         print(f"* {city_name}")
  38     print()
  39
  40
  41 def update_car_distances(ini_files: List[str]):
  42     site = WikiSite(ini_files)
  43     for result in site.query(list='categorymembers', cmtitle='Kategorie:Rodelbahn', cmlimit='max'):
  44         for page in result['categorymembers']:
  45             update_sledrun(site, page['title'])
  46
  47
  48 def main():
  49     parser = argparse.ArgumentParser(description='Update car distance information in sledrun JSON files.')
  50     parser.add_argument('inifile', nargs='+', help='inifile.ini, see: https://www.winterrodeln.org/trac/wiki/ConfigIni')
  51     args = parser.parse_args()
  52     update_car_distances(args.inifile)
  53
  54
  55 if __name__ == '__main__':
  56     main()