import datetime import json from peewee import * import resources.Environment as Env from resources import phrases from src.model.BaseModel import BaseModel from src.model.GroupChat import GroupChat from src.model.User import User from src.model.enums.GameStatus import GameStatus from src.model.enums.SavedMediaName import SavedMediaName from src.model.game.GameBoard import GameBoard from src.model.game.GameDifficulty import GameDifficulty from src.model.game.GameOutcome import GameOutcome from src.model.game.GameType import GameType from src.model.wiki.Character import Character from src.model.wiki.Terminology import Terminology from src.utils.string_utils import get_belly_formatted class Game(BaseModel): """ Game class """ challenger: User | ForeignKeyField = ForeignKeyField( User, backref="game_challengers", on_delete="CASCADE", on_update="CASCADE" ) opponent: User | ForeignKeyField = ForeignKeyField( User, backref="game_opponents", on_delete="CASCADE", on_update="CASCADE", null=True ) type: GameType | SmallIntegerField = SmallIntegerField(null=True) # Changed from CharField(max_length=9999) to TextField to fix the MySQL 65535 byte row limit error board: str | TextField = TextField(null=True) opponent_board: str | TextField = TextField(null=True) status: GameStatus | SmallIntegerField = SmallIntegerField( default=GameStatus.AWAITING_SELECTION ) group_chat: GroupChat | ForeignKeyField = ForeignKeyField( GroupChat, null=True, backref="game_groups_chats", on_delete="RESTRICT", on_update="CASCADE", ) message_id: int | IntegerField = IntegerField(null=True) challenger_message_id: int | IntegerField = IntegerField(null=True) opponent_message_id: int | IntegerField = IntegerField(null=True) wager: int | BigIntegerField = BigIntegerField(null=True) last_interaction_date: datetime.datetime | DateTimeField = DateTimeField( default=datetime.datetime.now ) global_challenger_start_date: datetime.datetime | DateTimeField = DateTimeField(null=True) global_challenger_end_date: datetime.datetime | DateTimeField = DateTimeField(null=True) global_opponent_start_date: datetime.datetime | DateTimeField = DateTimeField(null=True) global_opponent_end_date: datetime.datetime | DateTimeField = DateTimeField(null=True) global_challenger_has_finished: bool | BooleanField = BooleanField(default=False) last_hint_date = DateTimeField(null=True) last_hint_opponent_date = DateTimeField(null=True) thread_restart_count = IntegerField(default=0) class Meta: db_table = "game" def is_player(self, user: User) -> bool: """ Check if the user is a player of the game :param user: The user :return: True if the user is a player, False otherwise """ return self.challenger == user or self.opponent == user def get_saved_media_name(self) -> SavedMediaName: """ Get the SavedMediaName for this GameType :return: The SavedMediaName """ return GameType(self.type).get_saved_media_name() def get_difficulty(self) -> GameDifficulty: """ Get the difficulty level of the game :return: The difficulty level """ return GameDifficulty.get_from_total_wager(int(str(self.wager))) def get_name(self) -> str: """ Get the name of the game :return: The name """ return GameType(self.type).get_name() @staticmethod def get_total_win_or_loss_or_draw(user: User, status: GameStatus) -> int: """ Get the total amount of wins or losses or draws a user has param user: The user param status: The status (won or lost or draw) return: The total amount of wins or losses or draws """ if status in (GameStatus.WON, GameStatus.LOST): return ( Game().select().where((Game.challenger == user) & (Game.status == status)).count() + Game() .select() .where((Game.opponent == user) & (Game.status == status.get_opposite_status())) .count() ) return Game().select().where((Game.challenger == user) & (Game.status == status)).count() @staticmethod def get_total_belly_won_or_lost(user: User, status: GameStatus) -> int: """ Get the total amount of belly a user has won or lost param user: The user param status: The status (won or lost) return: The total amount of belly """ return ( Game.select(fn.SUM(Game.wager)).where( (Game.challenger == user) & (Game.status == status) ) ).scalar() + ( Game.select(fn.SUM(Game.wager)).where( (Game.opponent == user) & (Game.status == status.get_opposite_status()) ) ).scalar() @staticmethod def get_max_won_or_lost(user: User, status: GameStatus) -> "Game": """ Get the game with the max belly won or lost param user: The user param status: The status (won or lost) return: The game """ # Max game as challenger max_game_as_challenger: Game = ( Game() .select() .where((Game.challenger == user) & (Game.status == status)) .order_by(Game.wager.desc()) .first() ) # Max game as opponent max_game_as_opponent: Game = ( Game() .select() .where((Game.opponent == user) & (Game.status == status.get_opposite_status())) .order_by(Game.wager.desc()) .first() ) if max_game_as_challenger is None: return max_game_as_opponent if max_game_as_opponent is None: return max_game_as_challenger if max_game_as_challenger.wager > max_game_as_opponent.wager: return max_game_as_challenger return max_game_as_opponent @staticmethod def get_most_challenged_user(user: User) -> (User, int): """ Get the most challenged user and the amount of games param user: The user return: The most challenged user and the amount of games """ # noinspection DuplicatedCode most_challenged_user_as_challenger = ( Game.select(Game.opponent, fn.COUNT(Game.opponent).alias("count")) .where(Game.challenger == user) .group_by(Game.opponent) .order_by(SQL("count").desc()) .first() ) amount_as_challenger = most_challenged_user_as_challenger.count most_challenged_user_as_opponent = ( Game.select(Game.challenger, fn.COUNT(Game.challenger).alias("count")) .where(Game.opponent == user) .group_by(Game.challenger) .order_by(SQL("count").desc()) .first() ) amount_as_opponent = most_challenged_user_as_opponent.count if amount_as_challenger > amount_as_opponent: return most_challenged_user_as_challenger.opponent, amount_as_challenger return most_challenged_user_as_opponent.challenger, amount_as_opponent @staticmethod def get_most_played_game(user: User) -> (GameType, int): """ Get the most played game and the amount of games param user: The user return: The most played game and the amount of games """ # noinspection DuplicatedCode most_played_as_challenger = ( Game.select(Game.type, fn.COUNT(Game.type).alias("count")) .where(Game.challenger == user) .group_by(Game.type) .order_by(SQL("count").desc()) .first() ) amount_as_challenger = most_played_as_challenger.count most_played_as_opponent = ( Game.select(Game.type, fn.COUNT(Game.type).alias("count")) .where(Game.opponent == user) .group_by(Game.type) .order_by(SQL("count").desc()) .first() ) amount_as_opponent = most_played_as_opponent.count if amount_as_challenger > amount_as_opponent: return GameType(most_played_as_challenger.type), amount_as_challenger return GameType(most_played_as_opponent.type), amount_as_opponent def get_type(self) -> GameType: """ Get the GameType :return: The GameType """ return GameType(self.type) def get_status(self) -> GameStatus: """ Get the status of the game :return: The status of the game """ return GameStatus(self.status) def get_players(self) -> list[User]: """ Get the players :return: The players """ return [self.challenger, self.opponent] def is_global(self) -> bool: """ Check if the game is global :return: True if the game is global, False otherwise """ return self.global_challenger_start_date is not None @staticmethod def get_global_games() -> list["Game"]: """ Get all global games that are in progress and an opponent has not being found yet, first those of which the challenger has finished, then oldest :return: The global games """ return ( Game.select() .where( ( Game.global_challenger_start_date.is_null(False) & (Game.global_opponent_start_date.is_null()) & (Game.status == GameStatus.IN_PROGRESS) ) ) .order_by( Game.global_challenger_has_finished.desc(), Game.global_challenger_start_date ) ) def get_wager_formatted(self) -> str: """ Get the formatted wager :return: The formatted wager """ return get_belly_formatted(self.wager) def get_half_wager_formatted(self) -> str: """ Get the formatted half wager :return: The formatted half wager """ return get_belly_formatted(self.wager // 2) def is_challenger(self, user: User) -> bool: """ Check if the user is the challenger :param user: The user :return: True if the user is the challenger, False otherwise """ return user == self.challenger def is_opponent(self, user: User) -> bool: """ Check if the user is the opponent :param user: The user :return: True if the user is the opponent, False otherwise """ return not self.is_challenger(user) def get_other_player(self, user: User) -> User: """ Get the other player :param user: The user :return: The other player """ if self.is_challenger(user): return self.opponent return self.challenger def get_winner(self) -> User | None: """ Get the winner of the game :return: The winner """ if self.get_status() is GameStatus.WON: return self.challenger if self.get_status() is GameStatus.LOST: return self.opponent return None def has_winner(self) -> bool: """ Check if the game has a winner :return: True if the game has a winner, False otherwise """ return self.get_winner() is not None def get_outcome(self) -> GameOutcome: """ Get the outcome of the game :return: The outcome """ if self.get_status() is GameStatus.WON: return GameOutcome.CHALLENGER_WON if self.get_status() is GameStatus.LOST: return GameOutcome.OPPONENT_WON if self.get_status() is GameStatus.DRAW: return GameOutcome.DRAW return GameOutcome.NONE def is_finished(self) -> bool: """ Check if the game is finished :return: True if the game is finished, False otherwise """ return self.get_status().is_finished() def set_board(self, board: GameBoard, user: User) -> None: """ Set the board for the user :param board: The game board :param user: The user :return: None """ if self.is_challenger(user) or not self.is_global(): self.board = board.get_as_json_string() else: self.opponent_board = board.get_as_json_string() def is_turn_based(self) -> bool: """ Check if the game is turn based :return: True if the game is turn based, False otherwise """ return self.get_type().is_turn_based() def is_guess_based(self) -> bool: """ Check if the game is time based :return: True if the game is time based, False otherwise """ return self.get_type().is_guess_based() def has_opponent(self) -> bool: """ Check if the game has an opponent :return: True if the game has an opponent, False otherwise """ return self.opponent is not None def get_players_time_seconds(self) -> [int, int]: """ Get the players time in seconds :return: The players time in seconds """ from src.service.date_service import get_remaining_time_in_seconds challenger_total_time_seconds = get_remaining_time_in_seconds( ( self.global_challenger_end_date if self.global_challenger_end_date else datetime.datetime.now() ), start_datetime=self.global_challenger_start_date, ) opponent_total_time_seconds = get_remaining_time_in_seconds( ( self.global_opponent_end_date if self.global_opponent_end_date else datetime.datetime.now() ), start_datetime=self.global_opponent_start_date, ) return challenger_total_time_seconds, opponent_total_time_seconds def challenger_has_finished(self) -> bool: """ Check if the challenger has finished :return: True if the challenger has finished, False otherwise """ return self.is_global() and ( self.global_challenger_end_date is not None or self.is_finished() ) def opponent_has_finished(self) -> bool: """ Check if the opponent has finished :return: True if the opponent has finished, False otherwise """ return self.is_global() and ( self.global_opponent_end_date is not None or self.is_finished() ) def get_challenger_time_seconds(self) -> int: """ Get the challenger time in seconds :return: The challenger time in seconds """ def get_seconds_for_every_hint(self) -> int: """ Get the seconds for every hint :return: The seconds for every hint """ match self.get_type(): case GameType.GUESS_OR_LIFE: return Env.GUESS_OR_LIFE_NEW_LIFE_WAIT_TIME.get_int() case GameType.PUNK_RECORDS: return Env.PUNK_RECORDS_NEXT_DETAIL_WAIT_TIME.get_int() case GameType.SHAMBLES: return Env.SHAMBLES_NEXT_LEVEL_WAIT_TIME.get_int() case GameType.WHOS_WHO: return Env.WHOS_WHO_NEXT_LEVEL_WAIT_TIME.get_int() case _: raise ValueError(f"Unsupported game type: {self.get_type()}") def get_terminology(self) -> Terminology: """ Get the terminology from the game :return: The terminology """ # Parse the JSON string and create a Terminology object json_dict = json.loads(self.board) if "terminology" in json_dict: term_dict = json_dict.pop("terminology") terminology: Terminology = Terminology(**term_dict) elif "character" in json_dict: char_dict = json_dict.pop("character") terminology: Terminology = Character(**char_dict) else: raise ValueError("No terminology or character in JSON string") return terminology def get_global_item_text(self) -> str: """ Get the global item text :return: The global item text """ return phrases.GAME_GLOBAL_ITEM_TEXT.format( self.get_type().get_name(), self.get_wager_formatted() ) Game.create_table() # Auto-migrate: index status. Queried every minute (restart_hint_thread_if_down_all_games) and # every few minutes (end_inactive_games) with WHERE Game.status == ... and no index, forcing a # full table scan on the games table on every single run, 24/7. try: from playhouse.migrate import MySQLMigrator, migrate as pw_migrate from src.model.BaseModel import db_obj _migrator = MySQLMigrator(db_obj.get_db()) pw_migrate( _migrator.add_index("game", ("status",), unique=False), ) except Exception: pass # Index already exists