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

Made filters work with refactored changes

Browse source
This commit is contained in:
Yarne Coppens 2024-08-22 10:15:24 +02:00
parent 51cd2dda07
commit 339745daf7
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
src/filters/boardgame_filters.py
View file

@ -4,13 +4,13 @@ from src.classes import boardgame_classes
def filter_expansions_out(to_filter_boardgames: list[boardgame_classes.BoardGame]): def filter_expansions_out(to_filter_boardgames: list[boardgame_classes.BoardGame]):
filtered_boardgames = list(filter(lambda x: type(x) in [boardgame_classes.BoardGame], to_filter_boardgames)) filtered_boardgames = list(filter(lambda x: x.expansion_info == None, to_filter_boardgames))
return filtered_boardgames return filtered_boardgames
def filter_non_expansions_out(to_filter_boardgames: list[boardgame_classes.BoardGame]): def filter_non_expansions_out(to_filter_boardgames: list[boardgame_classes.BoardGame]):
#filtered_boardgames = list(filter(lambda x: type(x) in [boardgame_classes.BoardGameExpansion, boardgame_classes.OwnedBoardGameExpansion, boardgame_classes.WishlistBoardGameExpansion], to_filter_boardgames)) filtered_boardgames = list(filter(lambda x: x.expansion_info != None, to_filter_boardgames))
filtered_boardgames = to_filter_boardgames filtered_boardgames = to_filter_boardgames

4
src/filters/play_filters.py
View file

@ -3,14 +3,14 @@ from src.classes import play_classes, boardgame_classes
def filter_expansions_out(play_list: list[play_classes.Play]): def filter_expansions_out(play_list: list[play_classes.Play]):
to_return_plays = [] to_return_plays = []
to_return_plays = list(filter(lambda x: x.expansion_id == None, play_list)) to_return_plays = list(filter(lambda x: x.boardgame.expansion_info == None, play_list))
return to_return_plays return to_return_plays
def filter_non_expansions_out(play_list: list[play_classes.Play]): def filter_non_expansions_out(play_list: list[play_classes.Play]):
to_return_plays = [] to_return_plays = []
to_return_plays = list(filter(lambda x: x.expansion_id != None, play_list)) to_return_plays = list(filter(lambda x: x.boardgame.expansion_info != None, play_list))
return to_return_plays return to_return_plays