57 lines
1.2 KiB
Python
57 lines
1.2 KiB
Python
from fastapi import Body, FastAPI
|
|
from loguru import logger
|
|
|
|
from poker.constants import Suit, Value
|
|
from poker.models import Card, Hand
|
|
from poker.rank import rank_hand
|
|
|
|
app = FastAPI()
|
|
|
|
_CARD_PATTERN = r"\s".join(5 * ["(2|3|4|5|6|7|8|9|10|J|K|Q|A)[CDHS]"])
|
|
_SUIT_MAP = {
|
|
"C": Suit.CLUBS,
|
|
"D": Suit.DIAMONDS,
|
|
"H": Suit.HEARTS,
|
|
"S": Suit.SPADES,
|
|
}
|
|
_VALUE_MAP = {
|
|
"2": Value.TWO,
|
|
"3": Value.THREE,
|
|
"4": Value.FOUR,
|
|
"5": Value.FIVE,
|
|
"6": Value.SIX,
|
|
"7": Value.SEVEN,
|
|
"8": Value.EIGHT,
|
|
"9": Value.NINE,
|
|
"10": Value.TEN,
|
|
"J": Value.JACK,
|
|
"Q": Value.QUEEN,
|
|
"K": Value.KING,
|
|
"A": Value.ACE,
|
|
}
|
|
|
|
|
|
@app.get("/")
|
|
async def health_check() -> str:
|
|
"""Health check endpoint."""
|
|
return "OK"
|
|
|
|
|
|
@app.post("/rank")
|
|
def rank(body: str = Body(regex=_CARD_PATTERN, example="2H 3D 5S 10C KD")) -> str:
|
|
"""Rank hand.
|
|
|
|
TODO: Improve error response.
|
|
"""
|
|
logger.info(f"Input hand: {body}")
|
|
cards = [
|
|
Card(suit=_SUIT_MAP[card[-1]], value=_VALUE_MAP[card[:-1]])
|
|
for card in body.split()
|
|
]
|
|
|
|
ranked_hand = rank_hand(Hand(cards=cards))
|
|
|
|
logger.info(f"Hand rank: {ranked_hand.rank}")
|
|
|
|
return ranked_hand.description
|