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-10-09 23:11:52 +09:00
parent 522a5ff449
commit b9d551bbe8
4 changed files with 479 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
hts/HTS.py
View File

@@ -463,7 +463,7 @@ class HTS:
return data
def insertStockData(self, stocks, this_day):
def insertStockData(self, stock_code, 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)")
@@ -472,23 +472,21 @@ class HTS:
create_key = "CREATE INDEX IF NOT EXISTS hts_idx on hts(CODE, ymd, hms) "
self.cursor.execute(create_key)
for stock in stocks:
items = self.getStockInfo(stock_code, this_day.strftime('%Y%m%d'))
items = self.getStockInfo(stock["stock_code"], this_day.strftime('%Y%m%d'))
for item in items:
ymd = item[0]
hms = item[1]
open = item[2]
high = item[3]
low = item[4]
close = item[5]
volume = item[6]
for item in items:
ymd = item[0]
hms = item[1]
open = item[2]
high = item[3]
low = item[4]
close = item[5]
volume = item[6]
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.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()
self.conn.commit()
return
@@ -793,6 +791,7 @@ class HTS:
"disparity_avg20": -1,
"disparity_avg60": -1,
"disparity_avg120": -1,
"disparity_avg200": -1,
"bolingerband_upper": -1,
"bolingerband_lower": -1,
"bolingerband_middle": -1,