Source code for app.services.wallet_service

from app.repository.user_repository import UserRepository
from app.repository.wallet_repository import WalletRepository
from app.schema.wallet_schema import BaseWallet, ResponsePreviewConvertPoints
from app.services.base_service import BaseService




class WalletService(BaseService):
    """
    Service class for managing wallets.

    Attributes:
        wallet_repository (WalletRepository): Repository instance for wallets.
        user_repository (UserRepository): Repository instance for users.
    """

    def __init__(
        self, wallet_repository: WalletRepository, user_repository: UserRepository
    ):
        """
        Initializes the WalletService with the provided repositories.

        Args:
            wallet_repository (WalletRepository): The wallet repository
              instance.
            user_repository (UserRepository): The user repository instance.
        """
        self.wallet_repository = wallet_repository
        self.user_repository = user_repository
        super().__init__(wallet_repository)



    def get_wallet_by_user_id(self, externalUserId):
        """
        Retrieves the wallet associated with the given user ID.

        Args:
            externalUserId (str): The external user ID.

        Returns:
            BaseWallet: The wallet details.
        """
        user = self.user_repository.read_by_column(
            column="externalUserId",
            value=externalUserId,
            not_found_message=f"User with externalUserId "
            f"{externalUserId} not found",
        )

        wallet = self.wallet_repository.read_by_column(
            column="userId",
            value=user.id,
            not_found_message=f"Wallet with userId {user.id} not found",
        )

        wallet = BaseWallet(
            id=wallet.id,
            coinsBalance=wallet.coinsBalance,
            pointsBalance=wallet.pointsBalance,
            conversionRate=wallet.conversionRate,
            externalUserId=user.externalUserId,
            created_at=wallet.created_at,
            updated_at=wallet.updated_at,
        )

        return wallet




    def preview_convert(self, schema):
        """
        Previews the conversion of points to coins for a user.

        Args:
            schema: The schema containing conversion details.

        Returns:
            ResponsePreviewConvertPoints: The conversion preview details.
        """
        user = self.user_repository.read_by_column(
            column="externalUserId",
            value=schema.externalUserId,
            not_found_message=f"User with externalUserId "
            f"{schema.externalUserId} not found",
        )

        wallet = self.wallet_repository.read_by_column(
            column="userId",
            value=user.id,
            not_found_message=f"Wallet with userId {user.id} not found",
        )

        points_converted = schema.points * wallet.conversionRate

        response = ResponsePreviewConvertPoints(
            coins=points_converted,
            points_converted=schema.points,
            conversionRate=wallet.conversionRate,
            afterConversionPoints=wallet.pointsBalance - schema.points,
            afterConversionCoins=wallet.coinsBalance + points_converted,
        )

        return response