init

hts/OrderChecker.py

@@ -17,7 +17,7 @@ class OrderChecker:
         if os.path.isfile(saveFileName):
             order_df = pd.read_csv(saveFileName)
         else:
-            # type: 매수는 0 매도는 1
+            # (type) buy=0, sell=1
             order_df = pd.DataFrame(columns=["datetime", "stock_code", "type", "orderNum", "count", "price"])
 
         if 'Unnamed: 0' in order_df.columns:
@@ -40,26 +40,21 @@ class OrderChecker:
             return False
         return True
 
-    def add(self, ymd, stock_code: str, type: int, count: int, price: int, orderNum=None):
-        if self.exist(ymd, stock_code):
-            return False
+    def buy(self, ymd, stock_code: str, count: int, price: int, orderNum=""):
 
         order_df = self.read(ymd)
 
         # 새로운 주문을 추가한다.
-        order_df = order_df.append({"stock_code": "A" + stock_code, "type": type, "orderNum": 0, "count": count, "price": price, "datetime": datetime.now()}, ignore_index=True)
-        order_df = order_df.astype({"stock_code": str, "type": int, "orderNum": int, "count": int, "price": int})
+        order_df = order_df.append({"stock_code": "A" + stock_code, "type": 0, "orderNum": str(orderNum), "canceled": 0, "count": count, "price": price, "datetime": datetime.now()}, ignore_index=True)
+        order_df = order_df.astype({"stock_code": str, "type": int, "orderNum": str, "canceled": int, "count": int, "price": int})
 
         # 파일로 기록한다.
-        saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv")
+        saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv")
         order_df.to_csv(saveFileName)
 
         return True
 
-    def delete(self, ymd, stock_code: str):
-
-        if not self.exist(ymd, stock_code):
-            return False
+    def sell(self, ymd, stock_code: str):
 
         order_df = self.read(ymd)
 
@@ -67,7 +62,25 @@ class OrderChecker:
         order_df.loc[(order_df["stock_code"] == "A" + stock_code, 'type')] = 1
 
         # 파일로 기록 한다.
-        saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv")
+        saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv")
         order_df.to_csv(saveFileName)
 
+        return True
+
+    def cancel(self, ymd, stock_code, ORDER_LIST, min=10):
+
+        order_df = self.read(ymd)
+
+        now = datetime.now() - timedelta(minutes=min)
+        # min 분 이상 된 시간인 내용을 가지고 옴
+        df = order_df.loc[(order_df.index <= now) ]
+        # 취소가 되지 않은 것만 가지고 옴 (0: 취소 되지 않음, 1: 취소함)
+        df = df.loc[(order_df["canceled"] == 0)]
+
+        for i in range(len(df)):
+            order_df.loc[(order_df.index == df.index[i]), 'canceled'] = 1
+
+        saveFileName = os.path.join(self.RESOURCE_PATH, "etf_order", ymd + ".csv")
+        order_df.to_csv(saveFileName, index=False)
+
         return True