OnlyBiggg commited on
Commit
b0f3d0e
·
1 Parent(s): 835d9e6
app/dialogflow/services/dialog_service.py CHANGED
@@ -34,12 +34,12 @@ class DialogService:
34
  return raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day
35
 
36
  @staticmethod
37
- async def search_route_ids_one_way(origin_code: str = None, from_id: int = None, orign_ids: int = None, des_code: str = None, to_id: str = None, dest_ids: str = None):
38
  params = {
39
  "OriginCode": origin_code,
40
  "FromId": from_id,
41
  "OriginIds": orign_ids,
42
- "DestCode": des_code,
43
  "ToId": to_id,
44
  "DestIds": dest_ids
45
  }
@@ -49,9 +49,9 @@ class DialogService:
49
  route_ids = [route.get("routeId", None) for route in response]
50
  return route_ids
51
 
52
- async def search_all_route_ids(self, origin_code: str = None, from_id: int = None, orign_ids: int = None, des_code: str = None, to_id: str = None, dest_ids: str = None)
53
- route_ids_origin_dest = await self.search_route_ids_one_way(origin_code, from_id, orign_ids, des_code, to_id, dest_ids)
54
- route_ids_dest_origin = await self.search_route_ids_one_way(des_code, to_id, dest_ids, origin_code, from_id, orign_ids)
55
  route_ids = list(set(route_ids_origin_dest + route_ids_dest_origin))
56
  return route_ids
57
  @staticmethod
 
34
  return raw_departure_city, raw_destination_city, raw_ticket_number, raw_date, raw_time_of_day
35
 
36
  @staticmethod
37
+ async def search_route_ids_one_way(origin_code: str = None, from_id: int = None, orign_ids: int = None, dest_code: str = None, to_id: str = None, dest_ids: str = None):
38
  params = {
39
  "OriginCode": origin_code,
40
  "FromId": from_id,
41
  "OriginIds": orign_ids,
42
+ "DestCode": dest_code,
43
  "ToId": to_id,
44
  "DestIds": dest_ids
45
  }
 
49
  route_ids = [route.get("routeId", None) for route in response]
50
  return route_ids
51
 
52
+ async def search_all_route_ids(self, origin_code: str = None, from_id: int = None, orign_ids: int = None, dest_code: str = None, to_id: str = None, dest_ids: str = None)
53
+ route_ids_origin_dest = await self.search_route_ids_one_way(origin_code, from_id, orign_ids, dest_code, to_id, dest_ids)
54
+ route_ids_dest_origin = await self.search_route_ids_one_way(dest_code, to_id, dest_ids, origin_code, from_id, orign_ids)
55
  route_ids = list(set(route_ids_origin_dest + route_ids_dest_origin))
56
  return route_ids
57
  @staticmethod