arbitrage/submit_process_demo.py

import requests
import json
import time
import logging
from decimal import Decimal # 使用 Decimal 来表示精确的金额和价差

from checker.config import arb

logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')

# --- 配置 arb_executor.py 的 HTTP 地址和端口 ---
ARB_EXECUTOR_URL = arb["ARB_EXECUTOR_URL"]
OPEN_LIMIT = Decimal(arb["OPEN_LIMIT"])
CLOSE_LIMIT = Decimal(arb["CLOSE_LIMIT"])
MEXC_TARGET_PAIR_USDT = arb["CEX_PAIR"]
CEX_TRADE_AMOUNT = arb["COIN_TOKEN_TRADE_AMOUNT"]
STRATEGY = arb["STRATEGY"]

# --- 模拟的套利机会数据 ---
# 这是一个示例，实际数据应来自你的 price_checker 逻辑
def create_mock_arbitrage_data():
    query_price_endpoint = 'http://127.0.0.1:7777/table-data'
    # 构造提交给 arb_executor 的数据体
    data = {
        "pct": str(0.01),
        "openLimit": str(OPEN_LIMIT.quantize(Decimal('0.00001'))),
        "closeLimit": str(-0.2),
        "cexPrice": str('0.0067'),
        "dexPrice": str('0.0065'),
        "symbol": MEXC_TARGET_PAIR_USDT,
        "exchangeOutAmount": str(CEX_TRADE_AMOUNT), # CEX上期望卖出的目标币数量
        "strategy": STRATEGY,
        "queryPriceUrl": query_price_endpoint,
    }
    return data

# --- 发送请求函数 ---
def submit_arbitrage_process(arbitrage_data):
    """
    向 arb_executor 服务提交套利处理请求。
    """
    print(f"正在提交套利数据到 {ARB_EXECUTOR_URL}")

    try:
        # 发送 POST 请求
        response = requests.post(ARB_EXECUTOR_URL, json=arbitrage_data)

        # 检查响应状态码
        if response.status_code == 201:
            print("\n请求成功! 套利流程已启动。")
        elif response.status_code == 200:
             print("\n请求接收成功，但未达到利润阈值，未启动套利流程。")
        elif response.status_code == 400:
             print("\n请求失败! 无效的请求数据。")
        else:
            print(f"\n请求失败! 状态码: {response.status_code}")

        # 使用 response.json() 解析响应体
        try:
            response_data = response.json()
            logging.info(f"\n响应数据: {response_data}")
        except requests.exceptions.JSONDecodeError:
            logging.error("无法解析响应体为 JSON。")
            logging.info(f"原始响应文本: {response.text}")

    except requests.exceptions.ConnectionError as e:
        print(f"\n连接错误: 无法连接到 {ARB_EXECUTOR_URL}。请确保 arb_executor.py 正在运行。")
        print(f"错误详情: {e}")
    except Exception as e:
        print(f"\n发送请求时发生未知错误: {e}")

# --- 主执行逻辑 ---
if __name__ == "__main__":
    print("--- 模拟 Price Checker 发现套利机会 ---")

    # 模拟一个达到利润阈值的套利机会
    arb_opportunity_met = create_mock_arbitrage_data()

    submit_arbitrage_process(arb_opportunity_met)

    print("\n模拟完成。请检查 arb_executor.py 的控制台输出和 /processing, /history 端点查看结果。")