OnlyBiggg commited on
Commit
1f8c0c2
·
1 Parent(s): b6a3402
app/dialogflow/services/dialog_service.py CHANGED
@@ -35,19 +35,20 @@ class DialogService:
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
- }
46
  response = await api.get(f'/metadata/office/routes' , params=params)
47
  route_ids = []
48
  if isinstance(response, list):
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)
 
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 = {k: v for k, v in {
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
+ }.items() if v is not None}
46
  response = await api.get(f'/metadata/office/routes' , params=params)
47
  route_ids = []
48
  if isinstance(response, list):
49
  route_ids = [route.get("routeId", None) for route in response]
50
+ return route_ids
51
+ return []
52
 
53
  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):
54
  route_ids_origin_dest = await self.search_route_ids_one_way(origin_code, from_id, orign_ids, dest_code, to_id, dest_ids)