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

init

Browse Source
This commit is contained in:
dosangyoon
2021-10-03 21:06:05 +09:00
parent 015545d8e0
commit d30cfed3f7
1 changed files with 14 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

27
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 주문 상태 세팅용
@@ -788,7 +788,7 @@ class HTS:
bsLine['buy'] = [-1 for i in range(len(lower))] bsLine['buy'] = [-1 for i in range(len(lower))]
bsLine['sell'] = [-1 for i in range(len(lower))] bsLine['sell'] = [-1 for i in range(len(lower))]
for i in range(5, size-1): for i in range(5, size-5):
status = self.checkStatus(STOCK, i) status = self.checkStatus(STOCK, i)
count_1 = 0 count_1 = 0
if "arrange_" in status: count_1 += 1 if "arrange_" in status: count_1 += 1
@@ -819,9 +819,9 @@ class HTS:
# 시뮬레이션은 이번에 사고 파는 것으로 판단했기 때문에, 다음 봉에서 사고 판 위치를 표시한다. # 시뮬레이션은 이번에 사고 파는 것으로 판단했기 때문에, 다음 봉에서 사고 판 위치를 표시한다.
if count_0 == 0 and count_1 > 0: if count_0 == 0 and count_1 > 0:
bsLine['buy'][i + 1] = STOCK[i]['close'] - 5 bsLine['buy'][i + 1] = STOCK[i]['close'] - 5
bsLine['sell'][i + 1] = STOCK[i]['close'] bsLine['sell'][i + 2] = STOCK[i]['close']
if count_0 > 0: #if count_0 > 0:
bsLine['sell'][i+1] = STOCK[i]['close'] + 5 # bsLine['sell'][i + 2] = STOCK[i]['close'] + 5
return bsLine return bsLine
@@ -851,7 +851,7 @@ class HTS:
self.draw(stock_code, given_day, data, upper, lower, bsLine) self.draw(stock_code, given_day, data, upper, lower, bsLine)
# 가져온 만큼 데이터를 누적해서 파일로 작성한다. # 가져온 만큼 데이터를 누적해서 파일로 작성한다.
self.write(given_day, result) # self.write(given_day, result)
return return
@@ -962,19 +962,20 @@ if __name__ == "__main__":
RESOURCE_DIR = PROJECT_HOME + "/resources/analysis/"+today.strftime("%Y%m%d") RESOURCE_DIR = PROJECT_HOME + "/resources/analysis/"+today.strftime("%Y%m%d")
stock_codes = ["252670", "122630"] stock_codes = ["252670", "122630"]
#given_days = ['20210901','20210902','20210903','20210906','20210907','20210908','20210909','20210910','20210913','20210914','20210915','20210916','20210917','20210923','20210924','20210927','20210928','20210929','20210930','20211001'] stock_codes = ["252670"]
given_days = ['20210901','20210902','20210903','20210906','20210907','20210908','20210909','20210910','20210913','20210914','20210915','20210916','20210917','20210923','20210924','20210927','20210928','20210929','20210930','20211001']
hts = HTS() hts = HTS()
#hts.all_stocks() #hts.all_stocks()
#hts.getChartData(stock_codes) #hts.getChartData(stock_codes)
#hts.currentStock(stock_codes) #hts.currentStock(stock_codes)
#for given_day in given_days: for given_day in given_days:
#hts.writeStockData(stock_codes, given_day) #hts.writeStockData(stock_codes, given_day)
#for stock_code in stock_codes: for stock_code in stock_codes:
# hts.simulate(stock_code, given_day) hts.simulate(stock_code, given_day)
given_day = datetime.today().strftime('%Y%m%d') #given_day = datetime.today().strftime('%Y%m%d')
hts.buyRealTime(stock_codes[0], given_day) #hts.buyRealTime(stock_codes[0], given_day)
print ("done...") print ("done...")