61 lines
2.1 KiB
Python
61 lines
2.1 KiB
Python
import os
|
|
import pandas as pd
|
|
from datetime import datetime, timedelta
|
|
|
|
from hts.OrderType import OrderType
|
|
|
|
class OrderChecker:
|
|
|
|
RESOURCE_PATH = None
|
|
order_df = None
|
|
|
|
def __init__(self, RESOURCE_PATH):
|
|
self.RESOURCE_PATH = RESOURCE_PATH
|
|
self.order_df = None
|
|
return
|
|
|
|
def read(self, ymd):
|
|
saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv")
|
|
|
|
if os.path.isfile(saveFileName):
|
|
self.order_df = pd.read_csv(saveFileName)
|
|
else:
|
|
self.order_df = pd.DataFrame(columns=["datetime", "stock_code", "type", "orderNum", "count", "price"])
|
|
|
|
if 'Unnamed: 0' in self.order_df.columns:
|
|
self.order_df.drop(['Unnamed: 0'], axis=1, inplace=True)
|
|
self.order_df = self.order_df.fillna(0)
|
|
self.order_df = self.order_df.astype({"stock_code": str, "type": str, "orderNum": int, "count": int, "price": int})
|
|
self.order_df['datetime'] = pd.to_datetime(self.order_df['datetime'])
|
|
return
|
|
|
|
def exist(self, stock_code):
|
|
if self.order_df.loc[self.order_df["stock_code"] == stock_code] is None:
|
|
return False
|
|
return True
|
|
|
|
def add(self, ymd, stock_code, type, orderNum, count, price):
|
|
if self.exist(stock_code):
|
|
return False
|
|
|
|
# 새로운 주문을 추가한다.
|
|
self.order_df = self.order_df.append({"stock_code": stock_code, "type": type, "orderNum": orderNum, "count": count, "price": price, "datetime": datetime.now()}, ignore_index=True)
|
|
|
|
# 파일로 기록한다.
|
|
saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv")
|
|
self.order_df.to_csv(saveFileName)
|
|
|
|
return True
|
|
|
|
def delete(self, ymd, stock_code):
|
|
if not self.exist(stock_code):
|
|
return False
|
|
|
|
# 매도 주문을 삭제한다.
|
|
self.order_df = self.order_df.loc[(self.order_df["stock_code"] != stock_code)]
|
|
|
|
# 파일로 기록한다.
|
|
saveFileName = os.path.join(self.RESOURCE_PATH, "order", ymd + ".csv")
|
|
self.order_df.to_csv(saveFileName)
|
|
|
|
return True |