from datetime import datetime
import json

from sdk import Client


BASE_URL = "http://127.0.0.1:10086"
TOKEN = "QMTbyYanweidong"
STOCK_CODE = "000021.SZ"
VOLUME = 100

def main() -> None:
    """只读查询四类原始数据，输出 JSON 便于核对字段。"""
    with Client(BASE_URL, TOKEN) as client:
        for datatype in ("account", "order", "deal", "position"):
            print(f"\n=== {datatype} ===")
            try:
                result = client.org(datatype)
                print(f"记录数：{len(result)}")
                print(json.dumps(result, ensure_ascii=False, indent=2, default=str))
            except Exception as exc:
                print(f"查询失败：{exc}")


def main1() -> None:
    order = {
        "opType": 23,
        "orderType": 1101,
        "stockCode": STOCK_CODE,
        "prType": 5,
        "price": -1,
        "volume": VOLUME,
        "quickTrade": 2,
        "strategyName": "manual-test",
        "orderId": f"test-{datetime.now():%H%M%S}",
    }

    with Client(BASE_URL, TOKEN) as client:
        #client._post_json("/api/sys/shutdown")
        print("真实委托：", order)
        result = client.passorder(
            order["opType"],
            order["stockCode"],
            order["volume"],
            order_type=order["orderType"],
            pr_type=order["prType"],
            price=order["price"],
            quick_trade=order["quickTrade"],
            strategy_name=order["strategyName"],
            order_id=order["orderId"],
        )
        print("下单结果：", result)


if __name__ == "__main__":
    main()