from datetime import datetime
import json
from libs.collector import collector_push
from sdk import Assets, Client, PositionItem


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:
        portfolio = client.portfolio()
        assets = portfolio.assets
        positions = list(portfolio.positions.values())
        collector_push("86037237", assets, positions)
        


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()