import os
from datetime import datetime, timedelta

from hts.HTS import HTS

class HTS_Downloader (HTS):

    def __init__(self, RESOURCE_PATH):
        super().__init__(RESOURCE_PATH)

        return


if __name__ == "__main__":

    today = datetime.today()

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

    # KODEX 인버스 * 2
    stocks = [
        {"stock_code": "252670", "stock_name": "KODEX 200선물인버스2X", "start_date": datetime.strptime("20221121",'%Y%m%d'), "end_date": datetime.today()},
        {"stock_code": "122630", "stock_name": "KODEX 레버리지", "start_date": datetime.strptime("20220916",'%Y%m%d'), "end_date": datetime.today()}
    ]

    hts = HTS_Downloader(RESOURCE_PATH)
    db_filename = os.path.join(RESOURCE_PATH, "hts.db")
    for stock in stocks:
        this_day = stock["start_date"]
        while True:
            hts.insertStockData(db_filename, stock["stock_code"], stock["stock_name"], this_day.strftime('%Y%m%d'))
            this_day = this_day + timedelta(days=1)
            if this_day > stock["end_date"]:
                break

    print ("done...")