Update main.py

This commit is contained in:
Cal Corum 2023-06-21 10:07:42 -05:00
parent a7bfcaaff1
commit b445957e3c

10
main.py
View File

@ -1375,6 +1375,8 @@ async def v1_players_get_random(
final_players = [] final_players = []
p_list = [x.upper() for x in pos_exc] p_list = [x.upper() for x in pos_exc]
for x in all_players: for x in all_players:
if limit is not None and len(final_players) >= limit:
break
if not set(p_list).intersection(x.get_all_pos()): if not set(p_list).intersection(x.get_all_pos()):
final_players.append(x) final_players.append(x)
else: else:
@ -2566,10 +2568,6 @@ async def v1_events_get(
csv: Optional[bool] = None): csv: Optional[bool] = None):
all_events = Event.select() all_events = Event.select()
if all_events.count() == 0:
db.close()
raise HTTPException(status_code=404, detail=f'There are no events to filter')
if name is not None: if name is not None:
all_events = all_events.where(fn.Lower(Event.name) == name.lower()) all_events = all_events.where(fn.Lower(Event.name) == name.lower())
if in_desc is not None: if in_desc is not None:
@ -2580,10 +2578,6 @@ async def v1_events_get(
if active is not None: if active is not None:
all_events = all_events.where(Event.active == active) all_events = all_events.where(Event.active == active)
if all_events.count() == 0:
db.close()
raise HTTPException(status_code=404, detail=f'No events found')
if csv: if csv:
data_list = [['id', 'name', 'short_desc', 'long_desc', 'url', 'thumbnail', 'active']] data_list = [['id', 'name', 'short_desc', 'long_desc', 'url', 'thumbnail', 'active']]
for line in all_events: for line in all_events: