check_balance.py

"""
檢查錢包餘額的簡單腳本
"""
import asyncio
from config import load_config
from wallet_manager import WalletManager
from solana.rpc.async_api import AsyncClient


async def main():
    # 載入配置
    config = load_config()

    # 初始化 RPC 客戶端
    client = AsyncClient(config.rpc_endpoint)

    # 初始化錢包管理器
    wallet_manager = WalletManager(config.wallet_private_keys)

    for i, wallet in enumerate(wallet_manager, 1):
        print(f"\n{'='*60}")
        print(f"錢包 {i}: {wallet.address}")
        print(f"{'='*60}")

        # 獲取 SOL 餘額
        sol_balance = await wallet.get_sol_balance(client)
        print(f"SOL 餘額: {sol_balance:.9f} SOL")

        # 獲取 USDC 餘額
        usdc_balance = await wallet.get_usdc_balance(client, config.usdc_mint)
        print(f"USDC 餘額: {usdc_balance:.6f} USDC")

        print(f"{'='*60}\n")

    await client.close()


if __name__ == "__main__":
    asyncio.run(main())