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