Changeset 5 for hand.py

Timestamp:

01/23/08 08:23:37 (4 years ago)

Author:

mike

Message:

Wrote hand detection code
now, just need to write comparison code.

Files:

• hand.py (modified) (4 diffs)

hand.py

@@ -1 +1 @@
 import deck
 
+hand_orders = ["straight_flush",
+                "four_of_a_kind",
+                "full_house",
+                "flush",
+                "straight",
+                "three_of_a_kind",
+                "two_pair",
+                "pair",
+                "high_card"]
+
+#returns the best combination for the 5 cards
+def judge_hand(cards):
+    ret = is_straight_flush(cards)
+    if ret: return ret
+    else: return is_n_kinds([a for (a,b) in cards])
 
 #for all these functions, make sure cards are sorted DESCENDING!!!!
+#will return flush, straight or straight_flush
+def is_straight_flush(cards):
+    ret1 = is_flush(cards)
+    ret2 = is_straight(cards)
+    if ret1 and ret2: return ("straight_flush", [cards[0][0]])
+    elif ret1: return ret1
+    elif ret2: return ret2
+    else: return None
 
+#we don't need to call this except from is_straight_flush
 def is_flush(cards):
     last=cards[0][1]
@@ -11 +35 @@
             flush = False
             break
-    if flush: return ["flush", cards[0][0]] #return highest card
+    if flush: return ("flush", [cards[0][0]]) #return highest card
     else: return None
 
+#only is_straight_flush needs to call this
 def is_straight(cards):
     last=cards[0][0]
@@ -20 +45 @@
         if a != last-1:
             straight = False
-            last = a
             break
-    if straight: return ["straight", cards[0][0]] #return highest card
+        last = a
+    if straight: return ("straight", [cards[0][0]]) #return highest card
     else: return None
 
-def is_n_kinds(cards):
+#input: only a list of the card vals.  No suits please
+def is_n_kinds(card_vals):
     highest_n = 1   #greatest num in a row
     current_n = 1
-    highest_card = None  #highest card for greatest num in a row
-    last = cards[0][0]
-    for (a,b) in cards[1:]:
+    highest_card = card_vals[0]  #highest card for greatest num in a row
+    last = card_vals[0]
+    for a in card_vals[1:]:
         if a == last:
             current_n += 1
@@ -36 +62 @@
                 highest_n = current_n
                 highest_card = a
-    if highest_n == 1: return ["high"]+cards
+    if highest_n == 1: return ("high_card",card_vals)
     #check for full house
     elif highest_n == 2:
+        whats_left = [x for x in card_vals if x != highest_card]
         #this is the hand withotu the pairs
-        whats_left = [x for x in cards if x != highest_card]
         #now check if it's two pair
-        ret = is_n_kinds(whats_left)
-
-        #if the subset is a pair then this is a two pair
-        if ret[0] == "pair": return ["two_pair",highest_card]+ret[1:] 
-        else: return ["pair",highest_card]+whats_left
+        if len(whats_left) >= 2:
+            print highest_card
+            ret = is_n_kinds(whats_left)
+            #if the subset is a pair then this is a two pair
+            if ret[0] == "pair": return ("two_pair",[highest_card]+ret[1])
+        return ("pair",[highest_card]+whats_left)
     elif highest_n == 3:
-        whats_left = [x for x in cards if x != highest_card]
+        whats_left = [x for x in card_vals if x != highest_card]
         #now check if it's a full house
-        ret = is_n_kinds(whats_left)
-        if ret[0] == "pair": return ["full_house",highest_card]+ret[1:] 
+        if len(whats_left) >= 2:
+            ret = is_n_kinds(whats_left)
+            if ret[0] == "pair": return ("full_house",[highest_card]+ret[1]) 
         #well, it's just a 3 of a kind then
-        else: return ["three_of_a_kind",highest_card]+whats_left
+        return ("three_of_a_kind",[highest_card]+whats_left)
     elif highest_n == 4:
-        whats_left = [x for x in cards if x != highest_card]
-        return ["four_of_a_kind",highest_card]+whats_left
+        whats_left = [x for x in card_vals if x != highest_card]
+        return ("four_of_a_kind",[highest_card]+whats_left)
     #we shouldn't get to this point
     assert(False)