initial commit

main.py

@@ -0,0 +1,72 @@
+import json
+from typing import Any
+
+import requests
+
+BASE_URL = "https://www.nightjet.com"
+
+
+def dprint(txt) -> None:
+    print(txt)
+
+
+def request_init_token(endpoint: str = "/nj-booking-ocp/init/start") -> str:
+    DEBUG_TOKEN = "eyJhbGciOiJIUzI1NiJ9.eyJleHAiOjE3NTU1NDMyMjMsInB1YmxpY0lkIjoiYmU2N2ZlNDNjY2Y3NDI3Yjk0MjY3NmI0MjJmZmIzOWYifQ.Hvo7Ljm9iFof_w7RrQkVVACOX8wgY2qAzTKAYDm5QC4"
+    token = DEBUG_TOKEN
+
+    # headers = {
+    #     "Referer": "https://www.nightjet.com",
+    #     "Content-Type": "application/json"
+    # }
+    # body = {
+    #     "lang": "en"
+    # }
+    # resp_json = requests.post(f"{BASE_URL}{endpoint}", data=json.dumps(body), headers=headers).json()
+    # token = resp_json["token"]
+
+    dprint(f"Received init token: {token}")
+    return token
+
+
+START_STATION = "8096003"  # BerlinHBF
+END_STATION = "8796001"  # Paris Est
+TRAVEL_DATE = "2025-10-14"
+
+
+def request_connections(
+    token: str, endpoint: str = "/nj-booking-ocp/connection"
+) -> list[Any]:
+    DEBUG_CONNECTIONS = json.loads(
+        '{ "connections": [ { "from": { "name": "Berlin Hbf", "number": "8011160" }, "to": { "name": "Paris Est", "number": "8700011" }, "trains": [ { "train": "NJ 40424", "departure": { "utc": 1760461680000, "local": "2025-10-14T19:08:00" }, "arrival": { "utc": 1760513880000, "local": "2025-10-15T09:38:00" }, "trainType": "regular", "seatAsIC": false } ] }, { "from": { "name": "Berlin Hbf", "number": "8011160" }, "to": { "name": "Paris Est", "number": "8700011" }, "trains": [ { "train": "NJ 40424", "departure": { "utc": 1760634480000, "local": "2025-10-16T19:08:00" }, "arrival": { "utc": 1760686680000, "local": "2025-10-17T09:38:00" }, "trainType": "regular", "seatAsIC": false } ] }, { "from": { "name": "Berlin Hbf", "number": "8011160" }, "to": { "name": "Paris Est", "number": "8700011" }, "trains": [ { "train": "NJ 40424", "departure": { "utc": 1760893680000, "local": "2025-10-19T19:08:00" }, "arrival": { "utc": 1760945880000, "local": "2025-10-20T09:38:00" }, "trainType": "regular", "seatAsIC": false } ] } ] }'
+    )
+    resp_json = DEBUG_CONNECTIONS
+
+    #     uri = f"{BASE_URL}{endpoint}/{START_STATION}/{END_STATION}/{TRAVEL_DATE}"
+    #     headers = {
+    #         "Accept": "application/json",
+    #         "Accept-Language": "en-US,en;q=0.5",
+    #         "Referer": "https://www.nightjet.com/en/ticket-buchen/",
+    #         "x-token": token,
+    #         "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:139.0) Gecko/20100101 Firefox/139.0",
+    #     }
+    #
+    #     resp_json = requests.get(
+    #         uri,
+    #         headers=headers,
+    #     ).json()
+    return resp_json["connections"]
+
+
+def request_bookings(endpoint: str = "") -> dict[Any, Any]: ...
+
+
+def get_prices(bookings_dict: dict[Any, Any]) -> dict[Any, Any]: ...
+
+
+def main():
+    token = request_init_token()
+    print(request_connections(token))
+
+
+if __name__ == "__main__":
+    main()