FragmentAPI/pyfragment/types/results.py

import time
from dataclasses import dataclass, field

__all__ = ["AdsTopupResult", "PremiumResult", "StarsResult", "WalletInfo"]


@dataclass
class WalletInfo:
    """Wallet state returned by :meth:`FragmentClient.get_wallet`."""

    address: str
    state: str
    balance: float

    def __repr__(self) -> str:
        return f"WalletInfo(address='{self.address}', state='{self.state}', balance={self.balance} TON)"


@dataclass
class PremiumResult:
    """Result of a successful Telegram Premium gift."""

    transaction_id: str
    username: str
    months: int
    timestamp: int = field(default_factory=lambda: int(time.time()))

    def __repr__(self) -> str:
        return f"PremiumResult(username='{self.username}', months={self.months}, tx='{self.transaction_id}')"


@dataclass
class StarsResult:
    """Result of a successful Telegram Stars purchase."""

    transaction_id: str
    username: str
    stars: int
    timestamp: int = field(default_factory=lambda: int(time.time()))

    def __repr__(self) -> str:
        return f"StarsResult(username='{self.username}', stars={self.stars}, tx='{self.transaction_id}')"


@dataclass
class AdsTopupResult:
    """Result of a successful Telegram Ads balance top-up."""

    transaction_id: str
    username: str
    amount: int
    timestamp: int = field(default_factory=lambda: int(time.time()))

    def __repr__(self) -> str:
        return f"AdsTopupResult(username='{self.username}', amount={self.amount} TON, tx='{self.transaction_id}')"