import operator import random class InvalidCard(Exception): """Invalid Card Exception Thrown if an invalid face value or suit is supplied for a card """ pass class InvalidHand(Exception): """Invalid Hand Exception Thrown if the hand being created includes zero or more than 5 cards """ pass def unique_combinations(items, n): if n==0: yield [] else: for i in xrange(len(items)): for cc in unique_combinations(items[i+1:],n-1): yield [items[i]]+cc class Card: """Represents a single poker playing card""" VALUES = {'T': 10, 'J': 11, 'Q': 12, 'K': 13, 'A': 14} SUITS = {'Hearts': 'H', 'Diamonds': 'D', 'Clubs': 'C', 'Spades': 'S'} def __init__(self, string): self.value = string[0] if self.value in '23456789': self.value = int(self.value) else: try: self.value = Card.VALUES[self.value] except KeyError as e: raise InvalidCard('Invalid value: {0}'.format(string)) self.suit = string[1] if not self.suit in Card.SUITS.values(): raise InvalidCard('Invalid suit: {0}'.format(string)) def __cmp__(self, other): """Compare hand values Compares this card's value with another. Used for sorting. """ result = cmp(self.value, other.value) if result == 0: """Values are identical, suits differ. Doesn't affect ranking in any way.""" result = cmp(self.suit, other.suit) return result def __repr__(self): """Builds a string representation of the hand""" val = self.value for k, v in Card.VALUES.iteritems(): if self.value == v: val = k return str.format('{0}{1}', val, self.suit) class Deck(): def __init__(self): self.__cards = [] for suit in Card.SUITS.values(): for i in range(2,15): if i >= 10: for k, v in Card.VALUES.iteritems(): if i == v: value = k else: value = i self.__cards.append(Card('{0}{1}'.format(value, suit))) def shuffle(self): random.shuffle(self.__cards) def cards(self): return self.__cards def deal(self, n=1, players=[]): if players: for i in range(n): for player in players: player.add_card(self.__cards.pop()) else: cards = [] for i in range(n): cards.append(self.__cards.pop()) return cards class Hand: HIGH_CARD = 0 ONE_PAIR = 1 TWO_PAIRS = 2 THREE_OF_A_KIND = 3 STRAIGHT = 4 FLUSH = 5 FULL_HOUSE = 6 FOUR_OF_A_KIND = 7 STRAIGHT_FLUSH = 8 ROYAL_FLUSH = 9 RANKS = [ 'High Card', 'One Pair', 'Two Pairs', 'Three of a Kind', 'Straight', 'Flush', 'Full House', 'Four of a Kind', 'Straight Flush', 'Royal Flush' ] def __init__(self, cards): if len(cards) < 1 or len(cards) > 5: raise InvalidHand(cards) self.__rank = None self.__cards = sorted([Card(c) for c in cards], reverse=True) self.__values = [] self.rank() def __repr__(self): """Builds a string representation of the hand""" return str.format("Cards: {0} Rank: '{1}' Values: {2}", self.__cards, Hand.RANKS[self.rank()], self.values()) def rank(self): """Get the hand rank Determines the rank of the poker hand if it has not already been computed, and returns it. """ if self.__rank: return self.__rank flush = True straight = False last = None merged = {} for c in self.__cards: if last: if flush and c.suit != last.suit: flush = False last = c if c.value in merged: merged[c.value] = merged[c.value] + 1 else: merged[c.value] = 1 if (len(merged)) == 5: # All unique cards, check for a straight if self.__cards[0].value - self.__cards[4].value == 4: straight = True if self.__cards[4].value == 2 and self.__cards[1].value == 5 and self.__cards[0].value == 14: straight = True # Set the value of the ace to 1 and resort so hand comparisons work correctly self.__cards[0].value = 1 self.__cards = sorted(self.__cards, reverse=True) if straight and flush: if self.__cards[0].value == 14: self.__rank = Hand.ROYAL_FLUSH else: self.__rank = Hand.STRAIGHT_FLUSH elif flush: self.__rank = Hand.FLUSH elif straight: self.__rank = Hand.STRAIGHT else: self.__rank = Hand.HIGH_CARD self.__values = [c.value for c in self.__cards] else: multiples = [m for m in sorted(merged.items(), key = operator.itemgetter(1), reverse = True) if m[1] > 1] if len(multiples) > 1: if multiples[0][1] == multiples[1][1]: self.__rank = Hand.TWO_PAIRS else: self.__rank = Hand.FULL_HOUSE elif multiples: if multiples[0][1] > 3: self.__rank = Hand.FOUR_OF_A_KIND elif multiples[0][1] == 3: self.__rank = Hand.THREE_OF_A_KIND else: self.__rank = Hand.ONE_PAIR mvalues = sorted([m[0] for m in multiples], reverse=True) self.__values = mvalues + [c.value for c in self.__cards if c.value not in mvalues] if not self.__rank: self.__rank = Hand.HIGH_CARD return self.__rank def values(self): """Returns a list of card values for the hand Values for sets are provided first in order of importance and descending strength, followed by all additional card values in order of descending strength. Used for comparison and sorting. Examples: Full House ['5S', '5D', '5H', '8S', '8C'] Values = [5, 8] Two Pair ['9D', '9S', '5H', '5C', 'KH'] Values = [9, 5, 13] Straight ['AS', '2H', '3C', '4C', '5H'] Values = [5, 4, 3, 2, 1] """ if not self.__values: self.rank() return self.__values def cards(self): return self.__cards @staticmethod def create_best_hand(cards): """Create the strongest possible poker hand Using the supplied cards, this will build the strongest possible five- card poker hand. Uses smart hand creation if more than five cards are specified. """ if len(cards) <= 5: return Hand(cards) else: for hand in Hand.create_best_hand_smart(cards): return hand return false @staticmethod def create_best_hand_bruteforce(cards): """Create the strongest possible poker hand Builds every possible poker hand from the supplied set of cards, and returns the strongest result. """ combos = unique_combinations(cards, 5) hands = [Hand(combo) for combo in combos] hands = sorted(hands, reverse=True) return hands[0] @staticmethod def create_best_hand_smart(cards): """Create the strongest possible poker hand Intelligently seeks out the strongest possible poker hand from the supplied set of cards using the following steps: * Find all flushes * Find all straights ** Return best hand present in both flushes and straights, if applicable * Find all sets ** Return best quads with top remaining card ** Find and return best full house ** Find and return best three of a kind with top remaining cards ** Find and return best two pair with top remaining cards ** Find and return best single pair with top remaining cards ** Return top 5 cards """ cards = sorted([Card(c) for c in cards], reverse=True) # Get all flushes flushes = [] for suit in Card.SUITS.values(): suited = [str(c) for c in cards if c.suit == suit] if len(suited) >= 5: combos = unique_combinations(suited, 5) for combo in combos: flushes.append(Hand(combo)) flushes = sorted(flushes, reverse=True) if (flushes and flushes[0].rank() >= Hand.STRAIGHT_FLUSH): # Straight flush! No need to check anything else yield flushes[0] #Get all sets merged = {} for c in cards: if c.value in merged: merged[c.value] = merged[c.value] + 1 else: merged[c.value] = 1 multiples = [m for m in sorted(merged.items(), key = operator.itemgetter(1), reverse = True) if m[1] > 1] quads = [c[0] for c in multiples if c[1] == 4] quads = [c for c in cards if c.value in quads] trips = [c[0] for c in multiples if c[1] == 3] trips = [c for c in cards if c.value in trips] pairs = [c[0] for c in multiples if c[1] == 2] pairs = [c for c in cards if c.value in pairs] remaining = [c for c in cards if c.value not in [m[0] for m in multiples]] if quads: h = quads[:4] remaining = [c for c in cards if c.value not in [cc.value for cc in h]][:1] for r in remaining: h.append(r) yield Hand([str(c) for c in h]) if trips and pairs: # Get a full house together h = trips[:3] remaining = pairs[:2] for r in remaining: h.append(r) yield Hand([str(c) for c in h]) if flushes: # We've already got a flush, return it! yield flushes[0] # Look for a straight! mvals = sorted(merged.keys(), reverse=True) for i in range(0, len(mvals) -4, 1): if (mvals[i] - mvals[i + 4]) == 4: # Regular straight h = [[c for c in cards if c.value == v][0] for v in mvals[i:i + 5]] yield Hand([str(c) for c in h]) elif 14 in [c.value for c in cards] and mvals[i + 1] == 5 and mvals[i + 4] == 2: # Ace low straight h = [[c for c in cards if c.value == v][0] for v in mvals[i + 1:i + 5]] h.append([c for c in cards if c.value == 14][0]) yield Hand([str(c) for c in h]) if trips: h = trips[:3] remaining = [c for c in cards if c.value not in [cc.value for cc in h]][:2] for r in remaining: h.append(r) yield Hand([str(c) for c in h]) if pairs: if len(pairs) > 2: h = pairs[:4] remaining = [c for c in cards if c.value not in [cc.value for cc in h]][:1] for r in remaining: h.append(r) yield Hand([str(c) for c in h]) else: h = pairs remaining = [c for c in cards if c.value not in [cc.value for cc in h]][:3] for r in remaining: h.append(r) yield Hand([str(c) for c in h]) # High card, send the top 5 reverse-sorted cards yield Hand([str(c) for c in cards[:5]]) def __cmp__(self, other): """Compare hand rankings Compares this hand with another by first checking rank, and if they are equal in that regard, by their card values. Used for sorting. """ result = cmp(self.rank(), other.rank()) if (result == 0): # Compare hand values for i in range(len(self.values())): result = cmp(self.values()[i], other.values()[i]) if (result != 0): return result return result class ChinesePokerHand: def __init__(self, cards): if len(cards) != 13: raise InvalidHand(); self.__cards = cards self.build_hands() def build_hands(self): self.build_hands_dumb() def build_hands_dumb(self): self.__bottom = Hand.create_best_hand([str(c) for c in self.__cards]) self.__middle = Hand.create_best_hand([str(c) for c in self.__cards if c not in self.__bottom.cards()]) self.__top = Hand.create_best_hand([str(c) for c in self.__cards if c not in (self.__bottom.cards() + self.__middle.cards())]) def __repr__(self): return 'Chinese Poker Hand:\n\tTop: {0}\n\tMiddle: {1}\n\tBottom: {2}'.format(self.__top, self.__middle, self.__bottom) class Player: def __init__(self, name): self.__name = name self.__hand = None self.__cards = [] self.__community_cards = [] def add_card(self, card, community=False): if community: self.__community_cards.append(card) else: self.__cards.append(card) # Rebuild and re-evaluate hand self.__hand = Hand.create_best_hand([str(c) for c in self.__community_cards + self.__cards]) def discard(self, index): self.__cards.pop(index) self.__hand = Hand.create_best_hand([str(c) for c in self.__community_cards + self.__cards]) def hand(self): return self.__hand def __repr__(self): return '"{0}"\n\tCards: {1}\n\tHand: {2}'.format(self.__name, self.__cards, self.__hand) def __cmp__(self, other): return cmp(self.hand(), other.hand()) if __name__ == '__main__': deck = Deck() deck.shuffle() players = [] for i in range(1,10): players.append(Player('Player {0}'.format(i))) community_cards = [] deck.deal(2, players) deck.deal() community_cards.extend(deck.deal(3)) deck.deal() community_cards.extend(deck.deal()) deck.deal() community_cards.extend(deck.deal()) for player in players: for card in community_cards: player.add_card(card, True) players = sorted(players, reverse=True) print 'Community', community_cards for player in players: print player print '\nChinese Poker Example' deck = Deck() deck.shuffle() for i in range(4): cards = deck.deal(13) chinese = ChinesePokerHand(cards) print chinese