classes.py

"""
Classes to be used in the game
Credits to DJetelina https://github.com/djetelina, using the MIT OpenSource license
"""

# -*- coding: utf-8 -*-
"""
This is an example of pyCardDeck, it's not meant to be complete poker script,
but rather a showcase of pyCardDeck's usage.
"""

import pyCardDeck
# noinspection PyCompatibility
from typing import List
from pyCardDeck.cards import PokerCard

class Player:

    def __init__(self, name: str):
        self.hand = []
        self.name = name

    def __str__(self):
        return self.name
    
    def __getHand__(self):
        return self.hand


class PokerTable:

    def __init__(self, players: List[Player]):
        self.deck = pyCardDeck.Deck(
            cards=generate_deck(),
            name='Poker deck',
            reshuffle=False)
        self.players = players
        self.table_cards = []
        print("Created a table with {} players".format(len(self.players)))

    def texas_holdem(self):
        """
        Basic Texas Hold'em game structure
        """
        print("Starting a round of Texas Hold'em")
        self.deck.shuffle()
        self.deal_cards(2)
        # Imagine pre-flop logic for betting here
        self.flop()
        # Imagine post-flop, pre-turn logic for betting here
        self.river_or_flop()
        # Imagine post-turn, pre-river logic for betting here
        self.river_or_flop()
        # Imagine some more betting and winner decision here
        self.cleanup()

    def deal_cards(self, number: int):
        """
        Dealer will go through all available players and deal them x number of cards.

        :param number:  How many cards to deal
        :type number:   int
        """
        for _ in range(0, number):
            for player in self.players:
                card = self.deck.draw()
                player.hand.append(card)
                print("Dealt {} to player {}".format(card, player))

    def flop(self):
        """
        Burns a card and then shows 3 new cards on the table
        """
        # Burn a card
        burned = self.deck.draw()
        self.deck.discard(burned)
        print("Burned a card: {}".format(burned))
        for _ in range(0, 3):
            card = self.deck.draw()
            self.table_cards.append(card)
            print("New card on the table: {}".format(card))

    def river_or_flop(self):
        """
        Burns a card and then shows 1 new card on the table
        """
        burned = self.deck.draw()
        self.deck.discard(burned)
        print("Burned a card: {}".format(burned))
        card = self.deck.draw()
        self.table_cards.append(card)
        print("New card on the table: {}".format(card))

    def cleanup(self):
        """
        Cleans up the table to gather all the cards back
        """
        for player in self.players:
            for card in player.hand:
                self.deck.discard(card)
        for card in self.table_cards:
            self.deck.discard(card)
        self.deck.shuffle_back()
        print("Cleanup done")


def generate_deck() -> List[PokerCard]:
    """
    Function that generates the deck, instead of writing down 50 cards, we use iteration
    to generate the cards for use

    :return:    List with all 50 poker playing cards
    :rtype:     List[PokerCard]
    """
    suits = ['Hearts', 'Diamonds', 'Clubs', 'Spades']
    ranks = {'A': 'Ace',
             '2': 'Two',
             '3': 'Three',
             '4': 'Four',
             '5': 'Five',
             '6': 'Six',
             '7': 'Seven',
             '8': 'Eight',
             '9': 'Nine',
             '10': 'Ten',
             'J': 'Jack',
             'Q': 'Queen',
             'K': 'King'}
    cards = []
    for suit in suits:
        for rank, name in ranks.items():
            cards.append(PokerCard(suit, rank, name))
    print('Generated deck of cards for the table')
    return cards

if __name__ == '__main__':
    table = PokerTable([Player("Jack"), Player("John"), Player("Peter")])
    table.texas_holdem()