From bd7d56e1028ccb311e47d47ea418afe8fa420d2d Mon Sep 17 00:00:00 2001 From: dsyoon Date: Sun, 3 Oct 2021 19:06:35 +0900 Subject: [PATCH] init --- hts/HTS.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/hts/HTS.py b/hts/HTS.py index 8537712..6170e7e 100644 --- a/hts/HTS.py +++ b/hts/HTS.py @@ -546,7 +546,7 @@ class HTS: return data, upper, lower - def draw(self, given_day, data, upper, lower, bsLine): + def draw(self, stock_code, given_day, data, upper, lower, bsLine): buy_line = bsLine['buy'] sell_line = bsLine['sell'] @@ -593,7 +593,7 @@ class HTS: # 그래프를 그린다. fig = go.Figure(data=[candle_stick, bolinger_upper, bolinger_lower, buy_check, sell_check, avg5, avg20, avg60, avg120, avg240]) - fig.update_layout(title=given_day + "_2x") + fig.update_layout(title=stock_code + "_" + given_day) fig.show() return @@ -845,7 +845,7 @@ class HTS: bsLine = self.checkTransaction_Simulation(data, upper, lower) # 그래프를 그린다. - self.draw(given_day, data, upper, lower, bsLine) + self.draw(stock_code, given_day, data, upper, lower, bsLine) # 가져온 만큼 데이터를 누적해서 파일로 작성한다. self.write(given_day, result) @@ -982,7 +982,8 @@ if __name__ == "__main__": given_days = ['20210901','20210902','20210903','20210906','20210907','20210908','20210909','20210910','20210913','20210914','20210915','20210916','20210917','20210923','20210924','20210927','20210928','20210929','20210930','20211001'] for given_day in given_days: hts.writeStockData(stock_codes, given_day) - #hts.simulate(stock_codes, given_day) + #for stock_code in stock_codes: + #hts.simulate(stock_codes, given_day) #hts.buyRealTime(stock_codes, given_day)