40 lines
1.1 KiB
Python
40 lines
1.1 KiB
Python
from typing import List, Set, Dict, Tuple, Optional
|
|
import random
|
|
import cards
|
|
|
|
class Deck:
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.cards: List[cards.Card] = random.shuffle(cards.get_starting_deck())
|
|
self.scrap_pile: List[cards.Card] = []
|
|
|
|
def peek(self, n_cards: int) -> list:
|
|
return self.cards[:n_cards]
|
|
|
|
def peek_scrap_pile(self,) -> cards.Card:
|
|
if len(self.scrap_pile) > 0:
|
|
return self.scrap_pile[-1]
|
|
else:
|
|
return None
|
|
|
|
def pick_and_scrap(self) -> cards.Card:
|
|
card = self.cards.pop(0)
|
|
self.scrap_pile.append(card)
|
|
return card
|
|
|
|
def draw(self) -> cards.Card:
|
|
card = self.cards.pop(0)
|
|
if len(self.cards) == 0:
|
|
self.cards = self.scrap_pile[:-1].copy()
|
|
random.shuffle(self.cards)
|
|
self.scrap_pile = self.scrap_pile[-1:]
|
|
return card
|
|
|
|
def draw_from_scrap_pile(self) -> cards.Card:
|
|
if len(self.scrap_pile) > 0:
|
|
return self.scrap_pile.pop(0)
|
|
else:
|
|
return self.draw()
|
|
|
|
def scrap(self, card: cards.Card):
|
|
self.scrap_pile.append(card) |