import sys
from PyQt5.QtWidgets import *
from PyQt5 import uic
from PyQt5.QAxContainer import *
form_class = uic.loadUiType("upperLimitPriceTrading.ui")[0] # ui 파일을 로드하여 form_class 생성
class MyWindow(QMainWindow, form_class): # MyWindow 클래스 QMainWindow, form_class 클래스를 상속 받아 생성됨
def __init__(self): # MyWindow 클래스의 초기화 함수(생성자)
super().__init__() # 부모클래스 QMainWindow 클래스의 초기화 함수(생성자)를 호출
self.setupUi(self) # ui 파일 화면 출력
self.kiwoom = QAxWidget("KHOPENAPI.KHOpenAPICtrl.1") # 키움증권 Open API+의 ProgID를 사용하여 생성된 QAxWidget을 kiwoom 변수에 할당
self.btnStart.setDisabled(True) # 거래시작 버튼을 비활성화 상태로 변경
self.btnStop.setDisabled(True) # 거래중지 버튼을 비활성화 상태로 변경
self.btnReStart.setDisabled(True) # 재시작 버튼을 비활성화 상태로 변경
self.lePrice.setDisabled(True) # 투입금 입력란을 비활성화 상태로 변경
self.cbAcctNo.setDisabled(True) # 계좌번호 선택란을 비활성화 상태로 변경
self.leD2Price.setDisabled(True) # D2예수금 입력란을 비활성화 상태로 변경
self.cbCdtNm.setDisabled(True) # 조건검색명 선택란을 비활성화 상태로 변경
self.leTotalPrice.setDisabled(True) # 총투입금 입력란을 비활성화 상태로 변경
self.leProfits.setDisabled(True) # 수익금 입력란을 비활성화 상태로 변경
self.leProfitsPc.setDisabled(True) # 수익률 입력란을 비활성화 상태로 변경
self.pteBuyLog.setDisabled(True) # 매도 종목 노출란을 비활성화 상태로 변경
self.pteSellLog.setDisabled(True) # 매수 종목 노출란을 비활성화 상태로 변경
self.pteLog.setDisabled(True) # 전체 내역 노출란을 비활성화 상태로 변경
self.btnLogin.clicked.connect(self.btn_login) # ui 파일을 생성할때 작성한 로그인 버튼의 objectName 으로 클릭 이벤트가 발생할 경우 btn_login 함수를 호출
self.kiwoom.OnEventConnect.connect(self.event_connect) # 키움 서버 접속 관련 이벤트가 발생할 경우 event_connect 함수 호출
self.kiwoom.OnReceiveConditionVer.connect(self.receive_condition_var) # 키움 사용자 조건검색식 수신 관련 이벤트가 발생할 경우 receive_condition_var 함수 호출
def btn_login(self): # Login 버튼 클릭 시 실행되는 함수
ret = self.kiwoom.dynamicCall("CommConnect()") # 키움 로그인 윈도우를 실행
def event_connect(self, err_code): # 키움 서버 접속 관련 이벤트가 발생할 경우 실행되는 함수
if err_code == 0: # err_code가 0이면 로그인 성공 그외 실패
self.pteLog.appendPlainText("로그인 성공") # ui 파일을 생성할때 작성한 plainTextEdit의 objectName 으로 해당 plainTextEdit에 텍스트를 추가함
account_num = self.kiwoom.dynamicCall("GetLoginInfo(QString)", ["ACCNO"]) # 키움 dynamicCall 함수를 통해 GetLoginInfo 함수를 호출하여 계좌번호를 가져옴
self.pteLog.appendPlainText("계좌번호: " + account_num.rstrip(';')) # 키움은 전체 계좌를 반환하며 각 계좌 번호 끝에 세미콜론(;)이 붙어 있음으로 제거하여 plainTextEdit에 텍스트를 추가함
self.cbAcctNo.addItem(account_num.rstrip(';')) # 계좌번호 선택 박스에 계좌번호 셋팅
self.btnStart.setDisabled(False) # 거래시작 버튼을 활성화 상태로 변경
self.lePrice.setDisabled(False) # 투입금 입력란을 활성화 상태로 변경
self.cbAcctNo.setDisabled(False) # 계좌번호 선택란을 활성화 상태로 변경
self.cbCdtNm.setDisabled(False) # 조건검색명 선택란을 활성화 상태로 변경
self.pteBuyLog.setDisabled(False) # 매도 종목 노출란을 활성화 상태로 변경
self.pteSellLog.setDisabled(False) # 매수 종목 노출란을 활성화 상태로 변경
self.pteLog.setDisabled(False) # 전체 내역 노출란을 활성화 상태로 변경
self.btnLogin.setDisabled(True) # 로그인 버튼을 비활성화 상태로 변경
self.leTotalPrice.setDisabled(True) # 총투입금 입력란을 비활성화 상태로 변경
self.leProfits.setDisabled(True) # 수익금 입력란을 비활성화 상태로 변경
self.leProfitsPc.setDisabled(True) # 수익률 입력란을 비활성화 상태로 변경
self.leTotalPrice.setText('0') # 총투입금을 0원으로 셋팅
self.leProfits.setText('0') # 수익금을 0원으로 셋팅
self.leProfitsPc.setText('0%') # 수익률을 0%으로 셋팅
self.kiwoom.dynamicCall('GetConditionLoad()') # 키움 서버에 사용자 조건식 목록을 요청
QMessageBox.about(self, "message", "계좌 비밀번호 입력 및 환경 설정을 진행해주세요.")
else:
self.pteLog.appendPlainText("로그인 실패") # ui 파일을 생성할때 작성한 plainTextEdit의 objectName 으로 해당 plainTextEdit에 텍스트를 추가함
def receive_condition_var(self, bRet, sMsg): # 사용자 조건검색식 수신 함수
conditionNameList = self.kiwoom.dynamicCall('GetConditionNameList()') # 수신된 사용자 조건검색식 리스트를 받아옴 (ex. 인덱스^조건명;)
conditionNameListArray = conditionNameList.rstrip(';').split(';') # 조건검색식 리스트에 마지막 ";" 기호를 삭제하고 ";" 기호 기준 분리
for i in range(0, len(conditionNameListArray)): # 조건검색식 개수만큼 반복
self.cbCdtNm.addItem(conditionNameListArray[i]) # 콤보 박스에 조건 셋팅
index = self.cbCdtNm.findText('191^ULPTrading') # 조건검색식 명으로 순번을 찾음
if index >= 0: # 해당 순번이 있다면 (해당 조건검색식 명이 있다면)
self.cbCdtNm.setCurrentIndex(index) # 해당 순번을 선택 (해당 조검검색식을 선택)
# py 파일 실행시 제일 먼저 동작
if __name__ == "__main__":
app = QApplication(sys.argv)
myWindow = MyWindow() # MyWindow 클래스를 생성하여 myWondow 변수에 할당
myWindow.show() # MyWindow 클래스를 노출
app.exec_() # 메인 이벤트 루프에 진입 후 프로그램이 종료될 때까지 무한 루프 상태 대기
