From c3ecc09239dc611c2fc101f1a8ba00404101cac0 Mon Sep 17 00:00:00 2001 From: dsyoon Date: Sat, 16 Oct 2021 15:22:22 +0900 Subject: [PATCH] init --- hts/HTS.py | 4 ++-- hts/OrderChecker.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/hts/HTS.py b/hts/HTS.py index 09a5e52..617358a 100644 --- a/hts/HTS.py +++ b/hts/HTS.py @@ -161,7 +161,7 @@ class HTS: # acc = "782446178" # accFlag[0] = "01" objStockOrder = win32com.client.Dispatch("CpTrade.CpTd0311") - objStockOrder.SetInputValue(0, type) # 1: 매도, 2: 매수 + objStockOrder.SetInputValue(0, type.value) # 1: 매도, 2: 매수 objStockOrder.SetInputValue(1, acc) # 계좌번호 objStockOrder.SetInputValue(2, accFlag[0]) # 상품구분 - 주식 상품 중 첫번째 objStockOrder.SetInputValue(3, "A"+stock_code) # 종목코드 @@ -588,7 +588,7 @@ class HTS: # 미체결 기록을 가져온다. orderList = self.requestOrderList() # 매수 주문을 기록한다. - orderListToCancel = orderChecker.add("2", orderNum, BUY_COUNT, bs_buy_price, orderList) + orderListToCancel = orderChecker.add(OrderType.buy, orderNum, BUY_COUNT, bs_buy_price, orderList) # 두 시간 이전 미체결을 취소한다. self.cancelOrderList(orderListToCancel) # 로그 출력 diff --git a/hts/OrderChecker.py b/hts/OrderChecker.py index 66416a0..c3bc9b5 100644 --- a/hts/OrderChecker.py +++ b/hts/OrderChecker.py @@ -33,7 +33,7 @@ class OrderChecker: if len(orderList) > 0: # 만약 두시간 전 주문을 찾는다. orderNum_df = self.order_df.loc[self.order_df["datetime"] <= before_two_hour] - orderNum_df = orderNum_df.loc[orderNum_df["type"] == OrderType.buy] + orderNum_df = orderNum_df.loc[orderNum_df["type"] == OrderType.buy.value] orderNumSet = set(list(orderNum_df["orderNum"])) for item in orderList: if item.orderNum in orderNumSet: @@ -41,7 +41,7 @@ class OrderChecker: # 해당 orderNum 제외하기 self.order_df = self.order_df.loc[self.order_df["orderNum"] != item.orderNum] - self.order_df = self.order_df.append({"type": type, "orderNum": orderNum, "count": count, "price": price, "datetime": datetime.now()}, ignore_index=True) + self.order_df = self.order_df.append({"type": type.value, "orderNum": orderNum, "count": count, "price": price, "datetime": datetime.now()}, ignore_index=True) self.order_df.to_csv(self.saveFileName) return orderListToCancel