import os
from datetime import datetime
from HTS_etf import HTS_etf

if __name__ == "__main__":
    today = datetime.today()

    PROJECT_HOME = os.getcwd()
    RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources")

    stocks = [
        {'stock_code': '122630', 'stock_name': 'KODEX 레버리지'},
        {'stock_code': '233740', 'stock_name': 'KODEX 코스닥150레버리지'},
        {'stock_code': '251340', 'stock_name': 'KODEX 코스닥150선물인버스'},
        {'stock_code': '252670', 'stock_name': 'KODEX 200선물인버스2X'}
    ]

    hts = HTS_etf(RESOURCE_PATH)
    hts.connect2DB("hts.db")

    today_str = today.strftime('%Y%m%d')

    if not os.path.exists(os.path.join(RESOURCE_PATH, "log")):
        os.mkdir(os.path.join(RESOURCE_PATH, "log"))

    MAX_PRICE = 500000
    hts.buyRealTime(stocks, today_str, MAX_PRICE=MAX_PRICE)

    db_filename = os.path.join(RESOURCE_PATH, "hts.db")
    for stock in stocks:
        hts.insertStockData(today, stock['stock_code'], stock['stock_name'])
        hts.bot.sendMsg("done... {} ({})".format(stock['stock_code'], stock['stock_name']))
    hts.disconnect()

    print("done...")