dsyoon/DeepStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

init

Browse Source
This commit is contained in:
dsyoon
2023-02-03 21:46:17 +09:00
parent d27610fe67
commit 9a2fc96e42
3 changed files with 54 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

53
HTS_Downloader.py
View File

@@ -21,17 +21,64 @@ if __name__ == "__main__":
# 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()}
{"stock_code": "122630", "stock_name": "KODEX 레버리지", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "305720", "stock_name": "KODEX 2차전지산업", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "102780", "stock_name": "KODEX 삼성그룹", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139260", "stock_name": "TIGER 200 IT", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "091180", "stock_name": "KODEX 자동차", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "401470", "stock_name": "KODEX K-메타버스액티브", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "329200", "stock_name": "TIGER 리츠부동산인프라", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "091170", "stock_name": "KODEX 은행", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "091160", "stock_name": "KODEX 반도체", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "161510", "stock_name": "ARIRANG 고배당주", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "228800", "stock_name": "TIGER 여행레저", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "150460", "stock_name": "TIGER 중국소비테마", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "143860", "stock_name": "TIGER 헬스케어", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "228810", "stock_name": "TIGER 미디어컨텐츠", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139220", "stock_name": "TIGER 200 건설", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139280", "stock_name": "TIGER 경기방어", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "322400", "stock_name": "HANARO e커머스", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "157490", "stock_name": "TIGER 소프트웨어", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "228790", "stock_name": "TIGER 화장품", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139230", "stock_name": "TIGER 200 중공업", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "396500", "stock_name": "TIGER Fn반도체TOP10", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "365000", "stock_name": "TIGER KRX인터넷K-뉴딜", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "102970", "stock_name": "KODEX 증권", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "117680", "stock_name": "KODEX 철강", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "244580", "stock_name": "KODEX 바이오", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "266360", "stock_name": "KODEX 미디어&엔터테인먼트", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "375770", "stock_name": "KODEX 탄소효율그린뉴딜", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "364990", "stock_name": "TIGER KRX게임K-뉴딜", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "388420", "stock_name": "KBSTAR 비메모리반도체액티브", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "117460", "stock_name": "KODEX 에너지화학", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "300950", "stock_name": "KODEX 게임산업", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "266410", "stock_name": "KODEX 필수소비재", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "140700", "stock_name": "KODEX 보험", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139270", "stock_name": "TIGER 200 금융", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "395160", "stock_name": "KODEX Fn시스템반도체", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "140710", "stock_name": "KODEX 운송", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "139240", "stock_name": "TIGER 200 철강소재", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "395150", "stock_name": "KODEX Fn웹툰&드라마", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "307510", "stock_name": "TIGER 의료기기", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "315270", "stock_name": "TIGER 200커뮤니케이션서비스", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "132030", "stock_name": "KODEX 골드선물(H)", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "144600", "stock_name": "KODEX 은선물(H)", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "261220", "stock_name": "KODEX WTI원유선물(H)", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "271050", "stock_name": "KODEX WTI원유선물인버스(H)", "start_date": datetime.strptime("20230120",'%Y%m%d'), "end_date": datetime.today()},
{"stock_code": "138910", "stock_name": "KODEX 구리선물(H)", "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")
hts.connect2DB("hts.db")
for stock in stocks:
this_day = stock["start_date"]
print("insert...", stock["stock_code"], stock["stock_name"], this_day)
while True:
hts.insertStockData(db_filename, stock["stock_code"], stock["stock_name"], this_day.strftime('%Y%m%d'))
hts.insertStockData(stocks, this_day)
this_day = this_day + timedelta(days=1)
if this_day > stock["end_date"]:
break
hts.disconnect()
print ("done...")

3
HTS_etf.py
View File

@@ -21,7 +21,6 @@ class HTS_etf (HTS):
labelChecker = None
slackBot = None
stockStatus = None
MAX_BUY_PRICE = None
def __init__(self, RESOURCE_PATH):
super().__init__(RESOURCE_PATH)
@@ -83,7 +82,7 @@ class HTS_etf (HTS):
def getCount(self, stock_code, price):
if stock_code == "252670" or stock_code == "122630":
MAX_BUY_PRICE = 300000
MAX_BUY_PRICE = 500000
else:
MAX_BUY_PRICE = 100000

10
hts/HTS.py
View File

@@ -476,7 +476,7 @@ class HTS:
return data
def insertStockData(self, today, stocks):
def insertStockData(self, stocks, this_day):
# 테이블 생성
self.cursor.execute("CREATE TABLE IF NOT EXISTS hts (CODE text, NAME text, ymd text, hms text, close REAL, open REAL, high REAL, low REAL, volume REAL, label INTEGER DEFAULT 0)")
@@ -486,9 +486,9 @@ class HTS:
self.cursor.execute(create_key)
for stock in stocks:
items = self.getStockInfo(stock["stock_code"], today)
idx = 0
items = self.getStockInfo(stock["stock_code"], this_day.strftime('%Y%m%d'))
for item in items:
ymd = item[0]
hms = item[1]
@@ -498,14 +498,10 @@ class HTS:
close = item[5]
volume = item[6]
idx += 1
self.cursor.execute('DELETE FROM hts WHERE CODE=? and ymd=? and hms=?', (stock["stock_code"], ymd, hms,))
self.cursor.execute("INSERT INTO hts (CODE, NAME, ymd, hms, close, open, high, low, volume) VALUES(?, ?, ?, ?, ?, ?, ?, ?, ?)", (stock["stock_code"], stock["stock_name"], ymd, hms, close, open, high, low, volume))
self.conn.commit()
print("insert...", stock["stock_code"], stock["stock_name"], today)
return