import random from typing import List, Optional, Tuple from swiftstory.player import Player class Board: ''' This class automatically handles the reshuffling of different deck/heap of cards ''' def __init__(self) -> None: self.white_pick: List[Tuple[int, str]] = [] self.black_pick: List[Tuple[int, str]] = [] self.white_recycled: List[Tuple[int, str]] = [] self.black_recycled: List[Tuple[int, str]] = [] self.current_black_card: Optional[Tuple[int, str]] = None self.played_cards: List[Tuple[Tuple[int, str], Player]] = [] def reveal_next_black_card(self) -> None: if self.current_black_card is not None: self.black_recycled.append(self.current_black_card) self.current_black_card = None if not self.black_pick: # If we have no more cards in the deck, we need to reform one using # the cards that have been recycled. random.shuffle(self.black_recycled) self.black_pick, self.black_recycled = self.black_recycled, list() self.current_black_card = self.black_pick.pop() def pick_white_card(self) -> Tuple[int, str]: if not self.white_pick: # If we have no more cards in the deck, we need to reform one using # the cards that have been recycled. random.shuffle(self.white_recycled) self.white_pick, self.white_recycled = self.white_recycled, list() return self.white_pick.pop() def play_card(self, player: Player, card: Tuple[int, str]) -> None: self.played_cards.append((card, player)) def shuffle_played_cards(self) -> None: random.shuffle(self.played_cards) def recycle_played_cards(self) -> None: self.white_recycled += [i[0] for i in self.played_cards] self.played_cards = []