Files
DeepStock/HTS_Downloader.py
dsyoon 763fd79acd init
2023-01-29 21:16:10 +09:00

38 lines
1.1 KiB
Python

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("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "122630", "stock_name": "KODEX 레버리지", "start_date": datetime.strptime("20230120",'%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...")