web3_py_client.py 26 KB

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