introduce to tests

florida/tools/check_arbitrum_pool.py

@@ -0,0 +1,103 @@
+
+import os
+import sys
+import json
+from web3 import Web3
+from dotenv import load_dotenv
+from decimal import Decimal
+
+# Add project root
+sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
+
+# Load Env
+load_dotenv()
+
+RPC_URL = os.environ.get("MAINNET_RPC_URL") # Arbitrum RPC
+POOL_ADDRESS = "0xC6962004f452bE9203591991D15f6b388e09E8D0" # ARB/WETH 500
+
+ERC20_ABI = [
+    {"constant": True, "inputs": [], "name": "symbol", "outputs": [{"name": "", "type": "string"}], "payable": False, "stateMutability": "view", "type": "function"},
+    {"constant": True, "inputs": [], "name": "decimals", "outputs": [{"name": "", "type": "uint8"}], "payable": False, "stateMutability": "view", "type": "function"},
+    {"constant": True, "inputs": [{"name": "_owner", "type": "address"}], "name": "balanceOf", "outputs": [{"name": "balance", "type": "uint256"}], "payable": False, "stateMutability": "view", "type": "function"}
+]
+
+POOL_ABI = [
+    {"inputs": [], "name": "slot0", "outputs": [{"internalType": "uint160", "name": "sqrtPriceX96", "type": "uint160"}, {"internalType": "int24", "name": "tick", "type": "int24"}, {"internalType": "uint16", "name": "observationIndex", "type": "uint16"}, {"internalType": "uint16", "name": "observationCardinality", "type": "uint16"}, {"internalType": "uint16", "name": "observationCardinalityNext", "type": "uint16"}, {"internalType": "uint32", "name": "feeProtocol", "type": "uint32"}, {"internalType": "bool", "name": "unlocked", "type": "bool"}], "stateMutability": "view", "type": "function"},
+    {"inputs": [], "name": "liquidity", "outputs": [{"internalType": "uint128", "name": "", "type": "uint128"}], "stateMutability": "view", "type": "function"},
+    {"inputs": [], "name": "token0", "outputs": [{"internalType": "address", "name": "", "type": "address"}], "stateMutability": "view", "type": "function"},
+    {"inputs": [], "name": "token1", "outputs": [{"internalType": "address", "name": "", "type": "address"}], "stateMutability": "view", "type": "function"},
+    {"inputs": [], "name": "fee", "outputs": [{"internalType": "uint24", "name": "", "type": "uint24"}], "stateMutability": "view", "type": "function"}
+]
+
+def main():
+    if not RPC_URL:
+        print("Error: MAINNET_RPC_URL not found in .env")
+        return
+
+    w3 = Web3(Web3.HTTPProvider(RPC_URL))
+    if not w3.is_connected():
+        print("Error: Could not connect to RPC")
+        return
+
+    print(f"Connected to Arbitrum: {w3.eth.block_number}")
+    
+    pool_contract = w3.eth.contract(address=POOL_ADDRESS, abi=POOL_ABI)
+    
+    # 1. Metadata
+    t0_addr = pool_contract.functions.token0().call()
+    t1_addr = pool_contract.functions.token1().call()
+    fee = pool_contract.functions.fee().call()
+    
+    t0_contract = w3.eth.contract(address=t0_addr, abi=ERC20_ABI)
+    t1_contract = w3.eth.contract(address=t1_addr, abi=ERC20_ABI)
+    
+    t0_sym = t0_contract.functions.symbol().call()
+    t1_sym = t1_contract.functions.symbol().call()
+    t0_dec = t0_contract.functions.decimals().call()
+    t1_dec = t1_contract.functions.decimals().call()
+    
+    print(f"\nPool: {t0_sym} / {t1_sym} ({fee/10000}%)")
+    print(f"Token0: {t0_sym} ({t0_addr}) - {t0_dec} dec")
+    print(f"Token1: {t1_sym} ({t1_addr}) - {t1_dec} dec")
+    
+    # 2. State
+    slot0 = pool_contract.functions.slot0().call()
+    sqrt_price = slot0[0]
+    tick = slot0[1]
+    liquidity = pool_contract.functions.liquidity().call()
+    
+    print(f"\nState:")
+    print(f"Liquidity: {liquidity}")
+    print(f"Tick: {tick}")
+    print(f"SqrtPriceX96: {sqrt_price}")
+    
+    # 3. Price Calc
+    # price = (sqrt / 2^96)^2
+    p = (Decimal(sqrt_price) / Decimal(2**96)) ** 2
+    
+    # Adjust for decimals: Price = raw_price * 10^(d0 - d1)
+    adj_price = p * (Decimal(10) ** (t0_dec - t1_dec))
+    inv_price = Decimal(1) / adj_price if adj_price > 0 else 0
+    
+    print(f"\nPrices:")
+    print(f"1 {t0_sym} = {adj_price:.6f} {t1_sym}")
+    print(f"1 {t1_sym} = {inv_price:.6f} {t0_sym}")
+    
+    # 4. TVL Estimation (Balances)
+    t0_bal = t0_contract.functions.balanceOf(POOL_ADDRESS).call()
+    t1_bal = t1_contract.functions.balanceOf(POOL_ADDRESS).call()
+    
+    t0_human = Decimal(t0_bal) / Decimal(10**t0_dec)
+    t1_human = Decimal(t1_bal) / Decimal(10**t1_dec)
+    
+    print(f"\nTVL (Locked in Contract):")
+    print(f"{t0_human:,.2f} {t0_sym}")
+    print(f"{t1_human:,.2f} {t1_sym}")
+    
+    # Assume WETH is approx 3350 (or whatever current market is, we can use slot0 price if one is stable)
+    # If one is USD stable, we can calc total.
+    # ARB / WETH. WETH is ~$3300.
+    # Let's just output raw.
+
+if __name__ == "__main__":
+    main()