GramCore/services/players/services.py

from typing import List, Optional

from gram_core.base_service import BaseService
from gram_core.basemodel import RegionEnum
from gram_core.services.players.models import PlayersDataBase as Player
from gram_core.services.players.repositories import PlayersRepository

__all__ = "PlayersService"


class PlayersService(BaseService):
    def __init__(self, players_repository: PlayersRepository) -> None:
        self._repository = players_repository

    async def get(
        self,
        user_id: Optional[int] = None,
        player_id: Optional[int] = None,
        account_id: Optional[int] = None,
        region: Optional[RegionEnum] = None,
        is_chosen: Optional[bool] = None,
        offset: int = 0,
    ) -> Optional[Player]:
        return await self._repository.get(user_id, player_id, account_id, region, is_chosen, offset)

    async def get_player(
        self, user_id: int, region: Optional[RegionEnum] = None, player_id: int = None, offset: int = 0
    ) -> Optional[Player]:
        is_chosen = True
        if player_id is not None or (offset is not None and offset != 0):
            is_chosen = None
        return await self._repository.get(user_id, player_id, region=region, is_chosen=is_chosen, offset=offset)

    async def add(self, player: Player) -> None:
        await self._repository.add(player)

    async def update(self, player: Player) -> None:
        await self._repository.update(player)

    async def get_all_by_user_id(self, user_id: int) -> List[Player]:
        return await self._repository.get_all_by_user_id(user_id)

    async def remove_all_by_user_id(self, user_id: int):
        players = await self._repository.get_all_by_user_id(user_id)
        for player in players:
            await self._repository.delete(player)

    async def get_all_by_account_id(self, account_id: int, region: Optional[RegionEnum] = None) -> List[Player]:
        return await self._repository.get_all_by_account_id(account_id, region)

    async def set_player_to_main(self, user_id: int, player_id: int) -> bool:
        players = await self._repository.get_all_by_user_id(user_id)
        for player in players:
            player.is_chosen = player.player_id == player_id
            await self._repository.update(player)

    async def delete(self, player: Player):
        await self._repository.delete(player)