123 lines
4.6 KiB
Python
123 lines
4.6 KiB
Python
# coding:utf-8
|
|
import threading
|
|
import time, sys
|
|
sys.stdout.reconfigure(encoding='utf-8') # 设置标准输出编码为UTF-8
|
|
import strategy_db
|
|
from trade_thread import StockTradeThread
|
|
from util import getInstrumentName
|
|
from xtquant.xttrader import XtQuantTrader
|
|
from xtquant.xttype import StockAccount
|
|
|
|
|
|
class SFGridController:
|
|
def __init__(self, account_no: str = '99082560'):
|
|
print('\n=== 数据库模块初始化 ===\n')
|
|
strategy_db.db.connect()
|
|
strategy_db.db.create_tables([strategy_db.TradeTarget])
|
|
|
|
|
|
print('\n=== 三疯网络策略控制器初始化 ===\n')
|
|
self.init_instrument_pool()
|
|
self.init_trader(r'D:\\Programs\\DTQMT\\userdata_mini')
|
|
self.init_trade_account(account_no, 'STOCK')
|
|
self.grid_price = [11, 10, 9, 8, 7, 6, 5, 4, 3, 2, 1]
|
|
print('\n=== 三疯网络策略控制器初始化完成 ===\n')
|
|
|
|
def add_trade_target(self, stock_code):
|
|
try:
|
|
stock_name = getInstrumentName(stock_code)
|
|
new_target = strategy_db.TradeTarget.create(
|
|
stock_name=stock_name,
|
|
stock_code=stock_code,
|
|
current_position=0,
|
|
grid_index=0,
|
|
last_trade_price=0.0,
|
|
current_buy_price=0.0,
|
|
current_sell_price=0.0
|
|
)
|
|
new_target.save()
|
|
print(f'新增交易标的 {stock_code} {stock_name}, {new_target.id}')
|
|
except Exception as e:
|
|
print(f'新增交易标的失败 {stock_code} {e}')
|
|
|
|
def init_instrument_pool(self):
|
|
self.instrument_pool = strategy_db.TradeTarget.select()
|
|
print(f'初始化标的池初始化完成, {self.instrument_pool}')
|
|
|
|
def init_trader(self, path):
|
|
session_id = int(time.time())
|
|
self.xt_trader = XtQuantTrader(path, session_id)
|
|
self.xt_trader.start()
|
|
self.xt_trader.connect()
|
|
print(f'交易对象初始化完成, {self.xt_trader.connected}',)
|
|
|
|
def init_trade_account(self, account_id, account_type):
|
|
self.account= StockAccount(account_id, account_type)
|
|
print(f'交易账号对象初始化完成, {self.account}')
|
|
|
|
def print_account_info(self):
|
|
account_info = self.xt_trader.query_stock_asset(self.account)
|
|
|
|
print(f"\n=== 账户信息 {self.account.account_id} ===")
|
|
print(f"可用资金: {account_info.m_dCash}")
|
|
print(f"总资产: {account_info.m_dTotalAsset}")
|
|
print(f"证券市值: {account_info.m_dMarketValue}")
|
|
|
|
positions = self.xt_trader.query_stock_positions(self.account)
|
|
if positions:
|
|
print("\n=== 持仓信息 ===")
|
|
for pos in positions:
|
|
if pos.m_nVolume <=0:
|
|
continue
|
|
print(f"股票代码: {pos.stock_code}-{getInstrumentName(pos.stock_code)}")
|
|
print(f"总持仓: {pos.m_nVolume}")
|
|
print(f"可用持仓: {pos.m_nCanUseVolume}")
|
|
print(f"持仓成本: {pos.avg_price}")
|
|
print("---")
|
|
else:
|
|
print("\n当前无持仓")
|
|
|
|
def print_stock_orders(self):
|
|
orders = self.xt_trader.query_stock_orders(self.account, cancelable_only=True)
|
|
if orders:
|
|
print("\n=== 委托信息 ===")
|
|
for order in orders:
|
|
print(f"委托编号: {order.order_id}")
|
|
print(f"股票代码: {order.stock_code} {getInstrumentName(order.stock_code)}")
|
|
print(f"委托方向: {order.offset_flag} ")
|
|
print(f"委托价格: {order.price}")
|
|
print(f"委托数量: {order.order_volume}")
|
|
print(f"已成交数量: {order.traded_volume}")
|
|
print(f"委托状态: {order.order_status} ")
|
|
print("---")
|
|
else:
|
|
print("\n当前无委托记录")
|
|
|
|
def interact(self):
|
|
"""执行后进入repl模式"""
|
|
import code
|
|
code.InteractiveConsole(locals=globals()).interact()
|
|
|
|
def init_stock_trade_threads(self):
|
|
for stock_code in self.instrument_pool:
|
|
new_job = StockTradeThread(stock_code)
|
|
new_job.name = f"StockTradeThread-{stock_code}"
|
|
new_job.start()
|
|
|
|
def stock_trade_thread(self, stock_code):
|
|
print(f"启动标的交易线程 {stock_code} {getInstrumentName(stock_code)}\n")
|
|
threading.Event().wait(2) # 模拟交易操作的等待时间
|
|
|
|
|
|
if __name__ == '__main__':
|
|
ctrl = SFGridController('99082560')
|
|
ctrl.print_account_info()
|
|
ctrl.print_stock_orders()
|
|
|
|
ctrl.init_stock_trade_threads()
|
|
|
|
# 交互阻塞
|
|
ctrl.interact()
|
|
|
|
|