Make vao_ext_id_to_ifopt_stop_id( working for whole Austria.

[philipp/winterrodeln/wrpylib.git] / wrpylib / lib_update_public_transport.py

diff --git a/wrpylib/lib_update_public_transport.py b/wrpylib/lib_update_public_transport.py
index 898fc1e2441736388b890fc101588433c035b32f..8136cda1a2c6b0238ed7c642dacb0f7ee86f99a2 100644 (file)
--- a/wrpylib/lib_update_public_transport.py
+++ b/wrpylib/lib_update_public_transport.py
@@ -1,4 +1,6 @@
+import re
 from datetime import date, timedelta
+from typing import Optional
 
 
 def day_of_time_table_change(year: int):
@@ -20,3 +22,13 @@ def default_query_date(today: date) -> date:
     if query_date < today:
         return default_query_date(today + timedelta(days=14))
     return query_date
+
+
+def vao_ext_id_to_ifopt_stop_id(vao_ext_id: str) -> Optional[str]:
+    """Converts a VAO EXT ID like "476164600" to the IFOPT stop ID like "at:47:61646".
+    In case the vao_ext_id cannot be converted, None is returned."""
+    if match := re.match(r'(4[1-9])(\d{5})00$', vao_ext_id):
+        g1, g2 = match.groups()
+        g2 = int(g2)
+        return f"at:{g1}:{g2}"
+    return None