Documentação da API

import requests

# ==============================
# CONFIGURAÇÃO
# ==============================

API_KEY = "SUA_API_KEY_AQUI" # Substitua pela sua chave
BASE_URL = "https://olympuspay.cloud/api/v1"

HEADERS = {
    "Authorization": API_KEY,
    "Content-Type": "application/json"
}

# ==============================
# FUNÇÕES
# ==============================

def consultar_saldo():
    try:
        r = requests.get(f"{BASE_URL}/balance", headers=HEADERS)
        print("\n🔎 SALDO:")
        print(r.json())
    except Exception as e:
        print(f"Erro: {e}")

def criar_deposito():
    valor = input("Valor do depósito (ex: 5.90): ").replace(",", ".")
    try:
        valor = float(valor)
    except:
        print("❌ Valor inválido")
        return

    payload = {"amount": valor}

    try:
        r = requests.post(
            f"{BASE_URL}/pay",
            headers=HEADERS,
            json=payload
        )
        data = r.json()

        print("\n📥 DEPÓSITO CRIADO")
        print(data)

        # Tenta recuperar o ID de várias formas
        payment_id = (
            data.get("id")
            or data.get("payment_id")
            or data.get("order_id")
            or data.get("reference")
        )

        if payment_id:
            print(f"\n🆔 ID DO DEPÓSITO: {payment_id}")
            print("👉 Use esse ID para consultar o status")
        else:
            print("\n⚠️ ATENÇÃO: ID não identificado automaticamente.")

        if "qr_code" in data:
            print("\n📲 PIX COPIA E COLA:")
            print(data["qr_code"])
            
    except Exception as e:
        print(f"Erro na requisição: {e}")

def consultar_deposito():
    payment_id = input("ID do depósito: ").strip()
    if not payment_id:
        print("❌ ID inválido")
        return

    try:
        r = requests.get(
            f"{BASE_URL}/payment/{payment_id}",
            headers=HEADERS
        )
        print("\n📊 STATUS DO DEPÓSITO")
        print(r.json())
    except Exception as e:
        print(f"Erro: {e}")

def criar_saque():
    valor = input("Valor do saque (ex: 10.00): ").replace(",", ".")
    try:
        valor = float(valor)
    except:
        print("❌ Valor inválido")
        return

    pix_key = input("Chave PIX: ").strip()
    key_type = input("Tipo da chave (cpf/email/phone/random) [enter p/ ignorar]: ").strip()

    payload = {
        "amount": valor,
        "pix_key": pix_key
    }

    if key_type:
        payload["key_type"] = key_type

    try:
        r = requests.post(
            f"{BASE_URL}/withdraw",
            headers=HEADERS,
            json=payload
        )
        data = r.json()

        print("\n📤 SAQUE CRIADO")
        print(data)

        withdraw_id = data.get("id")

        if withdraw_id:
            print(f"\n🆔 ID DO SAQUE: {withdraw_id}")
            print("👉 Use esse ID para consultar o status")
            
    except Exception as e:
        print(f"Erro: {e}")

def consultar_saque():
    withdraw_id = input("ID do saque: ").strip()
    if not withdraw_id:
        print("❌ ID inválido")
        return

    try:
        r = requests.get(
            f"{BASE_URL}/withdraw/{withdraw_id}",
            headers=HEADERS
        )
        print("\n📊 STATUS DO SAQUE")
        print(r.json())
    except Exception as e:
        print(f"Erro: {e}")

# ==============================
# MENU
# ==============================

def menu():
    while True:
        print("""
==============================
        OLYMPUSPAY CLI
==============================
1 - Consultar saldo
2 - Criar depósito (PIX)
3 - Consultar depósito
4 - Criar saque (PIX)
5 - Consultar saque
0 - Sair
""")
        opcao = input("Escolha: ").strip()

        if opcao == "1": consultar_saldo()
        elif opcao == "2": criar_deposito()
        elif opcao == "3": consultar_deposito()
        elif opcao == "4": criar_saque()
        elif opcao == "5": consultar_saque()
        elif opcao == "0": 
            print("Saindo...")
            break
        else:
            print("❌ Opção inválida")

if __name__ == "__main__":
    menu()