import threading
import time
from strategy_db import TradeTarget
import strategy_db
from util import getInstrumentName, getStockPosition
from xtquant import xttrader, xtdata
from xtquant.xttype import StockAccount


class StockTradeController:

    def __init__(self, tradeTarget: TradeTarget, xt_trader: xttrader.XtQuantTrader, account: StockAccount, enabled: bool = False):
        super().__init__() #必须调用父类的初始化方法
        self.tradeTarget = tradeTarget
        self.xt_trader = xt_trader
        self.account = account
        self.currentPosition = getStockPosition(self.tradeTarget.stock_code, self.xt_trader, self.account)
        self.tradeRecords = strategy_db.TradeRecord.select().where(strategy_db.TradeRecord.stock_code == self.tradeTarget.stock_code).order_by(strategy_db.TradeRecord.trade_time.desc())

        print(f"创建标的交易线程 {self.tradeTarget.stock_code} {getInstrumentName(self.tradeTarget.stock_code)} {tradeTarget}\n")
        xtdata.subscribe_quote(tradeTarget.stock_code, period='tick', start_time='', end_time='', count=0, callback=self.onDataUpdate)
    
    def enabledTrading(self, enabled: bool):
        self.tradeTarget.enabled = enabled
        self.tradeTarget.save()
    
    def isEnabled(self) -> bool:
        return self.tradeTarget.enabled


    def onDataUpdate(self, data):
        if self.isEnabled():
            print(f"标的{self.tradeTarget.stock_code}-{self.tradeTarget.stock_name} 行情数据更新 {data[self.tradeTarget.stock_code]}\n")


    # Description: 程序启动后
    def check_stock_position(self):
        volume = getStockPosition(self.tradeTarget.stock_code, self.xt_trader, self.account)
        pass


    # Description: 新标的，建基础仓
    def init_stock_position(self):
        pass

    # Description: 双向下单
    def two_way_order(self):
        pass