diff --git a/HTS_Downloader.py b/HTS_Downloader.py new file mode 100644 index 0000000..9791a38 --- /dev/null +++ b/HTS_Downloader.py @@ -0,0 +1,37 @@ +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...")