cards.py

"""
module to implement the deck of cards
"""

from random import shuffle
from enum import Enum
from color import RED, RESET

if __name__ == "__main__":
    print(f"{RED}Please start the game by running the main.py file{RESET}")


class Suit(Enum):
    """
    suit of cards
    """

    HEARTS = f"{RED}♥{RESET}"
    DIAMONDS = f"{RED}♦{RESET}"
    CLUBS = "♣"
    SPADES = "♠"


class Rank(Enum):
    """
    rank of cards
    """

    ACE = ("A", 11, 1)
    TWO = ("2", 2)
    THREE = ("3", 3)
    FOUR = ("4", 4)
    FIVE = ("5", 5)
    SIX = ("6", 6)
    SEVEN = ("7", 7)
    EIGHT = ("8", 8)
    NINE = ("9", 9)
    TEN = ("10", 10)
    JACK = ("J", 10)
    QUEEN = ("Q", 10)
    KING = ("K", 10)

    def __init__(self, label, *values):
        self.label = label
        self.values = values  # This will be a tuple: (2,) or (1, 11)

    def get_label(self):
        return self.label

    def get_values(self):
        return self.values

    def is_ace(self):
        return len(self.values) == 2  # Only ACE has two values


class Card:
    """
    a playing card
    """

    def __init__(self, suit, rank):
        self.suit = suit
        self.rank = rank
        self.hidden = False

    def hide(self):
        self.hidden = True
        return self

    def unhide(self):
        self.hidden = False
        return self

    def __str__(self):
        return f"{self.rank.get_label()}{self.suit.value}" if not self.hidden else "🂠"

    def __eq__(self, other):
        return self.rank == other.rank


def get_shuffled_deck():
    deck = []

    for _ in range(5):
        deck.extend(generate_the_standard_deck())

    shuffle(deck)
    return deck


def generate_the_standard_deck():
    deck = []
    for suit in Suit:
        for rank in Rank:
            deck.append(Card(suit=suit, rank=rank))
    return deck