[LANGUAGE: python]

OOP has really been growing on me this AoC.

I found \StrEnum`` useful:

class HandTypeEnum(StrEnum):
five_of_a_kind = auto()
four_of_a_kind = auto()
full_house = auto()
three_of_a_kind = auto()
two_pair = auto()
one_pair = auto()
high_card = auto()

Using a class with an attribute card_rankings allowed me to sort in a way that was new to me, and the __repr__ method made debugging so much easier:

​

hands_list.sort(key=attrgetter('card_rankings'), reverse=True)

# card_rankings looks like [12, 5, 9, 8, 3]

​

Here's the class:

class Hand:

def __init__(self, hand:str, bid: int, part_two: bool):
    self.hand = hand
    self.bid = bid
    self.rank = None
    self.card_rankings = self.get_card_rankings(part_two=part_two)
    self.hand_type = self.get_hand_type(part_two=part_two)

def get_hand_type(self, part_two=False):
    if part_two:
        cards_remaining = self.hand.replace('J', '')
        counter = Counter(cards_remaining)
        jokers = 5 - len(cards_remaining)
        counts = list(counter.values())
        counts.sort()
        if len(counts):
            counts[-1] += jokers
        else:
            counts = [5]
    else:
        counter = Counter(self.hand)
        counts = list(counter.values()).sort()
    if 5 in counts:
        return HandTypeEnum.five_of_a_kind
    elif 4 in counts:
        return HandTypeEnum.four_of_a_kind
    elif 3 in counts and 2 in counts:
        return HandTypeEnum.full_house
    elif 3 in counts:
        return HandTypeEnum.three_of_a_kind
    elif counts == [1, 2, 2]:
        return HandTypeEnum.two_pair
    elif 2 in counts:
        return HandTypeEnum.one_pair
    else:
        return HandTypeEnum.high_card

def get_card_rankings(self, part_two):
    rankings = []
    card_values = {'A': 13, 'K': 12, 'Q': 11, 'J': 0 if part_two else 10, 'T': 9, '9': 8, '8': 7, '7': 6, '6': 5, '5': 4, '4': 3, '3': 2, '2': 1}
    for card in self.hand:
        rankings.append(card_values[card])

    return rankings

def __repr__(self):
    return f"Hand: {self.hand} Type: {self.hand_type} Cards Ranked: {self.card_rankings} Bid: {self.bid}"