From 8fdeba4ea52eadd508da4be011c7aa53dd4fb80c Mon Sep 17 00:00:00 2001 From: "dosang.yoon" Date: Tue, 30 Aug 2022 14:37:23 +0900 Subject: [PATCH] init --- hts/BuySellChecker.py | 19 ++++++++++++++++--- 1 file changed, 16 insertions(+), 3 deletions(-) diff --git a/hts/BuySellChecker.py b/hts/BuySellChecker.py index a8dd16b..93b750d 100644 --- a/hts/BuySellChecker.py +++ b/hts/BuySellChecker.py @@ -14,6 +14,8 @@ class BuySellChecker: macd = None ichimokuCloud = None + buy_type = -1 + def __init__(self): self.common = Common() self.stochastic = Stochastic() @@ -21,6 +23,8 @@ class BuySellChecker: self.macd = MACD() self.ichimokuCloud = IchimokuCloud() + self.buy_type = -1 + return @@ -978,8 +982,18 @@ class BuySellChecker: buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, last_index) sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, last_index) - bsLine['buy'] = [buy] - bsLine['buy_weight'] = [buy_weight] + if buy_type == 1: + self.buy_type = 1 + if buy_type == 0 or sell_type == 1: + self.buy_type = -1 + + if self.buy_type == 1: + bsLine['buy'] = [buy] + bsLine['buy_weight'] = [buy_weight] + else: + bsLine['buy'] = [-1] + bsLine['buy_weight'] = [-1] + bsLine['sell'] = [sell] bsLine['sell_weight'] = [sell_weight] else: @@ -989,7 +1003,6 @@ class BuySellChecker: bsLine['sell'] = [-1 for i in range(size)] bsLine['sell_weight'] = [-1 for i in range(size)] - self.buy_type = -1 for i in range(size): buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, i) sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, i)