Yarninator/bgg_api
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Wrote the missing boardgame id retrieval much more efficiently

Browse source
This commit is contained in:
Yarne Coppens 2024-08-14 10:08:07 +02:00
parent 40980ae89a
commit 896ef700cd
1 changed files with 6 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
src/modules/db_connection.py
View file

@ -66,19 +66,15 @@ def get_boardgame(session: Session, boardgame_id: int) -> Union[
def get_multiple_boardgames(session: Session, boardgame_type: SQLModel, boardgame_ids: list[int]) -> tuple[Union[
list[boardgame_classes.BoardGame], list[boardgame_classes.BoardGameExpansion]], list[int]]:
missing_boardgame_ids = []
to_return_boardgames = []
statement = select(boardgame_type).where(boardgame_type.id.in_(boardgame_ids))
results = session.exec(statement)
for boardgame_id in boardgame_ids:
statement = select(boardgame_type).where(boardgame_type.id == boardgame_id)
results = session.exec(statement).all()
if len(results) == 0:
missing_boardgame_ids.append(boardgame_id)
else:
to_return_boardgames.append(results[0])
boardgames = results.all()
return to_return_boardgames, missing_boardgame_ids
missing_boardgame_ids = list(filter(lambda x: x not in [boardgame.id for boardgame in boardgames], boardgame_ids))
return boardgames, missing_boardgame_ids
def get_all_boardgames(session: Session, boardgame_type: SQLModel) -> Union[
list[boardgame_classes.BoardGame], list[boardgame_classes.BoardGameExpansion],