diff --git a/app/routers_v2/cards.py b/app/routers_v2/cards.py index d56c848..683b33f 100644 --- a/app/routers_v2/cards.py +++ b/app/routers_v2/cards.py @@ -79,8 +79,8 @@ async def get_cards( all_cards = all_cards.where(Card.pack == this_pack) if value is not None: all_cards = all_cards.where(Card.value == value) - # if variant is not None: - # all_cards = all_cards.where(Card.variant == variant) + if variant is not None: + all_cards = all_cards.where(Card.variant == variant) if min_value is not None: all_cards = all_cards.where(Card.value >= min_value) if max_value is not None: @@ -114,8 +114,8 @@ async def get_cards( if csv: data_list = [ - ["id", "player", "cardset", "rarity", "team", "pack", "value"] - ] # , 'variant']] + ["id", "player", "cardset", "rarity", "team", "pack", "value", "variant"] + ] for line in all_cards: data_list.append( [ @@ -125,7 +125,8 @@ async def get_cards( line.player.rarity, line.team.abbrev, line.pack, - line.value, # line.variant + line.value, + line.variant, ] ) return_val = DataFrame(data_list).to_csv(header=False, index=False)