OnlyBiggg commited on
Commit
b8cbc0b
·
1 Parent(s): d7f3450
Files changed (1) hide show
  1. app/dialogflow/api/v1/dialogflow.py +10 -2
app/dialogflow/api/v1/dialogflow.py CHANGED
@@ -468,9 +468,13 @@ async def is_valid_select_time(request: Request) -> Response:
468
  body = await request.json()
469
  session_info = body.get("sessionInfo", {})
470
  parameters = session_info.get("parameters")
471
-
 
 
472
  raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day = dialog_service.get_param_from_dialogflow(body)
473
-
 
 
474
  from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
475
  ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
476
 
@@ -496,6 +500,10 @@ async def is_valid_select_time(request: Request) -> Response:
496
  route_name = parameters.get("route_name")
497
  is_has_time = parameters.get("is_has_time")
498
 
 
 
 
 
499
 
500
  if time_select:
501
  time_select = extra_time_dialogflow(time_select)
 
468
  body = await request.json()
469
  session_info = body.get("sessionInfo", {})
470
  parameters = session_info.get("parameters")
471
+ trip_list: list[Dict[str, any]] = parameters.get("trip_list", [])
472
+ raw_route_id = parameters.get("route_id")
473
+
474
  raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day = dialog_service.get_param_from_dialogflow(body)
475
+ route_id = int(raw_route_id) if raw_route_id else None
476
+
477
+
478
  from_time, to_time = dialog_service.process_dates_to_timestamp(raw_date)
479
  ticket_count = int(raw_ticket_number) if raw_ticket_number else 1
480
 
 
500
  route_name = parameters.get("route_name")
501
  is_has_time = parameters.get("is_has_time")
502
 
503
+ if len(time_list) == 0:
504
+ for trip in trip_list:
505
+ if (trip["route_id"]) == route_id:
506
+ time_list.append({"time": trip["departure_time"], "trip_id": trip["id"]})
507
 
508
  if time_select:
509
  time_select = extra_time_dialogflow(time_select)