Add ifopt_stop_id.

[philipp/winterrodeln/wrpylib.git] / scripts / update_public_transport.py

diff --git a/scripts/update_public_transport.py b/scripts/update_public_transport.py
index 620035000caf705faf2b519dc498ae8b008aa2fd..a8ee901ff5c5585b97c4b30c48da9741559dfbae 100644 (file)
--- a/scripts/update_public_transport.py
+++ b/scripts/update_public_transport.py
@@ -14,7 +14,7 @@ from termcolor import cprint  # python3-termcolor
 
 from wrpylib.cli_tools import unified_diff, input_yes_no_quit, Choice
 from wrpylib.json_tools import order_json_keys, format_json
 
 from wrpylib.cli_tools import unified_diff, input_yes_no_quit, Choice
 from wrpylib.json_tools import order_json_keys, format_json

-from wrpylib.lib_update_public_transport import default_query_date
+from wrpylib.lib_update_public_transport import default_query_date, vao_ext_id_to_ifopt_stop_id

 from wrpylib.mwapi import WikiSite, page_json
 from wrpylib.vao import Vao
 from wrpylib.wrvalidators import LonLat
 from wrpylib.mwapi import WikiSite, page_json
 from wrpylib.vao import Vao
 from wrpylib.wrvalidators import LonLat


@@ -318,6 +318,8 @@ def update_sledrun(vao: Vao, site: WikiSite, title: str, query_date: date):
             "walk_time": stop_with_dist.dist.time_min,
             "lines": lines,
         }
             "walk_time": stop_with_dist.dist.time_min,
             "lines": lines,
         }

+        if ifopt_stop_id := vao_ext_id_to_ifopt_stop_id(stop_with_dist.stop.ext_id):
+            public_transport_stop['ifopt_stop_id'] = ifopt_stop_id

         vvt_stop_id = get_vvt_stop_id(stop_with_dist.stop.ext_id)
         if vvt_stop_id is not None:
             public_transport_stop["vvt_stop_id"] = vvt_stop_id
         vvt_stop_id = get_vvt_stop_id(stop_with_dist.stop.ext_id)
         if vvt_stop_id is not None:
             public_transport_stop["vvt_stop_id"] = vvt_stop_id