import hashlib
import hmac
import json
import os
import secrets
import time
from urllib.request import Request, urlopen

def post(path, body):
    nonce = secrets.token_hex(16)
    timestamp = str(int(time.time()))
    signature = hmac.new(
        os.environ["QFEX_SECRET_KEY"].encode(),
        f"{nonce}:{timestamp}".encode(),
        hashlib.sha256,
    ).hexdigest()
    request = Request(
        "https://api.qfex.com" + path,
        data=json.dumps(body).encode(),
        headers={
            "Content-Type": "application/json",
            "x-qfex-public-key": os.environ["QFEX_PUBLIC_KEY"],
            "x-qfex-hmac-signature": signature,
            "x-qfex-nonce": nonce,
            "x-qfex-timestamp": timestamp,
        },
        method="POST",
    )
    with urlopen(request) as response:
        data = response.read()
        return json.loads(data) if data else None

# Create and fund with USD 50,000 at 2x portfolio leverage.
account = post("/managed-accounts/create", {"amount": 50000, "leverage_ratio": 2})

main_id = os.environ["QFEX_MAIN_ACCOUNT_ID"]
managed_id = account["account_id"]

# Deposit USD 1,000 from the main account into the managed account.
post(
    f"/user/transfer?src_account_id={main_id}&dst_account_id={managed_id}",
    {"amount": 1000},
)

# Withdraw USD 500 from the managed account back to the main account.
post(
    f"/user/transfer?src_account_id={managed_id}&dst_account_id={main_id}",
    {"amount": 500},
)

# Change portfolio leverage to 3x.
post("/managed-accounts/leverage", {"leverage_ratio": 3})