import httpx
import asyncio
import os
import time
import hmac
import hashlib
from dotenv import load_dotenv

load_dotenv()

async def check_wallet():
    api_key = os.getenv("BYBIT_API_KEY")
    api_secret = os.getenv("BYBIT_API_SECRET")
    
    url = "https://api.bybit.com/v5/account/wallet-balance"
    params = {"accountType": "UNIFIED"}
    
    timestamp = str(int(time.time() * 1000))
    recv_window = "5000"
    sorted_params = "&".join([f"{k}={v}" for k, v in sorted(params.items())])
    raw_str = timestamp + api_key + recv_window + sorted_params
    signature = hmac.new(api_secret.encode('utf-8'), 
                        raw_str.encode('utf-8'), hashlib.sha256).hexdigest()
    
    headers = {
        'X-BAPI-API-KEY': api_key,
        'X-BAPI-TIMESTAMP': timestamp,
        'X-BAPI-SIGN-TYPE': '2',
        'X-BAPI-RECV-WINDOW': recv_window,
        'X-BAPI-SIGN': signature
    }

    async with httpx.AsyncClient() as client:
        resp = await client.get(url, params=params, headers=headers)
        data = resp.json()
        print(f"Wallet Balance: {data}")

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