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

class StockTradeThread(threading.Thread):
    def __init__(self, tradeTarget: TradeTarget, xt_trader: xttrader.XtQuantTrader, account: StockAccount):
        super().__init__() #必须调用父类的初始化方法
        self.tradeTarget = tradeTarget
        self.xt_trader = xt_trader
        self.account = account

    def run(self) -> None:
        print(f"启动标的交易线程 {self.tradeTarget.stock_code} {getInstrumentName(self.tradeTarget.stock_code)}\n")
        while True:
            print('{} is running >> {}'.format(threading.current_thread().name, self.tradeTarget.stock_code))
            time.sleep(2)
    
    
    # Description: 程序启动后
    def check_stock_position(self):
        volume = getStockPosition(self.stock_code, self.xt_trader, self.account)
        pass

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

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