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

init

Browse Source
This commit is contained in:
dosang.yoon
2022-08-30 14:37:23 +09:00
parent 42710247be
commit 8fdeba4ea5
1 changed files with 16 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
hts/BuySellChecker.py
View File

@@ -14,6 +14,8 @@ class BuySellChecker:
macd = None macd = None
ichimokuCloud = None ichimokuCloud = None
buy_type = -1
def __init__(self): def __init__(self):
self.common = Common() self.common = Common()
self.stochastic = Stochastic() self.stochastic = Stochastic()
@@ -21,6 +23,8 @@ class BuySellChecker:
self.macd = MACD() self.macd = MACD()
self.ichimokuCloud = IchimokuCloud() self.ichimokuCloud = IchimokuCloud()
self.buy_type = -1
return return
@@ -978,8 +982,18 @@ class BuySellChecker:
buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, last_index) buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, last_index)
sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, last_index) sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, last_index)
bsLine['buy'] = [buy] if buy_type == 1:
bsLine['buy_weight'] = [buy_weight] 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'] = [sell]
bsLine['sell_weight'] = [sell_weight] bsLine['sell_weight'] = [sell_weight]
else: else:
@@ -989,7 +1003,6 @@ class BuySellChecker:
bsLine['sell'] = [-1 for i in range(size)] bsLine['sell'] = [-1 for i in range(size)]
bsLine['sell_weight'] = [-1 for i in range(size)] bsLine['sell_weight'] = [-1 for i in range(size)]
self.buy_type = -1
for i in range(size): for i in range(size):
buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, i) buy, buy_weight, buy_type = self.getBuyPriceAndWeight(data, i)
sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, i) sell, sell_weight, sell_type = self.getSellPriceAndWeight(data, i)