import win32com.client
import csv
import time
from datetime import datetime
from hts.OrderItem import OrderItem

class HTS:

    objCpCybos = None
    objCpCodeMgr = None

    def __init__(self):
        #self.connect()
        return

    def connect(self):
        # 연결 여부 체크
        self.objCpCybos = win32com.client.Dispatch("CpUtil.CpCybos")
        bConnect = self.objCpCybos.IsConnect
        if (bConnect == 0):
            print("PLUS가 정상적으로 연결되지 않음. ")
            exit()
        return

    def all_stocks(self):
        # 종목코드 리스트 구하기
        self.objCpCodeMgr = win32com.client.Dispatch("CpUtil.CpCodeMgr")
        codeList = self.objCpCodeMgr.GetStockListByMarket(1)  # 거래소
        codeList2 = self.objCpCodeMgr.GetStockListByMarket(2)  # 코스닥

        print("거래소 종목코드", len(codeList))
        for i, code in enumerate(codeList):
            secondCode = self.objCpCodeMgr.GetStockSectionKind(code)
            name = self.objCpCodeMgr.CodeToName(code)
            stdPrice = self.objCpCodeMgr.GetStockStdPrice(code)
            print(i, code, secondCode, stdPrice, name)

        print("코스닥 종목코드", len(codeList2))
        for i, code in enumerate(codeList2):
            secondCode = self.objCpCodeMgr.GetStockSectionKind(code)
            name = self.objCpCodeMgr.CodeToName(code)
            stdPrice = self.objCpCodeMgr.GetStockStdPrice(code)
            print(i, code, secondCode, stdPrice, name)

        print("거래소 + 코스닥 종목코드 ", len(codeList) + len(codeList2))
        return

    # 차트 데이터 구하기
    def getChartData(self, stock_code):
        # 차트 객체 구하기
        objStockChart = win32com.client.Dispatch("CpSysDib.StockChart")

        objStockChart.SetInputValue(0, 'A'+stock_code)  # 종목 코드 - 삼성전자
        objStockChart.SetInputValue(1, ord('2'))  # 개수로 조회
        objStockChart.SetInputValue(4, 100)  # 최근 100일 치
        objStockChart.SetInputValue(5, [0, 2, 3, 4, 5, 8])  # 날짜,시가,고가,저가,종가,거래량
        objStockChart.SetInputValue(6, ord('D'))  # '차트 주가 - 일간 차트 요청
        objStockChart.SetInputValue(9, ord('1'))  # 수정주가 사용
        objStockChart.BlockRequest()

        len = objStockChart.GetHeaderValue(3)

        print("날짜", "시가", "고가", "저가", "종가", "거래량")
        print("==============================================-")

        for i in range(len):
            day = objStockChart.GetDataValue(0, i)
            open = objStockChart.GetDataValue(1, i)
            high = objStockChart.GetDataValue(2, i)
            low = objStockChart.GetDataValue(3, i)
            close = objStockChart.GetDataValue(4, i)
            vol = objStockChart.GetDataValue(5, i)
            print(day, open, high, low, close, vol)

        return

    # 주식 현재가 조회
    def currentStock(self, stock_code):
        # 현재가 객체 구하기
        self.objStockMst = win32com.client.Dispatch("DsCbo1.StockMst")
        self.objStockMst.SetInputValue(0, 'A'+stock_code)  # 종목 코드 - 삼성전자
        self.objStockMst.BlockRequest()

        # 현재가 통신 및 통신 에러 처리
        rqStatus = self.objStockMst.GetDibStatus()
        rqRet = self.objStockMst.GetDibMsg1()
        print("통신상태", rqStatus, rqRet)
        if rqStatus != 0:
            exit()

        # 현재가 정보 조회
        code = self.objStockMst.GetHeaderValue(0)  # 종목코드
        name = self.objStockMst.GetHeaderValue(1)  # 종목명
        time = self.objStockMst.GetHeaderValue(4)  # 시간
        cprice = self.objStockMst.GetHeaderValue(11)  # 종가
        diff = self.objStockMst.GetHeaderValue(12)  # 대비
        open = self.objStockMst.GetHeaderValue(13)  # 시가
        high = self.objStockMst.GetHeaderValue(14)  # 고가
        low = self.objStockMst.GetHeaderValue(15)  # 저가
        offer = self.objStockMst.GetHeaderValue(16)  # 매도호가
        bid = self.objStockMst.GetHeaderValue(17)  # 매수호가
        vol = self.objStockMst.GetHeaderValue(18)  # 거래량
        vol_value = self.objStockMst.GetHeaderValue(19)  # 거래대금

        # 예상 체결관련 정보
        exFlag = self.objStockMst.GetHeaderValue(58)  # 예상체결가 구분 플래그
        exPrice = self.objStockMst.GetHeaderValue(55)  # 예상체결가
        exDiff = self.objStockMst.GetHeaderValue(56)  # 예상체결가 전일대비
        exVol = self.objStockMst.GetHeaderValue(57)  # 예상체결수량

        print("코드", code)
        print("이름", name)
        print("시간", time)
        print("종가", cprice)
        print("대비", diff)
        print("시가", open)
        print("고가", high)
        print("저가", low)
        print("매도호가", offer)
        print("매수호가", bid)
        print("거래량", vol)
        print("거래대금", vol_value)

        if (exFlag == ord('0')):
            print("장 구분값: 동시호가와 장중 이외의 시간")
        elif (exFlag == ord('1')):
            print("장 구분값: 동시호가 시간")
        elif (exFlag == ord('2')):
            print("장 구분값: 장중 또는 장종료")

        print("예상체결가 대비 수량")
        print("예상체결가", exPrice)
        print("예상체결가 대비", exDiff)
        print("예상체결수량", exVol)

        return

    # 주식 현금 매수주문
    def requestOrder(self, type, stock_code, count, price):
        # type = 2: buy, type=1: sell
        # 주문 초기화
        objTrade = win32com.client.Dispatch("CpTrade.CpTdUtil")
        initCheck = objTrade.TradeInit(0)
        if (initCheck != 0):
            print("주문 초기화 실패")
            exit()

        # 주식 매수 주문
        acc = objTrade.AccountNumber[0]  # 계좌번호
        accFlag = objTrade.GoodsList(acc, 1)  # 주식상품 구분
        # acc = "782446178"
        # accFlag[0] = "01"
        objStockOrder = win32com.client.Dispatch("CpTrade.CpTd0311")
        objStockOrder.SetInputValue(0, type.value)  # 1: 매도, 2: 매수
        objStockOrder.SetInputValue(1, acc)  # 계좌번호
        objStockOrder.SetInputValue(2, accFlag[0])  # 상품구분 - 주식 상품 중 첫번째
        objStockOrder.SetInputValue(3, "A"+stock_code)  # 종목코드
        objStockOrder.SetInputValue(4, count)  # 매수수량 count주
        objStockOrder.SetInputValue(5, price)  # 주문단가  - price 원
        objStockOrder.SetInputValue(7, "0")  # 주문 조건 구분 코드, 0: 기본 1: IOC 2:FOK
        objStockOrder.SetInputValue(8, "01")  # 주문호가 구분코드 - 01: 보통

        # 매수 주문 요청
        nRet = objStockOrder.BlockRequest()
        if (nRet != 0):
            print("order error", nRet)
            return None

        rqStatus = objStockOrder.GetDibStatus()
        rqRet = objStockOrder.GetDibMsg1()
        print("통신상태", rqStatus, rqRet)
        if rqStatus != 0:
            return None

        orderNum = objStockOrder.GetHeaderValue(0)

        if (type == "1"):
            print ("(SELL", count, price, ")")
        else:
            print ("(BUY", count, price, ")")
        return orderNum

    # 계좌 잔고 확인
    def requstJango(self):
        jangoDic = {}

        objTrade = win32com.client.Dispatch("CpTrade.CpTdUtil")
        initCheck = objTrade.TradeInit(0)
        if (initCheck != 0):
            print("주문 초기화 실패")
            exit()

        # 주식 매수 주문
        acc = objTrade.AccountNumber[0]  # 계좌번호
        accFlag = objTrade.GoodsList(acc, 1)  # 주식상품 구분

        objRq = win32com.client.Dispatch("CpTrade.CpTd6033")

        objRq.SetInputValue(0, acc)  # 계좌번호
        objRq.SetInputValue(1, accFlag[0])  # 상품구분 - 주식 상품 중 첫번째
        objRq.SetInputValue(2, 50)  # 요청 건수(최대 50)
        dicflag1 = {ord(' '): '현금',
                         ord('Y'): '융자',
                         ord('D'): '대주',
                         ord('B'): '담보',
                         ord('M'): '매입담보',
                         ord('P'): '플러스론',
                         ord('I'): '자기융자',
                         }
        while True:
            objRq.BlockRequest()
            # 통신 및 통신 에러 처리
            rqStatus = objRq.GetDibStatus()
            rqRet = objRq.GetDibMsg1()
            #print("통신상태", rqStatus, rqRet)
            if rqStatus != 0:
                return False

            cnt = objRq.GetHeaderValue(7)
            if cnt > 3:
                return jangoDic

            for i in range(cnt):
                item = {}
                code = objRq.GetDataValue(12, i)  # 종목코드
                item['종목코드'] = code
                item['종목명'] = objRq.GetDataValue(0, i)  # 종목명
                item['대출일'] = objRq.GetDataValue(2, i)  # 대출일
                item['잔고수량'] = objRq.GetDataValue(7, i)  # 체결잔고수량
                item['매도가능'] = objRq.GetDataValue(15, i)
                item['장부가'] = objRq.GetDataValue(17, i)  # 체결장부단가
                # item['평가금액'] = self.objRq.GetDataValue(9, i)  # 평가금액(천원미만은 절사 됨)
                # item['평가손익'] = self.objRq.GetDataValue(11, i)  # 평가손익(천원미만은 절사 됨)
                # 매입금액 = 장부가 * 잔고수량
                item['매입금액'] = item['장부가'] * item['잔고수량']
                item['현재가'] = 0
                item['대비'] = 0
                item['거래량'] = 0

                # 잔고 추가
                # key = (code, item['현금신용'],item['대출일'] )
                key = code
                jangoDic[key] = item

                if len(jangoDic) >= 3:  # 최대 3 종목만,
                    break

            if len(jangoDic) >= 3:
                break

            if (objRq.Continue == False):
                break

        check = False
        for item in jangoDic:
            if item:
                check = True
                break
        if not check:
            return None
        return jangoDic

    #  예약 주문 내역 조회 및 미체결 리스트 구하기
    def requestOrderList(self):
        # type = 2: buy, type=1: sell
        orderList = []
        # 주문 초기화
        objTrade = win32com.client.Dispatch("CpTrade.CpTdUtil")
        initCheck = objTrade.TradeInit(0)
        if (initCheck != 0):
            print("주문 초기화 실패")
            exit()

        # 주식 매수 주문
        acc = objTrade.AccountNumber[0]  # 계좌번호
        accFlag = objTrade.GoodsList(acc, 1)  # 주식상품 구분

        objResult = win32com.client.Dispatch("CpTrade.CpTd5339")
        objResult.SetInputValue(0, acc)  # 계좌번호
        objResult.SetInputValue(1, accFlag[0])  # 상품구분 - 주식 상품 중 첫번째
        objResult.SetInputValue(4, "0") # 전체
        objResult.SetInputValue(5, "1") # 정렬 기준 - 역순
        objResult.SetInputValue(6, "0") # 전체
        objResult.SetInputValue(7, 20) # 요청개수 - 최대 20개

        while True:
            ret = objResult.BlockRequest()
            if objResult.GetDibStatus() != 0:
                print("통신상태", objResult.GetDibStatus(), objResult.GetDibMsg1())
                return False

            if (ret == 2 or ret == 3):
                print("통신 오류", ret)
                return False;

            # 통신 초과 요청 방지에 의한 요류 인 경우
            while (ret == 4):  # 연속 주문 오류 임. 이 경우는 남은 시간동안 반드시 대기해야 함.
                time.sleep(1)
                ret = objResult.BlockRequest()

            # 수신 개수
            cnt = objResult.GetHeaderValue(5)
            print("[Cp5339] 수신 개수 ", cnt)
            if cnt == 0:
                break

            for i in range(cnt):
                item = OrderItem()
                item.orderNum = objResult.GetDataValue(1, i)
                item.orderPrev = objResult.GetDataValue(2, i)
                item.code = objResult.GetDataValue(3, i)  # 종목코드
                item.name = objResult.GetDataValue(4, i)  # 종목명
                item.orderDesc = objResult.GetDataValue(5, i)  # 주문구분내용
                item.amount = objResult.GetDataValue(6, i)  # 주문수량
                item.price = objResult.GetDataValue(7, i)  # 주문단가
                item.ContAmount = objResult.GetDataValue(8, i)  # 체결수량
                item.credit = objResult.GetDataValue(9, i)  # 신용구분
                item.modAvali = objResult.GetDataValue(11, i)  # 정정취소 가능수량
                item.buysell = objResult.GetDataValue(13, i)  # 매매구분코드
                item.creditdate = objResult.GetDataValue(17, i)  # 대출일
                item.orderFlagDesc = objResult.GetDataValue(19, i)  # 주문호가구분코드내용
                item.orderFlag = objResult.GetDataValue(21, i)  # 주문호가구분코드

                orderList.append(item)

                # 연속 처리 체크 - 다음 데이터가 없으면 중지
            if objResult.Continue == False:
                print("[Cp5339] 연속 조회 여부: 다음 데이터가 없음")
                break

        return orderList

    #  미체결 취소하기
    def cancelOrderList(self, orderList):

        if len(orderList) < 1:
            return

        objTrade = win32com.client.Dispatch("CpTrade.CpTdUtil")
        initCheck = objTrade.TradeInit(0)
        if (initCheck != 0):
            print("주문 초기화 실패")
            exit()
        acc = objTrade.AccountNumber[0]  # 계좌번호
        accFlag = objTrade.GoodsList(acc, 1)  # 주식상품 구분

        objCancelOrder = win32com.client.Dispatch("CpTrade.CpTd0314")  # 취소

        onums = []
        codes = []
        amounts = []
        for item in orderList:
            onums.append(item.orderNum)
            codes.append(item.code)
            amounts.append(item.amount)

        for i in range(len(onums)):
            ordernum = onums[i]
            code = codes[i]
            amount = amounts[i]
            objCancelOrder.SetInputValue(1, ordernum)  # 원주문 번호 - 정정을 하려는 주문 번호
            objCancelOrder.SetInputValue(2, acc)  # 상품구분 - 주식 상품 중 첫번째
            objCancelOrder.SetInputValue(3, accFlag[0])  # 상품구분 - 주식 상품 중 첫번째
            objCancelOrder.SetInputValue(4, code)  # 종목코드
            objCancelOrder.SetInputValue(5, amount)  # 정정 수량, 0 이면 잔량 취소임

            # 취소주문 요청
            ret = objCancelOrder.BlockRequest()
            print("[CpRPOrder/BlockRequestCancel] 주문결과", objCancelOrder.GetDibStatus(), objCancelOrder.GetDibMsg1())
            if objCancelOrder.GetDibStatus() != 0:
                break

        return

    # 주식 현재가 조회
    def writeStockData(self, stock_code, today):
        objCpCybos = win32com.client.Dispatch("CpUtil.CpCybos")
        bConnect = objCpCybos.IsConnect
        if (bConnect == 0):
            print("PLUS가 정상적으로 연결되지 않음. ")
            exit()

        # 차트 객체 구하기
        objStockChart = win32com.client.Dispatch("CpSysDib.StockChart")

        outfp = open("./data/"+stock_code+"_"+today+".csv", mode="w", encoding="utf-8")
        objStockChart.SetInputValue(0, 'A' + stock_code)  # 종목 코드
        objStockChart.SetInputValue(1, ord('1'))  # 1: 기간으로 조회, 2: 개수로 조회
        objStockChart.SetInputValue(2, today)  # 기간 조회 시, 시작일
        objStockChart.SetInputValue(3, today)             # 기간 조회 시, 종료일
        objStockChart.SetInputValue(4, 400)  # 조회 시 가져오는 Line 개수
        objStockChart.SetInputValue(5, [0, 1, 2, 3, 4, 5, 8])  # 날짜,시간,시가,고가,저가,종가,거래량
        objStockChart.SetInputValue(6, ord('m'))  # '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
        objStockChart.SetInputValue(7, 1)
        objStockChart.SetInputValue(9, ord('1'))  # 수정주가 사용
        objStockChart.BlockRequest()

        size = objStockChart.GetHeaderValue(3)

        outfp.write("%s,%s,%s,%s,%s,%s,%s\n" % ("날짜", "시간", "시가", "고가", "저가", "종가", "거래량"))
        for i in range(size - 1, -1, -1):
            day = objStockChart.GetDataValue(0, i)
            time = objStockChart.GetDataValue(1, i)
            start = objStockChart.GetDataValue(2, i)
            high = objStockChart.GetDataValue(3, i)
            low = objStockChart.GetDataValue(4, i)
            close = objStockChart.GetDataValue(5, i)
            vol = objStockChart.GetDataValue(6, i)
            outfp.write("%d,%s,%d,%d,%d,%d,%d\n" % (day, str(time).zfill(4), start, high, low, close, vol))
        outfp.close()

        return


    def write(self, day, result):
        #날짜,시간,시가,고가,저가,종가,거래량
        #20210909,900,2070,2070,2070,2070,0

        outFp = open(day+".csv", mode="w", encoding="UTF-8")
        outFp.write("날짜,시간,시가,고가,저가,종가,거래량\n")
        for i in range(len(result["time"])):
            outFp.write("%s,%s,%s,%s,%s,%s,%s\n"%(
                result["time"][i].strftime('%Y%m%d'),
                result["time"][i].strftime('%H%M'),
                result["open"][i],
                result["high"][i],
                result["low"][i],
                result["close"][i],
                result["vol"][i]))
        outFp.close()
        return

    def getCSV(self, fileName, given_day, result):
        with open(fileName, 'r') as infp:
            reader = csv.reader(infp)
            next(reader)

            for rows in reader:
                days = rows[0]  # data.날짜
                time = rows[1]  # data.시간
                open_v = rows[2]  # data.시가
                high = rows[3]  # data.고가
                low = rows[4]  # data.저가
                close = rows[5]  # data.종가
                vol = rows[6]  # data.거래량

                start_time = datetime.strptime(given_day + " 090000", '%Y%m%d %H%M%S')

                temp = datetime.strptime(str(days) + " " + str(time).zfill(4) + "00", '%Y%m%d %H%M%S')
                if temp < start_time:
                    continue

                result["time"].append(temp)
                result["open"].append(int(open_v))
                result["close"].append(int(close))
                result["high"].append(int(high))
                result["low"].append(int(low))
                result["vol"].append(int(vol))
        return

    # 주식 현재가 조회
    def getRealTime(self, stock_code, lastday, today):
        result = {"check": set(),
                  "time": [],
                  "open": [],
                  "close": [],
                  "high": [],
                  "low": [],
                  "vol": []}

        self.getCSV("./data/" + stock_code + "_" + lastday + ".csv", today, result)

        int_given_day = int(today)
        objCpCybos = win32com.client.Dispatch("CpUtil.CpCybos")
        bConnect = objCpCybos.IsConnect
        if (bConnect == 0):
            print("PLUS가 정상적으로 연결되지 않음. ")
            exit()

        # 차트 객체 구하기
        objStockChart = win32com.client.Dispatch("CpSysDib.StockChart")

        objStockChart.SetInputValue(0, 'A'+stock_code)          # 종목 코드
        objStockChart.SetInputValue(1, ord('1'))                # 1: 기간으로 조회, 2: 개수로 조회
        objStockChart.SetInputValue(2, today)                   # 기간 조회 시, 시작일
        objStockChart.SetInputValue(3, today)                   # 기간 조회 시, 종료일
        objStockChart.SetInputValue(4, 400)                     # 조회 시 가져오는 Line 개수
        objStockChart.SetInputValue(5, [0, 1, 2, 3, 4, 5, 8])   # 날짜,시간,시가,고가,저가,종가,거래량
        objStockChart.SetInputValue(6, ord('m'))                # '차트 주가 - 월(M), 주(W), 일(D), 시(H), 분(m), 초(S) 차트 요청
        objStockChart.SetInputValue(7, 1)
        objStockChart.SetInputValue(9, ord('1'))                # 수정주가 사용
        objStockChart.BlockRequest()

        size = objStockChart.GetHeaderValue(3)

        #print("날짜", "시간", "시가", "고가", "저가", "종가", "거래량")
        start_time = datetime.strptime(today + " 090000", '%Y%m%d %H%M%S')
        for i in range(size-1, -1, -1):
            int_day = objStockChart.GetDataValue(0, i)
            int_time = objStockChart.GetDataValue(1, i)

            if int_day < int_given_day:
                continue
            time = datetime.strptime(str(int_day)+" "+str(int_time).zfill(4)+"00", '%Y%m%d %H%M%S')
            if time < start_time:
                continue

            open = objStockChart.GetDataValue(2, i)
            close = objStockChart.GetDataValue(5, i)
            high = objStockChart.GetDataValue(3, i)
            low = objStockChart.GetDataValue(4, i)
            vol = objStockChart.GetDataValue(6, i)

            if len(result["check"]) == 0:
                result["check"].add(start_time)
                result["time"].append(start_time)
                result["open"].append(open)
                result["close"].append(open)
                result["high"].append(open)
                result["low"].append(open)
                result["vol"].append(0)

            if time not in result["check"]:
                result["check"].add(time)
                result["time"].append(time)

                result["open"].append(open)
                result["close"].append(close)
                result["high"].append(high)
                result["low"].append(low)
                result["vol"].append(vol)

        return