ahtcg_discord_bot/validate_decks.py

33 lines
1.2 KiB
Python
Raw Normal View History

from collections import defaultdict
from collections.abc import Iterator
from arkhamdb import ArkhamDBCard, ArkhamDBDeck
class Validator:
cards_by_code: dict[str, ArkhamDBCard]
def __init__(self, cards: list[ArkhamDBCard]) -> None:
cards: list[ArkhamDBCard]
self.cards_by_code = {card["code"]: card for card in cards}
def validate(self, decks: list[ArkhamDBDeck]) -> Iterator[str]:
card_total_counts = defaultdict(lambda: 0)
for deck in decks:
for code, count in deck["slots"].items():
card_total_counts[code] += count
for code, count in card_total_counts.items():
try:
card = self.cards_by_code[code]
quantity = card["quantity"]
# assume two copies of the core set
if card["pack_code"] == "core":
quantity = quantity * 2
if count > quantity:
in_decks = [deck["id"] for deck in decks if code in deck["slots"]]
yield f"Too many copies of {card['name']}! ({count} > {quantity}), in decks {in_decks}"
except KeyError:
yield f'Did not find card with code "{code}"'