diff --git a/HTS_122630.py b/HTS_122630.py index 07d2bc9..42821c4 100644 --- a/HTS_122630.py +++ b/HTS_122630.py @@ -8,7 +8,7 @@ from hts.OrderType import OrderType from hts.BuySellChecker import BuySellChecker from hts.OrderChecker import OrderChecker - +from stock.util.LabelMaker import LabelMaker class HTS_122630 (HTS): RESOURCE_PATH = None @@ -16,6 +16,7 @@ class HTS_122630 (HTS): buy_count = None orderChecker = None buySellChecker = None + labelMaker = None def __init__(self, RESOURCE_PATH, stock_code, buy_count): super().__init__(RESOURCE_PATH) @@ -26,6 +27,7 @@ class HTS_122630 (HTS): self.orderChecker = OrderChecker(self.RESOURCE_PATH, self.stock_code) self.buySellChecker = BuySellChecker() + self.labelMaker = LabelMaker() return def getSellingPrice(self, final_price): @@ -177,6 +179,11 @@ class HTS_122630 (HTS): return + def updteTodayStock(self, db_filename, stock_code, today_str): + bsLine, data = self.labelMaker.makeCandidate(stock_code, today_str) + self.labelMaker.updateLabel(db_filename, stock_code, bsLine, data, today_str) + return + if __name__ == "__main__": today = datetime.today() @@ -195,6 +202,6 @@ if __name__ == "__main__": hts.buyRealTime(today_str) db_filename = os.path.join(RESOURCE_PATH, "hts.db") - hts.insertStockData(db_filename, stock_code, stock_name, today_str) + hts.updteTodayStock(db_filename, stock_code, today_str) print ("done...") diff --git a/HTS_252670.py b/HTS_252670.py index 97bb909..54add6d 100644 --- a/HTS_252670.py +++ b/HTS_252670.py @@ -8,7 +8,7 @@ from hts.OrderType import OrderType from hts.BuySellChecker import BuySellChecker from hts.OrderChecker import OrderChecker - +from stock.util.LabelMaker import LabelMaker class HTS_252670 (HTS): @@ -17,6 +17,7 @@ class HTS_252670 (HTS): buy_count = None orderChecker = None buySellChecker = None + labelMaker = None def __init__(self, RESOURCE_PATH, stock_code, buy_count): super().__init__(RESOURCE_PATH) @@ -27,6 +28,7 @@ class HTS_252670 (HTS): self.orderChecker = OrderChecker(self.RESOURCE_PATH, self.stock_code) self.buySellChecker = BuySellChecker() + self.labelMaker = LabelMaker() return def getSellingPrice(self, final_price): @@ -178,6 +180,12 @@ class HTS_252670 (HTS): return + def updteTodayStock(self, db_filename, stock_code, today_str): + bsLine, data = self.labelMaker.makeCandidate(stock_code, today_str) + self.labelMaker.updateLabel(db_filename, stock_code, bsLine, data, today_str) + return + + if __name__ == "__main__": today = datetime.today() @@ -196,6 +204,6 @@ if __name__ == "__main__": hts.buyRealTime(today_str) db_filename = os.path.join(RESOURCE_PATH, "hts.db") - hts.insertStockData(db_filename, stock_code, stock_name, today_str) + hts.updteTodayStock(db_filename, stock_code, today_str) print ("done...")