import time import os from datetime import datetime, timedelta from hts.HTS import HTS from hts.OrderType import OrderType from hts.BuySellChecker import BuySellChecker from hts.OrderChecker import OrderChecker from stock.util.LabelChecker import LabelChecker from stock.util.SlackBot import SlackBot from stock.analysis.StockStatus import StockStatus class HTS_252670 (HTS): RESOURCE_PATH = None stock_code = None buy_count = None orderChecker = None buySellChecker = None labelChecker = None slackBot = None stockStatus = None def __init__(self, RESOURCE_PATH, stock_code, buy_count): super().__init__(RESOURCE_PATH) self.RESOURCE_PATH = RESOURCE_PATH self.stock_code = stock_code self.buy_count = buy_count self.orderChecker = OrderChecker(self.RESOURCE_PATH) self.buySellChecker = BuySellChecker() self.labelChecker = LabelChecker(RESOURCE_PATH) self.slackBot = SlackBot() self.stockStatus = StockStatus(RESOURCE_PATH) return def getSellingPrice(self, log_time, stock_code, final_price, check=False): # final_price와 diff를 받으면, 해당 가격으로 그냥 매도한다는 의미 # final_price와 diff가 None이면 장부가와 final 중 max로 팔겠다는 의미 # final_price가 0이고 diff가 None이면 장부가로 팔겠다는 의미임 sell_price = -1 jangoDic = self.requstJango() if jangoDic and len(jangoDic.keys()) > 0: for code in jangoDic: if jangoDic[code]['매도가능'] > 0: if check: if jangoDic[code]['장부가']*0.05 < jangoDic[code]['장부가'] - final_price: sell_price = jangoDic[code]['장부가'] if code == "A" + stock_code: orderNum = self.requestOrder(OrderType.sell, stock_code, jangoDic[code]['매도가능'], sell_price) print("ORDER_SELL", stock_code, log_time.strftime('%Y%m%d %H%M%S'), jangoDic[code]['매도가능'], sell_price) else: max_price = max(jangoDic[code]['장부가'], final_price) sell_price = (int(max_price) - int(max_price) % 5) + 5 if code == "A"+stock_code: orderNum = self.requestOrder(OrderType.sell, stock_code, jangoDic[code]['매도가능'], sell_price) print("ORDER_SELL", stock_code, log_time.strftime('%Y%m%d %H%M%S'), jangoDic[code]['매도가능'], sell_price) return def makeTickData(self, data, mins=30): result = {"check": set(), "time": [], "open": [], "close": [], "high": [], "low": [], "vol": [], "label": []} for i in range(mins, len(data['time'])+1): result["check"].add(data['time'][i-1]) result["time"].append(data['time'][i-1]) result["open"].append(data['open'][i-mins]) result["close"].append(data['close'][i-1]) result["high"].append(max(data['high'][i - mins: i])) result["low"].append(min(data['low'][i - mins: i])) result["vol"].append(sum(data['vol'][i - mins: i])) return result def buyRealTime(self, today, analyzed_day=1000): print ("START...") THIS_TIME = datetime.now() final_sell_check = False LAST_DATA = self.getLastData(self.stock_code, today) while datetime.strptime(today + " 070000", '%Y%m%d %H%M%S') < THIS_TIME < datetime.strptime(today + " 153100", '%Y%m%d %H%M%S'): if datetime.strptime(today + " 090000", '%Y%m%d %H%M%S') < THIS_TIME < datetime.strptime(today + " 151500", '%Y%m%d %H%M%S'): # 3시 까지만 매수를 시도한다. if THIS_TIME.strftime('%S') == "03": # 매분 3초마다 실행한다. # 데이터를 가지고 온다. result = self.getRealTime(self.stock_code, today, LAST_DATA) result_5 = self.makeTickData(result, mins=5) result_30 = self.makeTickData(result, mins=30) data = self.buySellChecker.analyze(result) data.drop(data.index[:len(data) - analyzed_day], inplace=True) # 이동평균, RSI, MACD, 일목균형, 볼린저밴드 상/하단을 계산한다. data_5 = self.buySellChecker.analyze(result_5) # 분석일 데이터만 활용한다 (이전 데이터는 제거) data_5.drop(data_5.index[:len(data_5) - analyzed_day], inplace=True) data_30 = self.buySellChecker.analyze(result_30) # 분석일 데이터만 활용한다 (이전 데이터는 제거) data_30.drop(data_30.index[:len(data_30) - analyzed_day], inplace=True) # 사야 할 시점과 팔아야 할 시점을 체크한다. bsLine = self.buySellChecker.checkTransaction(stock_code, data, data_5, data_30, isRealTime=True) bs_buy_price = bsLine['buy'][0] bs_buy_weight = bsLine['buy_weight'][0] bs_sell_price = bsLine['sell'][0] data_size = len(data["close"]) final_price = data["close"][data_size-1] if bs_buy_price > 0: # 기본 100 주에 가중치를 추가해서 매수한다. BUY_COUNT = int(self.buy_count * bs_buy_weight) # 매수를 주문한다. orderNum = self.requestOrder(OrderType.buy, self.stock_code, BUY_COUNT , bs_buy_price) # slackbot에 메시지를 보냄 self.slackBot.post_to_slack(stock_code, stock_name, "BUY", bsLine['buy'][len(bsLine['buy']) - 1], buy_count) self.orderChecker.add(today, stock_code, 1, buy_count, bs_buy_price, orderNum) # 로그 출력 print("BUY", THIS_TIME.strftime('%Y%m%d %H%M%S'), orderNum, stock_code, stock_name, bs_buy_price, buy_count) if bs_sell_price > 0: # 매도한다. orderNum = self.getSellingPrice(THIS_TIME, self.stock_code, final_price) # slackbot에 메시지를 보냄 self.slackBot.post_to_slack(stock_code, stock_name, "SELL", bsLine['sell'][len(bsLine['sell']) - 1], -1) self.orderChecker.delete(today, stock_code) # 로그 출력 print("SELL", THIS_TIME.strftime('%Y%m%d %H%M%S'), str(orderNum), stock_code, stock_name, bs_sell_price) # 로그 출력 print("TIMECHECK: %s, price: %d, avg5: %.2f, avg30: %.2f, slow_k: %.2f, open: %d, high: %d, low: %d" % (str(THIS_TIME), final_price, data["avg5"][data_size - 1], data["avg30"][data_size - 1], data["slow_k"][data_size - 2], data["open"][data_size - 1], data["high"][data_size - 1], data["low"][data_size - 1],)) elif datetime.strptime(today + " 151530", '%Y%m%d %H%M%S') < THIS_TIME < datetime.strptime(today + " 151600", '%Y%m%d %H%M%S'): # 3시 15분 30초부터 3시 16분 사이는 잔량을 매도한다. if not final_sell_check: #### # 손해 보지 않는 가격에 매도한다. #### # 주문 리스트를 가져온다. orderList = self.requestOrderList() # 15:10:00 이후라면 모든 미체결 취소한다. self.cancelOrderList(orderList) # 매도 가격을 가져온다. result = self.getRealTime(self.stock_code, today, LAST_DATA) final_price = result["close"][len(result["close"]) - 1] self.getSellingPrice(THIS_TIME, self.stock_code, final_price) final_sell_check = True time.sleep(0.9) THIS_TIME = datetime.now() return def updteTodayStock(self, db_filename, stock_code, today_str): bsLine, data = self.labelChecker.makeCandidate(stock_code, today_str) self.labelChecker.updateLabel(db_filename, stock_code, bsLine, data, today_str) return if __name__ == "__main__": today = datetime.today() PROJECT_HOME = os.getcwd() RESOURCE_PATH = os.path.join(PROJECT_HOME, "resources") # KODEX 인버스 * 2 stock_code = "252670" stock_name = "KODEX 200선물인버스2X" buy_count = 2000 hts = HTS_252670(RESOURCE_PATH, stock_code, buy_count) today_str = today.strftime('%Y%m%d') #today_str = "20220916" 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...")