mirror of
https://github.com/bohd4nx/FragmentAPI.git
synced 2026-07-25 06:14:29 +00:00
bohd4nx
01a5befd87
- Removed wallet transaction and transfer logic from the wallet domain. - Introduced a new tonapi module to handle transactions and balance checks. - Updated tests to reflect the new structure and ensure functionality remains intact. - Added functionality for sending TON and USDT transfers through the tonapi module. - Improved error handling and validation for payment balances. - Cleaned up and organized imports across the codebase.
92 lines
3.3 KiB
Python
92 lines
3.3 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import TYPE_CHECKING
|
|
|
|
from ton_core import Address, NetworkGlobalID
|
|
from tonutils.clients import ToncenterClient
|
|
from tonutils.contracts import JettonTransferBuilder, TONTransferBuilder
|
|
|
|
from pyfragment.core.constants import USDT_TON_MASTER_ADDRESS, WALLET_CLASSES
|
|
from pyfragment.exceptions import TransactionError, WalletError
|
|
from pyfragment.models.wallet import TonTransferResult, UsdtTransferResult
|
|
|
|
if TYPE_CHECKING:
|
|
from pyfragment.client import FragmentClient
|
|
|
|
|
|
async def send_ton_transfer(
|
|
client: FragmentClient,
|
|
destination: str,
|
|
amount: int,
|
|
body: str | None = None,
|
|
) -> TonTransferResult:
|
|
"""Send a direct TON transfer from the seeded wallet.
|
|
|
|
Args:
|
|
client: Authenticated `FragmentClient` instance.
|
|
destination: Recipient address in any TON-compatible format.
|
|
amount: Amount in nanotons.
|
|
body: Optional on-chain comment.
|
|
"""
|
|
try:
|
|
async with ToncenterClient(network=NetworkGlobalID.MAINNET) as ton:
|
|
wallet_cls = WALLET_CLASSES[client.wallet_version]
|
|
wallet, _, _, _ = wallet_cls.from_mnemonic(client=ton, mnemonic=client.seed)
|
|
result = await wallet.transfer_message(
|
|
TONTransferBuilder(
|
|
destination=Address(destination),
|
|
amount=amount,
|
|
body=body,
|
|
)
|
|
)
|
|
return TonTransferResult(
|
|
transaction_id=str(result.normalized_hash),
|
|
destination=destination,
|
|
amount=amount,
|
|
)
|
|
except (TransactionError, WalletError):
|
|
raise
|
|
except Exception as exc:
|
|
raise TransactionError(TransactionError.BROADCAST_FAILED.format(exc=exc)) from exc
|
|
|
|
|
|
async def send_usdt_transfer(
|
|
client: FragmentClient,
|
|
destination: str,
|
|
usdt_amount: int,
|
|
forward_payload: str | None = None,
|
|
ton_for_gas: int = 50_000_000,
|
|
) -> UsdtTransferResult:
|
|
"""Send a direct USDT transfer from the seeded wallet.
|
|
|
|
Args:
|
|
client: Authenticated `FragmentClient` instance.
|
|
destination: Recipient address in any TON-compatible format.
|
|
usdt_amount: Amount in USDT base units (6 decimals).
|
|
forward_payload: Optional comment passed through to the recipient.
|
|
ton_for_gas: TON attached for gas in nanotons.
|
|
"""
|
|
try:
|
|
async with ToncenterClient(network=NetworkGlobalID.MAINNET) as ton:
|
|
wallet_cls = WALLET_CLASSES[client.wallet_version]
|
|
wallet, _, _, _ = wallet_cls.from_mnemonic(client=ton, mnemonic=client.seed)
|
|
result = await wallet.transfer_message(
|
|
JettonTransferBuilder(
|
|
destination=Address(destination),
|
|
jetton_amount=usdt_amount,
|
|
jetton_master_address=Address(USDT_TON_MASTER_ADDRESS),
|
|
forward_payload=forward_payload,
|
|
forward_amount=1,
|
|
amount=ton_for_gas,
|
|
)
|
|
)
|
|
return UsdtTransferResult(
|
|
transaction_id=str(result.normalized_hash),
|
|
destination=destination,
|
|
amount=usdt_amount,
|
|
)
|
|
except (TransactionError, WalletError):
|
|
raise
|
|
except Exception as exc:
|
|
raise TransactionError(TransactionError.BROADCAST_FAILED.format(exc=exc)) from exc
|