tcloud compatibility, dotnet bug fixes

This commit is contained in:
Steve Androulakis
2025-02-14 11:10:16 -08:00
parent a9d8a2a631
commit d6bbb900b7
12 changed files with 227 additions and 53 deletions

View File

@@ -2,50 +2,34 @@ import requests
import os
from dotenv import load_dotenv
BASE_URL = 'http://localhost:8080/'
def search_trains(args: dict) -> dict:
load_dotenv(override=True)
raise NotImplementedError("TODO implement :)")
origin = args.get("origin")
destination = args.get("destination")
outbound_time = args.get("outbound_time")
return_time = args.get("return_time")
if not origin or not destination or not outbound_time or not return_time:
return {"error": "Origin, destination, outbound_time and return_time are required."}
search_url = f'{BASE_URL}/api/search'
params = {
'from': origin,
'to': destination,
'outbound_time': outbound_time,
'return_time': return_time,
}
response = requests.get(search_url, params=params)
if response.status_code != 200:
print(response.content)
return {"error": "Failed to fetch journey data."}
journey_data = response.json()
return journey_data
def book_trains(args: dict) -> dict:
load_dotenv(override=True)
raise NotImplementedError("TODO implement :)")
train_ids = args.get("train_ids")
if not train_ids:
return {"error": "Train IDs is required."}
# todo clean this up
# BASE_URL = "http://localhost:8080/"
book_url = f'{BASE_URL}/api/book/{train_ids}'
response = requests.post(book_url)
if response.status_code != 200:
return {"error": "Failed to book ticket."}
# def book_trains(args: dict) -> dict:
# load_dotenv(override=True)
# train_ids = args.get("train_ids")
# if not train_ids:
# return {"error": "Train IDs is required."}
# book_url = f"{BASE_URL}/api/book/{train_ids}"
# response = requests.post(book_url)
# if response.status_code != 200:
# return {"error": "Failed to book ticket."}
# booking_data = response.json()
# return booking_data
booking_data = response.json()
return booking_data
# Example usage
if __name__ == "__main__":
@@ -53,7 +37,7 @@ if __name__ == "__main__":
"origin": "London Gatwick",
"destination": "Manchester",
"outbound_time": "2025-03-15T14:00",
"return_time": "2025-03-20T14:00"
"return_time": "2025-03-20T14:00",
}
search_results = search_trains(search_args)
print(search_results)
@@ -62,4 +46,32 @@ if __name__ == "__main__":
"train_ids": "12345",
}
booking_results = book_trains(book_args)
print(booking_results)
print(booking_results)
# def search_trains(args: dict) -> dict:
# load_dotenv(override=True)
# origin = args.get("origin")
# destination = args.get("destination")
# outbound_time = args.get("outbound_time")
# return_time = args.get("return_time")
# if not origin or not destination or not outbound_time or not return_time:
# return {"error": "Origin, destination, outbound_time and return_time are required."}
# search_url = f'{BASE_URL}/api/search'
# params = {
# 'from': origin,
# 'to': destination,
# 'outbound_time': outbound_time,
# 'return_time': return_time,
# }
# response = requests.get(search_url, params=params)
# if response.status_code != 200:
# print(response.content)
# return {"error": "Failed to fetch journey data."}
# journey_data = response.json()
# return journey_data