| from peewee import * |
|
|
| import constants as c |
| 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 |
|
|
|
|
| class DocQGame(BaseModel): |
| """ |
| Doc Q Game class |
| """ |
|
|
| user: User | ForeignKeyField = ForeignKeyField( |
| User, backref="doc_q_game_users", on_delete="CASCADE", on_update="CASCADE" |
| ) |
| status: GameStatus | SmallIntegerField = SmallIntegerField(default=GameStatus.IN_PROGRESS) |
| correct_choices_index: str | CharField = CharField(max_length=99, null=True) |
| group_chat: GroupChat | ForeignKeyField = ForeignKeyField( |
| GroupChat, |
| null=True, |
| backref="doc_q_game_groups_chats", |
| on_delete="RESTRICT", |
| on_update="CASCADE", |
| ) |
| message_id: int | IntegerField = IntegerField(null=True) |
| belly: int | BigIntegerField = BigIntegerField(null=True) |
|
|
| class Meta: |
| db_table = "doc_q_game" |
|
|
| @staticmethod |
| def get_total_win_or_loss(user: User, status: GameStatus) -> int: |
| """ |
| Get the total amount of wins or losses a user has |
| param user: The user |
| param status: The status (won or lost) |
| return: The total amount of wins or losses |
| """ |
|
|
| return ( |
| DocQGame() |
| .select() |
| .where((DocQGame.user == user) & (DocQGame.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 won or lost |
| """ |
|
|
| return ( |
| DocQGame() |
| .select(fn.SUM(DocQGame.belly)) |
| .where((DocQGame.user == user) & (DocQGame.status == status)) |
| ).scalar() |
|
|
| @staticmethod |
| def get_max_won_or_lost(user: User, status: GameStatus) -> "DocQGame": |
| """ |
| Get the DocQGame with the max amount of belly won or lost |
| param user: The user |
| param status: The status (won or lost) |
| return: The max amount of belly won or lost |
| """ |
|
|
| return ( |
| DocQGame() |
| .select() |
| .where((DocQGame.user == user) & (DocQGame.status == status)) |
| .order_by(DocQGame.belly.desc()) |
| .first() |
| ) |
|
|
| def get_status(self) -> GameStatus: |
| """ |
| Get the status of the game |
| :return: The status of the game |
| """ |
|
|
| return GameStatus(self.status) |
|
|
| def get_correct_apple_number(self) -> int: |
| """ |
| Get the correct apple number |
| :return: The correct apple number |
| """ |
|
|
| return int((str(self.correct_choices_index).split(c.STANDARD_SPLIT_CHAR))[0]) + 1 |
|
|
|
|
| DocQGame.create_table() |
|
|
