# API com元件初始化

import comtypes.client

comtypes.client.GetModule(r'SKCOM.dll')

import comtypes.gen.SKCOMLib as sk

import time, pythoncom



# 群益API元件導入Python code內用的物件宣告

m_pSKCenter = comtypes.client.CreateObject(sk.SKCenterLib,interface=sk.ISKCenterLib)

m_pSKReply = comtypes.client.CreateObject(sk.SKReplyLib,interface=sk.ISKReplyLib)

m_pSKOrder = comtypes.client.CreateObject(sk.SKOrderLib,interface=sk.ISKOrderLib)



# 全域變數



# 登入帳號:交易帳號

dictUserID = {}

dictUserID["更新帳號"] = ["無"]



userID = "身分證字號"

password = "密碼"



# ReplyLib事件

class SKReplyLibEvent():

    def OnReplyMessage(self, bstrUserID, bstrMessages):

        nConfirmCode = -1

        msg = "【註冊公告OnReplyMessage】" + bstrUserID + "_" + bstrMessages;

        #richTextBoxMessage.insert('end', msg + "\n")

        #richTextBoxMessage.see('end')

        print(msg)

        return nConfirmCode

SKReplyEvent = SKReplyLibEvent();

SKReplyLibEventHandler = comtypes.client.GetEvents(m_pSKReply, SKReplyEvent);



# OrderLib事件

class SKOrderLibEvent():

    # 帳號資訊

    def OnAccount(self, bstrLogInID, bstrAccountData):

        msg = "【OnAccount】" + bstrLogInID + "_" + bstrAccountData;        

        #richTextBoxMessage.insert('end', msg + "\n")

        #richTextBoxMessage.see('end')

        print(msg)



        values = bstrAccountData.split(',')

        # broker ID (IB)4碼 + 帳號7碼

        Account = values[1] + values[3]



        if bstrLogInID in dictUserID:

            accountExists = False

            for value in dictUserID[bstrLogInID]:

                if value == Account:

                    accountExists = True

                    break

            if accountExists == False:

                dictUserID[bstrLogInID].append(Account)

        else:

            dictUserID[bstrLogInID] = [Account]

            

    # (新)期貨未平倉GW。透過呼叫 GetOpenInterestGW 後，資訊由該事件回傳

    def OnOpenInterest(self, bstrData):

        msg = "【OnOpenInterest】" + bstrData

        #richTextBoxMessage.insert('end', msg + "\n")

        #richTextBoxMessage.see('end')

        print(msg)

        

    # 國內期貨權益數。透過呼叫 GetFutureRights 後，資訊由該事件回傳

    def OnFutureRights(self, bstrData):

        msg = "【OnFutureRights】" + bstrData;

        #richTextBoxMessage.insert('end', msg + "\n")

        #richTextBoxMessage.see('end')

        print(msg)

          

SKOrderEvent = SKOrderLibEvent()

SKOrderLibEventHandler = comtypes.client.GetEvents(m_pSKOrder, SKOrderEvent)



def capital_Login(bstrUserID, bstrPassword):

    nCode = m_pSKCenter.SKCenterLib_Login(bstrUserID, bstrPassword)

    msg = "【SKCenterLib_Login】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

    print(msg)

    

    # 這裡有做下單初始化

    nCode = m_pSKOrder.SKOrderLib_Initialize()

    msg = "【SKOrderLib_Initialize】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

    print(msg)

    

    # 拿到交易帳號

    nCode = m_pSKOrder.GetUserAccount()

    msg = "【GetUserAccount】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

    print(msg)



def capital_GetOpenInterestGW(bstrLogInID, bstrAccount):

    nCode = m_pSKOrder.GetOpenInterestGW(bstrLogInID, bstrAccount, 1)

    msg = "【GetOpenInterestGW】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

    print(msg)



def capital_GetFutureRights(bstrLogInID, bstrAccount, sCoinType):    

    nCode = m_pSKOrder.GetFutureRights(bstrLogInID, bstrAccount, sCoinType)

    msg = "【GetFutureRights】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

    print(msg)

        

if __name__ == '__main__':

    capital_Login(userID, password)

    account = dictUserID[userID][0]

    capital_GetOpenInterestGW(userID, account)

    # sCoinType 0:全幣別 1:基幣(台幣TWD) 2:人民幣RMB

    capital_GetFutureRights(userID, account, 1)

    #每秒推一次 event, 推5秒

    for i in range(5):

        pythoncom.PumpWaitingMessages()

        time.sleep(1)

        def on_comboBoxSKCenterLib_SetAuthority(event):

            if comboBoxSKCenterLib_SetAuthority.get() == "正式環境":

                nAuthorityFlag = 0

            elif comboBoxSKCenterLib_SetAuthority.get() == "正式環境SGX":

                nAuthorityFlag = 1

            elif comboBoxSKCenterLib_SetAuthority.get() == "測試環境":

                nAuthorityFlag = 2

            elif comboBoxSKCenterLib_SetAuthority.get() == "測試環境SGX":

                nAuthorityFlag = 3

            nCode = m_pSKCenter.SKCenterLib_SetAuthority(nAuthorityFlag)

            msg = "【SKCenterLib_SetAuthority】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

            richTextBoxMethodMessage.insert('end',  msg + "\n")

            richTextBoxMethodMessage.see('end')

            nCode = m_pSKOrder.ReadCertByID(comboBoxUserID.get())

            msg = "【ReadCertByID】" + m_pSKCenter.SKCenterLib_GetReturnCodeMessage(nCode)

            richTextBoxMethodMessage.insert('end',  msg + "\n")

            richTextBoxMethodMessage.see('end')

        hour = str(nTimehms)[:2]

        min = str(nTimehms)[2:4]

        sec = str(nTimehms)[4:6]

        if len(str(nTimehms)) == 6:

            hour = str(nTimehms)[:2]

            min = str(nTimehms)[2:4]

            sec = str(nTimehms)[4:6]

        elif len(str(nTimehms)) == 5:

            hour = str(nTimehms)[:1]

            min = str(nTimehms)[1:3]

            sec = str(nTimehms)[3:5]

        else:

            hour = "0"

            min = str(nTimehms)[:1]

            sec = str(nTimehms)[1:3]

pip install comtypes pywin32