Add booking requests

main.py

@@ -1,4 +1,5 @@
 import json
+from pprint import pprint
 from typing import Any
 
 import requests
@@ -57,7 +58,50 @@ def request_connections(
     return resp_json["connections"]
 
 
-def request_bookings(endpoint: str = "") -> dict[Any, Any]: ...
+TRAVELLER_BIRTHDATE = "2000-07-15"  # TODO: randomize a little
+
+
+def connection_data_to_booking_requests(connections):
+    b_requests = []
+    for c in connections:
+        train = c["trains"][0]
+        dep = train["departure"]["utc"]
+        req = {
+            "njFrom": c["from"]["number"],  # from station,
+            "njTo": c["to"]["number"],  # to station
+            "njDep": dep,  # departure time,
+            "maxChanges": 0,
+            "connections": 1,
+            "filter": {
+                "njTrain": train["train"],  # train number
+                "njDeparture": dep,  # departure time again
+            },
+            "objects": [  # traveller
+                {"type": "person", "birthDate": TRAVELLER_BIRTHDATE, "cards": []}
+            ],
+            "relations": [],
+            "lang": "en",
+        }
+        b_requests.append(req)
+    return b_requests
+
+
+def request_bookings(
+    token: str, booking_req: dict[str, Any], endpoint: str = "/nj-booking-ocp/offer/get"
+) -> dict[Any, Any]:
+    headers = {
+        "Accept": "application/json",
+        "Accept-Language": "en-US,en;q=0.5",
+        "Content-Type": "application/json",
+        "Referer": "https://www.nightjet.com/en/ticket-buchen/",
+        "Origin": "https://www.nightjet.com",
+        "x-token": token,
+        "User-Agent": "Mozilla/5.0 (X11; Linux x86_64; rv:139.0) Gecko/20100101 Firefox/139.0",
+    }
+    resp_json = requests.post(
+        f"{BASE_URL}{endpoint}", headers=headers, data=json.dumps(booking_req)
+    ).json()
+    return resp_json
 
 
 def get_prices(bookings_dict: dict[Any, Any]) -> dict[Any, Any]: ...
@@ -65,7 +109,9 @@ def get_prices(bookings_dict: dict[Any, Any]) -> dict[Any, Any]: ...
 
 def main():
     token = request_init_token()
-    print(request_connections(token))
+    connections = request_connections(token)
+    booking_requests = connection_data_to_booking_requests(connections)
+    dprint(request_bookings(token, booking_requests[0]))
 
 
 if __name__ == "__main__":