web3_py_client.py 26 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576
  1. import os
  2. import json
  3. import logging
  4. from decimal import Decimal, ROUND_DOWN
  5. from web3 import Web3
  6. from web3.middleware import ExtraDataToPOAMiddleware # For PoA networks like Goerli, Sepolia, BSC etc.
  7. from eth_account import Account
  8. from dotenv import load_dotenv
  9. from checker.logger_config import get_logger
  10. # 配置日志
  11. logger = get_logger('as')
  12. # 加载环境变量
  13. load_dotenv()
  14. # 标准 IERC20 ABI (只包含常用函数)
  15. IERC20_ABI = json.loads('''
  16. [
  17. {
  18. "constant": true,
  19. "inputs": [],
  20. "name": "name",
  21. "outputs": [{"name": "", "type": "string"}],
  22. "payable": false,
  23. "stateMutability": "view",
  24. "type": "function"
  25. },
  26. {
  27. "constant": false,
  28. "inputs": [
  29. {"name": "_spender", "type": "address"},
  30. {"name": "_value", "type": "uint256"}
  31. ],
  32. "name": "approve",
  33. "outputs": [{"name": "", "type": "bool"}],
  34. "payable": false,
  35. "stateMutability": "nonpayable",
  36. "type": "function"
  37. },
  38. {
  39. "constant": true,
  40. "inputs": [],
  41. "name": "totalSupply",
  42. "outputs": [{"name": "", "type": "uint256"}],
  43. "payable": false,
  44. "stateMutability": "view",
  45. "type": "function"
  46. },
  47. {
  48. "constant": false,
  49. "inputs": [
  50. {"name": "_from", "type": "address"},
  51. {"name": "_to", "type": "address"},
  52. {"name": "_value", "type": "uint256"}
  53. ],
  54. "name": "transferFrom",
  55. "outputs": [{"name": "", "type": "bool"}],
  56. "payable": false,
  57. "stateMutability": "nonpayable",
  58. "type": "function"
  59. },
  60. {
  61. "constant": true,
  62. "inputs": [],
  63. "name": "decimals",
  64. "outputs": [{"name": "", "type": "uint8"}],
  65. "payable": false,
  66. "stateMutability": "view",
  67. "type": "function"
  68. },
  69. {
  70. "constant": true,
  71. "inputs": [{"name": "_owner", "type": "address"}],
  72. "name": "balanceOf",
  73. "outputs": [{"name": "balance", "type": "uint256"}],
  74. "payable": false,
  75. "stateMutability": "view",
  76. "type": "function"
  77. },
  78. {
  79. "constant": true,
  80. "inputs": [],
  81. "name": "symbol",
  82. "outputs": [{"name": "", "type": "string"}],
  83. "payable": false,
  84. "stateMutability": "view",
  85. "type": "function"
  86. },
  87. {
  88. "constant": false,
  89. "inputs": [
  90. {"name": "_to", "type": "address"},
  91. {"name": "_value", "type": "uint256"}
  92. ],
  93. "name": "transfer",
  94. "outputs": [{"name": "", "type": "bool"}],
  95. "payable": false,
  96. "stateMutability": "nonpayable",
  97. "type": "function"
  98. },
  99. {
  100. "constant": true,
  101. "inputs": [
  102. {"name": "_owner", "type": "address"},
  103. {"name": "_spender", "type": "address"}
  104. ],
  105. "name": "allowance",
  106. "outputs": [{"name": "", "type": "uint256"}],
  107. "payable": false,
  108. "stateMutability": "view",
  109. "type": "function"
  110. }
  111. ]
  112. ''')
  113. class EthClient:
  114. def __init__(self, rpc_url: str = None, hash: str = None):
  115. self.rpc_url = rpc_url or os.getenv("RPC_URL")
  116. _hash = hash or os.getenv("HASH")
  117. if not self.rpc_url:
  118. raise ValueError("RPC_URL not provided or found in environment variables.")
  119. if not _hash:
  120. raise ValueError("HASH not provided or found in environment variables.")
  121. self.w3 = Web3(Web3.HTTPProvider(self.rpc_url))
  122. # 如果连接的是 PoA 网络 (如 Goerli, Sepolia, BSC, Polygon), 需要注入中间件
  123. # 对于主网,不需要此操作。可以根据 chain_id 动态判断,或者让用户明确。
  124. # 例如:if self.w3.eth.chain_id in [5, 11155111, 56, 137]: # Goerli, Sepolia, BSC, Polygon
  125. self.w3.middleware_onion.inject(ExtraDataToPOAMiddleware, layer=0)
  126. if not self.w3.is_connected():
  127. raise ConnectionError(f"Failed to connect to Ethereum node at {self.rpc_url}")
  128. self.account = Account.from_key(_hash)
  129. self.address = self.account.address
  130. logger.info(f"EthClient initialized. Address: {self.address}, RPC: {self.rpc_url}, Connected: {self.w3.is_connected()}")
  131. def _get_nonce(self) -> int:
  132. """获取账户的下一个 nonce"""
  133. return self.w3.eth.get_transaction_count(self.address)
  134. def _estimate_gas(self, tx: dict) -> int:
  135. """估算交易的 gas limit"""
  136. return self.w3.eth.estimate_gas(tx)
  137. def _sign(self, tx: dict, gas_limit_multiplier: float = 1.2) -> dict:
  138. """签署并返回簽名結果"""
  139. try:
  140. # 填充 gas 和 nonce (如果未提供)
  141. if 'nonce' not in tx:
  142. logger.info('TODO nonce应该提前管理好')
  143. tx['nonce'] = self._get_nonce()
  144. '''
  145. 在使用 web3.py 手动构建和签名交易时,需要使用支持 EIP-1559 交易类型的签名函数。
  146. 通常,eth_account 库及其 sign_transaction 方法是支持的,只要您提供的交易字典包含了正确的 EIP-1559 字段 (chainId, nonce, to, value, gas, maxFeePerGas, maxPriorityFeePerGas 等)。
  147. 您构建的交易字典可能混合了传统 Gas 字段 (gasPrice) 和 EIP-1559 字段 (maxFeePerGas, maxPriorityFeePerGas)。一个交易只能使用其中一种方式来指定 Gas 费用。
  148. 解决方案: 确保您的交易字典中只有 EIP-1559 相关的 Gas 字段(maxFeePerGas, maxPriorityFeePerGas 和 gas),或者只有传统 Gas 字段(gasPrice 和 gas)。
  149. 不要同时包含 gasPrice 和 maxFeePerGas/maxPriorityFeePerGas。
  150. '''
  151. # # 对于支持 EIP-1559 的网络,应该使用:
  152. # if 'maxPriorityFeePerGas' not in tx and 'maxFeePerGas' not in tx:
  153. # latest_block = self.w3.eth.get_block('latest')
  154. # tx['maxPriorityFeePerGas'] = '1000000000'
  155. # tx['maxFeePerGas'] = int(latest_block['baseFeePerGas']) * 2 + int(tx['maxPriorityFeePerGas'])
  156. if 'chainId' not in tx:
  157. tx['chainId'] = self.w3.eth.chain_id
  158. # 增加成交成功率
  159. tx['gas'] = int(int(tx['gas']) * gas_limit_multiplier)
  160. signed_tx = self.w3.eth.account.sign_transaction(tx, self.account.key)
  161. return signed_tx
  162. except Exception as e:
  163. logger.info(f"Error signing transaction: {e}")
  164. def _sign_and_send_transaction(self, tx: dict, gas_limit_multiplier: float = 1.2) -> str:
  165. """签署并发送交易,返回交易哈希"""
  166. try:
  167. signed_tx = self._sign(tx, gas_limit_multiplier)
  168. tx_hash = self.w3.eth.send_raw_transaction(signed_tx.raw_transaction)
  169. return self.w3.to_hex(tx_hash)
  170. except Exception as e:
  171. logger.info(f"Error signing or sending transaction: {e}")
  172. # 可以进一步处理特定错误,例如 nonce 过低,余额不足等
  173. raise
  174. def wait_for_transaction_receipt(self, tx_hash: str, timeout: int = 120, poll_latency: int = 1):
  175. """等待交易被打包并返回收据"""
  176. try:
  177. receipt = self.w3.eth.wait_for_transaction_receipt(tx_hash, timeout=timeout, poll_latency=poll_latency)
  178. return receipt
  179. except Exception as e: # Web3.exceptions.TimeExhausted as e
  180. logger.info(f"Transaction {tx_hash} timed out after {timeout} seconds.")
  181. raise
  182. def send_eth(self, to_address: str, amount_ether: float, gas_limit: int = None, gas_price_gwei: float = None) -> str:
  183. """发送 ETH"""
  184. if not self.w3.is_address(to_address):
  185. raise ValueError(f"Invalid recipient address: {to_address}")
  186. value_wei = self.w3.to_wei(amount_ether, 'ether')
  187. tx = {
  188. 'to': self.w3.to_checksum_address(to_address),
  189. 'value': value_wei,
  190. 'from': self.address # web3.py 会自动从签名账户中获取 from,但显式写上更清晰
  191. }
  192. if gas_limit:
  193. tx['gas'] = gas_limit
  194. if gas_price_gwei:
  195. tx['gasPrice'] = self.w3.to_wei(gas_price_gwei, 'gwei')
  196. logger.info(f"Preparing to send {amount_ether} ETH to {to_address}...")
  197. return self._sign_and_send_transaction(tx)
  198. def _get_erc20_contract(self, token_address: str):
  199. """获取 ERC20 合约实例"""
  200. if not self.w3.is_address(token_address):
  201. raise ValueError(f"Invalid token address: {token_address}")
  202. return self.w3.eth.contract(address=self.w3.to_checksum_address(token_address), abi=IERC20_ABI)
  203. def get_erc20_decimals(self, token_address: str) -> int:
  204. """获取 ERC20 代币的精度"""
  205. contract = self._get_erc20_contract(token_address)
  206. return contract.functions.decimals().call()
  207. def _to_token_units(self, token_address: str, amount_readable: float) -> int:
  208. """将可读的代币数量转换为最小单位 (例如 1.0 USDT -> 1000000 if decimals is 6)"""
  209. decimals = self.get_erc20_decimals(token_address)
  210. factor = Decimal(10) ** Decimal(decimals)
  211. return int(Decimal(str(amount_readable)) * factor)
  212. def _from_token_units(self, token_address: str, amount_units: int) -> Decimal:
  213. """将最小单位的代币数量转换为可读数量"""
  214. decimals = self.get_erc20_decimals(token_address)
  215. factor = Decimal(10) ** Decimal(decimals)
  216. return (Decimal(amount_units) / factor).quantize(Decimal('0.1') ** decimals, rounding=ROUND_DOWN)
  217. def transfer_erc20(self, token_address: str, to_address: str, amount_readable: float,
  218. gas_limit: int = None, gas_price: float = None) -> str:
  219. """
  220. 转移 ERC20 代币。
  221. :param token_address: 代币合约地址
  222. :param to_address: 接收者地址
  223. :param amount_readable: 要转移的代币数量 (例如 1.2345)
  224. :param gas_limit: 手动设置 gas limit (可选)
  225. :param gas_price: 手动设置 gas price (可选)
  226. :return: 交易哈希
  227. """
  228. if not self.w3.is_address(to_address):
  229. raise ValueError(f"Invalid recipient address: {to_address}")
  230. contract = self._get_erc20_contract(token_address)
  231. amount_in_smallest_units = self._to_token_units(token_address, amount_readable)
  232. tx_data = contract.functions.transfer(
  233. self.w3.to_checksum_address(to_address),
  234. amount_in_smallest_units
  235. ).build_transaction({
  236. 'from': self.address,
  237. 'nonce': self._get_nonce(),
  238. 'gas': 200000, # 通常 transfer ERC20 消耗 50k-100k gas,可以估算或手动设置
  239. 'gasPrice': self.w3.eth.gas_price # 或手动设置
  240. })
  241. if gas_limit:
  242. tx_data['gas'] = gas_limit
  243. if gas_price:
  244. tx_data['gasPrice'] = gas_price
  245. logger.info(f"Preparing to transfer {amount_readable} of token {token_address} to {to_address}...")
  246. return self._sign_and_send_transaction(tx_data, 1.2, 2)
  247. def approve_erc20(self, token_address: str, spender_address: str, amount_readable: float,
  248. gas_limit: int = None, gas_price_gwei: float = None) -> str:
  249. """
  250. 授权给 spender 地址一定数量的 ERC20 代币。
  251. :param token_address: 代币合约地址
  252. :param spender_address: 被授权者地址
  253. :param amount_readable: 要授权的代币数量 (例如 1.2345)
  254. :param gas_limit: 手动设置 gas limit (可选)
  255. :param gas_price_gwei: 手动设置 gas price Gwei (可选)
  256. :return: 交易哈希
  257. """
  258. if not self.w3.is_address(spender_address):
  259. raise ValueError(f"Invalid spender address: {spender_address}")
  260. contract = self._get_erc20_contract(token_address)
  261. amount_in_smallest_units = self._to_token_units(token_address, amount_readable)
  262. tx_data = contract.functions.approve(
  263. self.w3.to_checksum_address(spender_address),
  264. amount_in_smallest_units
  265. ).build_transaction({
  266. 'from': self.address,
  267. 'nonce': self._get_nonce(),
  268. })
  269. if gas_limit:
  270. tx_data['gas'] = gas_limit
  271. if gas_price_gwei:
  272. tx_data['gasPrice'] = self.w3.to_wei(gas_price_gwei, 'gwei')
  273. logger.info(f"Preparing to approve {amount_readable} of token {token_address} for spender {spender_address}...")
  274. return self._sign_and_send_transaction(tx_data)
  275. def get_erc20_balance(self, token_address: str, owner_address: str = None) -> Decimal:
  276. """获取指定地址的 ERC20 代币余额 (可读数量)"""
  277. target_address = owner_address or self.address
  278. if not self.w3.is_address(target_address):
  279. raise ValueError(f"Invalid owner address: {target_address}")
  280. contract = self._get_erc20_contract(token_address)
  281. balance_units = contract.functions.balanceOf(self.w3.to_checksum_address(target_address)).call()
  282. return self._from_token_units(token_address, balance_units)
  283. def get_erc20_allowance(self, token_address: str, spender_address: str, owner_address: str = None) -> Decimal:
  284. """获取 owner 授权给 spender 的 ERC20 代币数量 (可读数量)"""
  285. target_owner = owner_address or self.address
  286. if not self.w3.is_address(target_owner):
  287. raise ValueError(f"Invalid owner address: {target_owner}")
  288. if not self.w3.is_address(spender_address):
  289. raise ValueError(f"Invalid spender address: {spender_address}")
  290. contract = self._get_erc20_contract(token_address)
  291. allowance_units = contract.functions.allowance(
  292. self.w3.to_checksum_address(target_owner),
  293. self.w3.to_checksum_address(spender_address)
  294. ).call()
  295. return self._from_token_units(token_address, allowance_units)
  296. def get_erc20_total_supply(self, token_address: str) -> Decimal:
  297. """获取 ERC20 代币的总供应量 (可读数量)"""
  298. contract = self._get_erc20_contract(token_address)
  299. total_supply_units = contract.functions.totalSupply().call()
  300. return self._from_token_units(token_address, total_supply_units)
  301. def get_erc20_name(self, token_address: str) -> str:
  302. """获取 ERC20 代币的名称"""
  303. contract = self._get_erc20_contract(token_address)
  304. return contract.functions.name().call()
  305. def get_erc20_symbol(self, token_address: str) -> str:
  306. """获取 ERC20 代币的符号"""
  307. contract = self._get_erc20_contract(token_address)
  308. return contract.functions.symbol().call()
  309. def get_eth_balance(self, address: str = None) -> Decimal:
  310. """获取ETH余额 (单位 Ether)"""
  311. target_address = address or self.address
  312. if not self.w3.is_address(target_address):
  313. raise ValueError(f"Invalid address: {target_address}")
  314. balance_wei = self.w3.eth.get_balance(self.w3.to_checksum_address(target_address))
  315. return self.w3.from_wei(balance_wei, 'ether')
  316. if __name__ == "__main__":
  317. import traceback
  318. import time
  319. from pprint import pprint
  320. from checker import ok_chain_client
  321. # from ok_chain_client import swap, broadcast, orders
  322. from decimal import Decimal
  323. from config import wallet
  324. pprint(ok_chain_client.api_config)
  325. client = EthClient()
  326. CHAIN_ID = 1
  327. # IN_AMOUNT_TO_QUERY = decimal.Decimal('1')
  328. # IN_TOKEN_ADDRESS = '0xdAC17F958D2ee523a2206206994597C13D831ec7' # USDT on Ethereum
  329. # IN_TOKEN_DECIMALS = decimal.Decimal(6)
  330. # OUT_TOKEN_ADDRESS = '0xf816507E690f5Aa4E29d164885EB5fa7a5627860' # RATO on Ethereum
  331. USER_WALLET = wallet['user_wallet']
  332. # SLIPPAGE = 1
  333. # USER_EXCHANGE_WALLET = '0xc71835a042F4d870B0F4296cc89cAeb921a9f3DA'
  334. # rst = swap(CHAIN_ID, IN_AMOUNT_TO_QUERY * (10 ** IN_TOKEN_DECIMALS), IN_TOKEN_ADDRESS, OUT_TOKEN_ADDRESS, SLIPPAGE, USER_WALLET, USER_EXCHANGE_WALLET)
  335. # data = rst['data'][0]
  336. # tx = data['tx']
  337. try:
  338. # tx.pop('gasPrice', None)
  339. # tx.pop('value', None)
  340. # tx.pop('minReceiveAmount', None)
  341. # tx.pop('slippage', None)
  342. # tx.pop('maxSpendAmount', None)
  343. # tx.pop('signatureData', None)
  344. tx = {
  345. 'from': USER_WALLET,
  346. 'to': USER_WALLET,
  347. 'gas': '40000',
  348. 'value': 1,
  349. 'maxPriorityFeePerGas': '1800000000'
  350. }
  351. latest_block = client.w3.eth.get_block('latest')
  352. tx['maxPriorityFeePerGas'] = int(tx['maxPriorityFeePerGas'])
  353. tx['maxFeePerGas'] = int(int(latest_block['baseFeePerGas']) * 2 + tx['maxPriorityFeePerGas'])
  354. '''
  355. {'code': '0',
  356. 'data': {'chainId': '1',
  357. 'chainIndex': '1',
  358. 'dexRouter': '0x6088d94c5a40cecd3ae2d4e0710ca687b91c61d0',
  359. 'errorMsg': '',
  360. 'fromAddress': '0xb1f33026db86a86372493a3b124d7123e9045bb4',
  361. 'fromTokenDetails': {'amount': '10000000',
  362. 'symbol': 'USDT',
  363. 'tokenAddress': '0xdac17f958d2ee523a2206206994597c13d831ec7'},
  364. 'gasLimit': '285000',
  365. 'gasPrice': '2526360593',
  366. 'gasUsed': '216783',
  367. 'height': '22586403',
  368. 'referralAmount': '',
  369. 'status': 'success',
  370. 'toAddress': '0xc71835a042f4d870b0f4296cc89caeb921a9f3da',
  371. 'toTokenDetails': {'amount': '655477388084022',
  372. 'symbol': 'RATO',
  373. 'tokenAddress': '0xf816507e690f5aa4e29d164885eb5fa7a5627860'},
  374. 'txFee': '',
  375. 'txHash': '0xe54acd58923685fb132bc0d103b72fa538e59d54dec44b3dc8e0314de7d8b126',
  376. 'txTime': '1748497463',
  377. 'txType': 'swap'},
  378. 'msg': ''}
  379. '''
  380. # pprint(ok_chain_client.history(CHAIN_ID, '0xe54acd58923685fb132bc0d103b72fa538e59d54dec44b3dc8e0314de7d8b126'))
  381. '''
  382. {
  383. 'code': '0',
  384. 'data': {'chainId': '1',
  385. 'chainIndex': '1',
  386. 'errorMsg': 'execution reverted',
  387. 'fromAddress': '0xc36b5466d88d3ebe9e538ed650f61b7f9902e6cc',
  388. 'gasLimit': '285000',
  389. 'gasPrice': '4907400618',
  390. 'gasUsed': '282898',
  391. 'height': '22587649',
  392. 'status': 'fail',
  393. 'txFee': '',
  394. 'txHash': '0x344d2d0a9efdfb46b6130a58a40e117bb3bf6181b03b09b65b4d6cc8256f1e2f',
  395. 'txTime': '1748512535',
  396. 'txType': ''},
  397. 'msg': ''
  398. }
  399. '''
  400. pprint(ok_chain_client.history(CHAIN_ID, '0x344d2d0a9efdfb46b6130a58a40e117bb3bf6181b03b09b65b4d6cc8256f1e2f'))
  401. # pprint(ok_chain_client.gas_limit(CHAIN_ID, tx['from'], tx['to'], tx['value']))
  402. # pprint(tx)
  403. # estimated_gas = client.w3.eth.estimate_gas(tx)
  404. # estimated_wei = estimated_gas * (tx['maxPriorityFeePerGas'] + tx['maxFeePerGas'])
  405. # estimated_eth = estimated_wei / (10 ** 18)
  406. # logger.info(f"估算的燃气量: {estimated_gas}, eth消耗: {estimated_eth}")
  407. # logger.info(f"餘額:{client.w3.eth.get_balance(USER_WALLET) / Decimal('1e18')}")
  408. # tx_hash = client._sign_and_send_transaction(tx)
  409. # receipt = client.wait_for_transaction_receipt(tx_hash)
  410. # logger.info(f"{tx_hash} 交易已确认! Status: {'Success' if receipt.status == 1 else 'Failed'}")
  411. # # ok api發交易測試
  412. # signed_tx = client._sign(tx)
  413. # raw_transaction = client.w3.to_hex(signed_tx.raw_transaction)
  414. # broadcast_rst = ok_chain_client.broadcast(CHAIN_ID, USER_WALLET, raw_transaction)
  415. # pprint(broadcast_rst)
  416. # order_id = broadcast_rst['data'][0]['orderId']
  417. # while True:
  418. # wallet_orders = ok_chain_client.orders(CHAIN_ID, USER_WALLET)
  419. # pprint(wallet_orders)
  420. # time.sleep(1)
  421. except Exception as e:
  422. print(f"測試失败: {e}")
  423. traceback.print_exc()
  424. # # --- 使用示例 ---
  425. # # 确保你的 .env 文件配置正确
  426. # # 并且你的账户中有足够的 ETH 来支付 Gas 费
  427. # # 替换为实际的ERC20代币地址和接收者地址 (例如USDT on Sepolia testnet)
  428. # # Sepolia USDT: 0xaA8E23Fb1079EA71e0a56F48S1a3ET28wpD1RLf
  429. # # Sepolia WETH: 0x7b79995e5f793A07Bc00c21412e50Ecn10yt2e_ (错误,应为 0x7b79995e5f793A07Bc00c21412e50Ecn10yt2eH)
  430. # # 更正: Sepolia WETH: 0xfFf9976782d46CC05630D1f6eBAb18b2324d6B14 (常用)
  431. # # 有些测试网可能没有标准的USDT,你可以找一个存在的ERC20代币进行测试,或者自己部署一个
  432. # # 为了演示,这里假设使用的是 Sepolia 测试网
  433. # # !!重要!!: 以下地址和代币地址仅为示例, 请替换为您测试网络上的真实地址和代币
  434. # # 如果您在主网操作,请务必小心,并使用小额资金测试。
  435. # TEST_RECIPIENT_ADDRESS = "0xb1f33026db86a86372493a3b124d7123e9045bb4" # 替换为你的测试接收地址
  436. # # Sepolia 上的一个示例 ERC20 token (你可以找一个你有的测试币)
  437. # # 用于测试的代币地址
  438. # TEST_ERC20_TOKEN_ADDRESS_SEPOLIA_LINK = "0xdAC17F958D2ee523a2206206994597C13D831ec7"
  439. # try:
  440. # client = EthClient() # RPC_URL 和 HASH 会从 .env 文件加载
  441. # logger.info(f"\nMy ETH Balance: {client.get_eth_balance()} ETH")
  442. # # 1. 发送 ETH (取消注释以测试, 确保接收地址正确且你有足够ETH)
  443. # # logger.info(f"\nAttempting to send ETH...")
  444. # # eth_tx_hash = client.send_eth(TEST_RECIPIENT_ADDRESS, 0.0001) # 发送 0.0001 ETH
  445. # # logger.info(f"ETH transaction sent! Hash: {eth_tx_hash}")
  446. # # receipt = client.wait_for_transaction_receipt(eth_tx_hash)
  447. # # logger.info(f"ETH transaction confirmed! Status: {'Success' if receipt.status == 1 else 'Failed'}")
  448. # # --- ERC20 操作示例 ---
  449. # # 使用 Sepolia LINK 代币进行演示
  450. # token_address = TEST_ERC20_TOKEN_ADDRESS_SEPOLIA_LINK
  451. # if not client.w3.is_address(token_address): # 简单检查
  452. # logger.info(f"Warning: {token_address} does not look like a valid address. Skipping ERC20 tests.")
  453. # else:
  454. # logger.info(f"\n--- ERC20 Token Operations for: {token_address} ---")
  455. # token_name = client.get_erc20_name(token_address)
  456. # token_symbol = client.get_erc20_symbol(token_address)
  457. # token_decimals = client.get_erc20_decimals(token_address)
  458. # logger.info(f"Token: {token_name} ({token_symbol}), Decimals: {token_decimals}")
  459. # # ERC20余额查询以及基础功能测试(总供应量)
  460. # my_token_balance = client.get_erc20_balance(token_address)
  461. # logger.info(f"My {token_symbol} Balance: {my_token_balance} {token_symbol}")
  462. # total_supply = client.get_erc20_total_supply(token_address)
  463. # logger.info(f"Total Supply of {token_symbol}: {total_supply} {token_symbol}")
  464. # # # 2. ERC20 转账 (取消注释以测试, 确保你有该代币且接收地址正确)
  465. # # amount_to_transfer = 0.01 # 转移 0.01 个代币
  466. # # if my_token_balance >= Decimal(str(amount_to_transfer)):
  467. # # logger.info(f"\nAttempting to transfer {amount_to_transfer} {token_symbol}...")
  468. # # erc20_tx_hash = client.transfer_erc20(token_address, TEST_RECIPIENT_ADDRESS, amount_to_transfer)
  469. # # logger.info(f"{token_symbol} transfer transaction sent! Block: {client.w3.eth.block_number} Hash: {erc20_tx_hash}")
  470. # # receipt = client.wait_for_transaction_receipt(erc20_tx_hash)
  471. # # logger.info(f"{token_symbol} transfer transaction confirmed! Block: {client.w3.eth.block_number} Status: {'Success' if receipt.status == 1 else 'Failed'}")
  472. # # logger.info(f"My new {token_symbol} Balance: {client.get_erc20_balance(token_address)} {token_symbol}")
  473. # # else:
  474. # # logger.info(f"Insufficient {token_symbol} balance to transfer {amount_to_transfer} {token_symbol}.")
  475. # # # 3. ERC20 Approve 和 Allowance (取消注释以测试)
  476. # # spender_for_allowance = '0x156ACd2bc5fC336D59BAAE602a2BD9b5e20D6672' # 可以是任何你想授权的地址
  477. # # amount_to_approve = 74547271788
  478. # # token_address = "0xdAC17F958D2ee523a2206206994597C13D831ec7"
  479. # # # current_allowance = client.get_erc20_allowance(token_address, spender_for_allowance)
  480. # # # logger.info(f"\nCurrent allowance for {spender_for_allowance} to spend my {token_symbol}: {current_allowance} {token_symbol}")
  481. # # # if my_token_balance >= Decimal(str(amount_to_approve)): # 确保有足够的代币去授权(虽然授权本身不消耗代币)
  482. # # logger.info(f"\nAttempting to approve {amount_to_approve} {token_symbol} for spender {spender_for_allowance}...")
  483. # # approve_tx_hash = client.approve_erc20(token_address, spender_for_allowance, amount_to_approve)
  484. # # logger.info(f"{token_symbol} approve transaction sent! Hash: {approve_tx_hash}")
  485. # # receipt = client.wait_for_transaction_receipt(approve_tx_hash)
  486. # # logger.info(f"{token_symbol} approve transaction confirmed! Status: {'Success' if receipt.status == 1 else 'Failed'}")
  487. # # new_allowance = client.get_erc20_allowance(token_address, spender_for_allowance)
  488. # # logger.info(f"New allowance for {spender_for_allowance}: {new_allowance} {token_symbol}")
  489. # # else:
  490. # # logger.info(f"Not enough balance to consider approving {amount_to_approve} {token_symbol} (though approval itself doesn't spend tokens).")
  491. # except ValueError as ve:
  492. # logger.info(f"Configuration Error: {ve}")
  493. # except ConnectionError as ce:
  494. # logger.info(f"Connection Error: {ce}")
  495. # except Exception as e:
  496. # logger.info(f"An unexpected error occurred: {e}")
  497. # import traceback
  498. # traceback.logger.info_exc()