azpsen/tailfin
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix filter validation

Browse Source
This commit is contained in:
april 2024-01-11 13:12:46 -06:00
parent cf9784b770
commit 03ea9d0235
1 changed files with 5 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
api/database/flights.py
View File

@ -27,17 +27,16 @@ async def retrieve_flights(user: str = "", sort: str = "date", order: int = -1,
:param filter_val: Value to filter field by :param filter_val: Value to filter field by
:return: List of flights :return: List of flights
""" """
if filter not in FlightDisplaySchema.__annotations__.keys():
raise HTTPException(400, f"Invalid filter field: {filter}")
filter_options = {} filter_options = {}
if user != "": if user != "":
filter_options["user"] = ObjectId(user) filter_options["user"] = ObjectId(user)
if filter != "": if filter != "" and filter_val != "":
fs_keys = list(FlightCreateSchema.__annotations__.keys())
fs_keys.extend(list(FlightDisplaySchema.__annotations__.keys()))
if filter not in fs_keys:
raise HTTPException(400, f"Invalid filter field: {filter}")
filter_options[filter] = filter_val filter_options[filter] = filter_val
print(filter_options)
flights = [] flights = []
async for flight in flight_collection.find(filter_options).sort({sort: order}): async for flight in flight_collection.find(filter_options).sort({sort: order}):
flights.append(FlightConciseSchema(**flight_display_helper(flight))) flights.append(FlightConciseSchema(**flight_display_helper(flight)))