32 lines
1.4 KiB
Python
32 lines
1.4 KiB
Python
|
from fastapi import APIRouter, HTTPException
|
||
|
|
|
||
|
|
from genetic_algorithm.genetic_algorithm import genetic_algorithm, load_graph_from_request
|
||
|
|
from schemas import TripRequest
|
||
|
|
|
||
|
|
router = APIRouter(
|
||
|
|
prefix="/flight",
|
||
|
|
tags=["Flight"],
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
@router.post("/get/")
|
||
|
|
async def get_flight(request: TripRequest):
|
||
|
|
graphTo, graphBack, start_point, end_point, flightsDataTo, flightsDataBack, departure_date, departure_date_return = load_graph_from_request(request)
|
||
|
|
if request.returnDate:
|
||
|
|
resultTo = genetic_algorithm(start=start_point, end=end_point, graph=graphTo, flights_data=flightsDataTo, type = "to", departure_date = departure_date)
|
||
|
|
resulFrom = genetic_algorithm(end=start_point, start=end_point, graph=graphBack, flights_data=flightsDataBack, type = "back", departure_date = departure_date_return)
|
||
|
|
if not resultTo or not resulFrom:
|
||
|
|
raise HTTPException(status_code=404, detail="No valid paths found")
|
||
|
|
return resultTo, resulFrom
|
||
|
|
if not request.returnDate:
|
||
|
|
resultTo = genetic_algorithm(start=start_point, end=end_point, graph=graphTo, flights_data=flightsDataTo,
|
||
|
|
type="to", departure_date = departure_date)
|
||
|
|
|
||
|
|
if resultTo:
|
||
|
|
resultTo.append({
|
||
|
|
"back": []
|
||
|
|
})
|
||
|
|
else:
|
||
|
|
raise HTTPException(status_code=404, detail="No valid paths found")
|
||
|
|
return resultTo
|