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
2021-10-08 15:44:01 +09:00
parent c6254cbe12
commit 59b3fb64d3
1 changed files with 6 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
hts/HTS.py
View File

@@ -1,10 +1,10 @@
#import win32com.client import win32com.client
import time import time
import os import os
from datetime import datetime, timedelta from datetime import datetime, timedelta
import pandas as pd import pandas as pd
from enum import Enum from enum import Enum
import plotly.graph_objects as go #import plotly.graph_objects as go
from stockpredictor.analysis.Common import Common from stockpredictor.analysis.Common import Common
# enum 주문 상태 세팅용 # enum 주문 상태 세팅용
@@ -781,7 +781,7 @@ class HTS:
size = len(data["Close"]) size = len(data["Close"])
STOCK = [] STOCK = []
for i in range(size): for i in range(size):
STOCK.append({'volume': data['Volume'][i], 'close': data["Close"][i], 'open': data["Open"][i], 'high': data["High"][i], 'low': data["Low"][i], 'avg5': data["avg5"][i], 'avg20': data["avg20"][i], 'avg60': data["avg60"][i], 'avg120': data["avg120"][i], 'avg240': data["avg240"][i]}) STOCK.append({'volume': data['Volume'][i], 'close': data["Close"][i], 'open': data["Open"][i], 'high': data["High"][i], 'low': data["Low"][i], 'avg2': data["avg2"][i], 'avg5': data["avg5"][i], 'avg10': data["avg10"][i], 'avg20': data["avg20"][i], 'avg30': data["avg30"], 'avg40': data["avg40"], 'avg50': data["avg50"], 'avg60': data["avg60"][i]})
bsLine = {} bsLine = {}
bsLine['buy'] = [-1 for i in range(size)] bsLine['buy'] = [-1 for i in range(size)]
@@ -960,7 +960,7 @@ class HTS:
while datetime.strptime(given_day + " 083000", '%Y%m%d %H%M%S') < datetime.now() < datetime.strptime(given_day + " 15200", '%Y%m%d %H%M%S'): while datetime.strptime(given_day + " 083000", '%Y%m%d %H%M%S') < datetime.now() < datetime.strptime(given_day + " 15200", '%Y%m%d %H%M%S'):
second = datetime.now().strftime('%Y%m%d %H%M%S') second = datetime.now().strftime('%Y%m%d %H%M%S')
if datetime.now() < datetime.strptime(given_day + " 144000", '%Y%m%d %H%M%S'): if datetime.now() < datetime.strptime(given_day + " 151000", '%Y%m%d %H%M%S'):
if second in timecheck and not timecheck[second]: if second in timecheck and not timecheck[second]:
print("TIMECHECK", second) print("TIMECHECK", second)
logFp.write("%s,%s,\n" %("TIMECHECK", second)) logFp.write("%s,%s,\n" %("TIMECHECK", second))
@@ -971,7 +971,6 @@ class HTS:
# 분석을 통해서 볼린저밴드 상/하단을 계산한다. # 분석을 통해서 볼린저밴드 상/하단을 계산한다.
data, upper, lower = self.analyze(result) data, upper, lower = self.analyze(result)
final_price = data["Close"][len(data["Close"])- 1]
# 사야 할 시점/가격과 팔아야 할 시점/가격을 체크한다. # 사야 할 시점/가격과 팔아야 할 시점/가격을 체크한다.
bs_buy_price, bs_sell_price = self.checkTransaction_Realtime(data, upper, lower) bs_buy_price, bs_sell_price = self.checkTransaction_Realtime(data, upper, lower)
@@ -1068,7 +1067,7 @@ if __name__ == "__main__":
given_day = datetime.today().strftime('%Y%m%d') given_day = datetime.today().strftime('%Y%m%d')
#hts.writeStockData(stock_codes, given_day) #hts.writeStockData(stock_codes, given_day)
hts.simulate(stock_codes[0], given_day) #hts.simulate(stock_codes[0], given_day)
#hts.buyRealTime(stock_codes[0], given_day) hts.buyRealTime(stock_codes[0], given_day)
print ("done...") print ("done...")