from xtquant import xttrader
from xtquant.xtdata import download_history_data, get_market_data
from xtquant.xttype import StockAccount
import random

##订阅账户
# 设置 QMT 交易端的数据路径和会话ID
min_path = r"C:\\QMT\\中金财富QMT个人版交易端\\userdata_mini"
session_id = int(random.randint(100000, 999999))

# 创建 XtQuantTrader 实例并启动
xt_trader = xttrader.XtQuantTrader(min_path, session_id)
xt_trader.start()

# 连接 QMT 交易端
connect_result = xt_trader.connect()
if connect_result == 0:
    print('连接成功')
else:
    print('连接失败')
    xt_trader.stop()
    exit()

# 设置账户信息
account = StockAccount('8176081580')

# 订阅账户
res = xt_trader.subscribe(account)
if res == 0:
    print('订阅成功')
else:
    print('订阅失败')