import enum from typing import List, Tuple from hanabi import hanab_game from database import conn_manager class GameOutcome(enum.Enum): win = 0 discard_crit = 1 bomb_crit = 2 strikeout = 3 bottom_deck = 4 vote_to_kill = 5 out_of_pace = 6 class GameAnalysisResult: def __init__(self, outcomes: List[GameOutcome], bdrs: List[Tuple[hanab_game.DeckCard, int]]): self.outcome = GameOutcome self.bdrs = bdrs def analyze_game(instance: hanab_game.HanabiInstance, actions: List[hanab_game.Action]) -> GameAnalysisResult: # List of bdrs bdrs = [] # This is the default value if we find no other reason why the game was lost (or won) outcomes = [] game = hanab_game.GameState(instance) def handle_lost_card(card, game, play: bool): if not game.is_trash(card): if game.is_critical(card): outcomes.append(GameOutcome.bomb_crit if play else GameOutcome.discard_crit) elif card.rank != 1: if card in game.deck[game.progress:]: bdrs.append((card, game.draw_pile_size)) else: if game.deck[game.progress:].count(card) == 2: bdrs.append((card, game.draw_pile_size)) for action in actions: if action.type == hanab_game.ActionType.Discard: discarded_card = instance.deck[action.target] handle_lost_card(discarded_card, game, False) if action.type == hanab_game.ActionType.Play: played_card = instance.deck[action.target] if not game.is_playable(played_card) and not game.is_trash(played_card): bombed_card = instance.deck[action.target] handle_lost_card(bombed_card, game, True) game.make_action(action) if game.pace < 0 and GameOutcome.out_of_pace not in outcomes: outcomes.append(GameOutcome.out_of_pace) if game.strikes == 3: outcomes.append(GameOutcome.strikeout) elif actions[-1].type in [hanab_game.ActionType.EndGame, hanab_game.ActionType.VoteTerminate]: outcomes.append(GameOutcome.vote_to_kill) if game.score == 5 * instance.num_suits: outcomes.append(GameOutcome.win) return GameAnalysisResult(outcomes, bdrs) def update_user_statistics(user_ids: List[int]): """ Update the cumulative user statistics for this user, assuming that the corresponding game statistics have been computed already. @param user_ids: @return: """