52 lines
1.5 KiB
Python
52 lines
1.5 KiB
Python
from functools import total_ordering
|
|
from enum import Enum
|
|
from typing import List
|
|
from poker.hand_ranking import HandType, HandRanking
|
|
from poker.card import ShortDeckRank, Card
|
|
|
|
|
|
@total_ordering
|
|
class ShortDeckHandType(Enum):
|
|
"""
|
|
标准牌型扑克与短牌型扑克的区别:
|
|
- 同花(Flush) > 满堂红(Full House)
|
|
- 其他牌型等级不变
|
|
"""
|
|
HIGH_CARD = (1, "High Card")
|
|
ONE_PAIR = (2, "One Pair")
|
|
TWO_PAIR = (3, "Two Pair")
|
|
THREE_OF_A_KIND = (4, "Three of a Kind")
|
|
STRAIGHT = (5, "Straight")
|
|
FULL_HOUSE = (6, "Full House")
|
|
FLUSH = (7, "Flush")
|
|
FOUR_OF_A_KIND = (8, "Four of a Kind")
|
|
STRAIGHT_FLUSH = (9, "Straight Flush")
|
|
ROYAL_FLUSH = (10, "Royal Flush")
|
|
|
|
def __new__(cls, strength, type_name):
|
|
obj = object.__new__(cls)
|
|
obj._value_ = strength
|
|
obj.strength = strength
|
|
obj.type_name = type_name
|
|
return obj
|
|
|
|
def __str__(self):
|
|
return self.type_name
|
|
|
|
def __eq__(self, other):
|
|
if isinstance(other, ShortDeckHandType):
|
|
return self.strength == other.strength
|
|
return NotImplemented
|
|
|
|
def __lt__(self, other):
|
|
if isinstance(other, ShortDeckHandType):
|
|
return self.strength < other.strength
|
|
return NotImplemented
|
|
|
|
|
|
class ShortDeckHandRanking(HandRanking):
|
|
def __init__(self, hand_type: ShortDeckHandType, key_ranks: List[ShortDeckRank], cards: List[Card]):
|
|
super().__init__(hand_type, key_ranks, cards)
|
|
|
|
def __str__(self):
|
|
return super().__str__() |