From b8bb824115a615dcabcaf99129ee513a674a7944 Mon Sep 17 00:00:00 2001 From: Yarne Coppens Date: Mon, 12 Aug 2024 09:31:27 +0200 Subject: [PATCH] Added filters --- src/filters/boardgame_filters.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 src/filters/boardgame_filters.py diff --git a/src/filters/boardgame_filters.py b/src/filters/boardgame_filters.py new file mode 100644 index 0000000..4b60707 --- /dev/null +++ b/src/filters/boardgame_filters.py @@ -0,0 +1,21 @@ +from typing import Union + +from src.classes import boardgame_classes + +def filter_expansions_out(to_filter_boardgames: list[Union[ + list[boardgame_classes.BoardGame], list[boardgame_classes.BoardGameExpansion], + list[boardgame_classes.OwnedBoardGame], list[boardgame_classes.OwnedBoardGameExpansion], + list[boardgame_classes.WishlistBoardGame], list[boardgame_classes.WishlistBoardGameExpansion]]]): + + filtered_boardgames = list(filter(lambda x: type(x) in [boardgame_classes.BoardGame, boardgame_classes.OwnedBoardGame, boardgame_classes.WishlistBoardGame], to_filter_boardgames)) + + return filtered_boardgames + +def filter_non_expansions_out(to_filter_boardgames: list[Union[ + list[boardgame_classes.BoardGame], list[boardgame_classes.BoardGameExpansion], + list[boardgame_classes.OwnedBoardGame], list[boardgame_classes.OwnedBoardGameExpansion], + list[boardgame_classes.WishlistBoardGame], list[boardgame_classes.WishlistBoardGameExpansion]]]): + + filtered_boardgames = list(filter(lambda x: type(x) in [boardgame_classes.BoardGameExpansion, boardgame_classes.OwnedBoardGameExpansion, boardgame_classes.WishlistBoardGameExpansion], to_filter_boardgames)) + + return filtered_boardgames \ No newline at end of file