주식자동매매 36강. 자동매매 종목 선정(5), 최종 정리 및 코드 공유

32강부터 35강까지 자동매매 종목을 선정하는 법을 알아보았습니다. 우리가 원하는 종목을 찾아 넣는 것이죠. 구성된 GUI에 종목명/코드/현재가/신용비율을 넣는 것이었습니다. 한번 더 복습해봅시다.

 

이해가 어려우시면 32강부터 다시 복습하세요

 

[유튜브 강의, 링크]와 같이 보시면 많은 도움이 되실 겁니다.

1. GUI 구성

 GUI 구성은 간단합니다. 종목명 입력에 우리가 원하는 종목을 넣고 종목 추가를 누르면 Table Widget에 종묙명/코드/현재가/신용비율이 입력되는 것이죠.

GUI 구성

 - Table Widget : 특정 값을 행과 열로 입력할 수 있습니다. 관련 함수를 살펴봅시다.

  * setColumnCount : 열의 개수를 구성합니다.

  * setRowCount : 행의 개수를 구성합니다.

  * setHorizontalHeaderLabels : 행의 이름을 정의합니다.

  * setItem과 QTableWidgetItem : 행과 열에 원하는 값을 입력하는 함수입니다.

 - Label : 한글/영어로 설명을 적을 수 있습니다.

 - Text Edit : 원하는 값을 적어 넣을 수 있습니다.

  * toPlainText : Edit에 적힌 값을 가져올 수 있습니다.

 - Push Button : 크릭 시 특정 이벤트가 실행됩니다.

  * clicked.connect() : 클릭되면 ()안의 함수를 실행합니다.

 

 위의 함수를 정의하면 우리는 Text Edit에 삼성전자를 넣고 종목 추가(Push Button)를 누르면 종목명/코드/현재가/신용비율을 Table Widget에 입력하면 되는 것입니다. 그따서 아래와 같이 코딩을 하였습니다. 

 

2. 주요 코딩 설명

 -  Push Button(additemlast, Deletcode)가 클릭되면 종목 추가 또는 종목 삭제 함수가 실행되게 합니다.

self.additmelast.clicked.connect(self.searchItem2)                # 종목 추가
self.Deletcode.clicked.connect(self.deltecode)                        # 종목 삭제

 -  종목명은 우리가 입력한 Text Edit에서 가져오고 코드 번호는 메타클래스기반 싱글턴의 All_Stock_Cdoe에서 가져옵니다.

def searchItem2(self):            # 종목추가시 사용됨.
    itemName = self.searchItemTextEdit2.toPlainText()
    if itemName != "":
        for code in self.k.All_Stock_Code.keys():  # 포트폴리오에 저장된 코드들을 실시간 등록
            # 주식체결 정보 가져오기(틱 데이터) : 현재가, 전일대비, 등락률, 매도호가, 매수호가, 거래량, 누적거래량, 고가, 시가, 저가
            if itemName == self.k.All_Stock_Code[code]['종목명']:
                self.new_code = code

 - 그리고 종목명과 코드는 아래 코드를 이용해 Table Widget에 입력합니다.

 - 특정 값 입력 시 위의 Table Widget함수를 사용합니다.

 - 열의 이름은 : 종목코드, 종목명, 현재가, 신용비율이빈다.

 - 따라서 열의 개수는 4개가 됩니다.

 - 행의 개수는 입력한 종목의 개수와 동일합니다.

column_head = ["종목코드", "종목명", "현재가", "신용비율"]
colCount = len(column_head)
row_count = self.buylast.rowCount()

self.buylast.setColumnCount(colCount)  # 행 갯수
self.buylast.setRowCount(row_count+1)  # colum_haed가 한 행을 잡아 먹는다. 실제 입력 되는 값은 1행 부터이다.
self.buylast.setHorizontalHeaderLabels(column_head)  # 행의 이름 삽입

self.buylast.setItem(row_count, 0, QTableWidgetItem(str(self.new_code))) # 실제 입력값은 1행부터이나 0부터 들어가야 된다.
self.buylast.setItem(row_count, 1, QTableWidgetItem(str(itemName)))

 - 원하는 종목의 현재가와 신용비율은 opt10001 : 주식기본정보 가져오기 함수를 사용합니다.

 - getItemInfo 함수에 주식기본정보요청하는 명령 코드를 작성합니다.

def getItemInfo(self, new_code):
    self.k.kiwoom.dynamicCall("SetInputValue(QString, QString)", "종목코드", new_code)
    self.k.kiwoom.dynamicCall("CommRqData(QString, QString, int, QString)", "주식기본정보요청", "opt10001", 0, "100")

 - 전송된 명령은 trdata_slot에 받아옵니다.

 - 이때 받아오는 값은 현재가와 신용비율이며, Table widget의 2열과 3열에 넣습니다.

def trdata_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):

    if sTrCode == "opt10001":
        if sRQName == "주식기본정보요청":
            currentPrice = abs(int(self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "현재가")))
            D_R = (self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "신용비율")).strip()
            row_count = self.buylast.rowCount()
            self.buylast.setItem(row_count - 1, 2, QTableWidgetItem(str(currentPrice)))
            self.buylast.setItem(row_count - 1, 3, QTableWidgetItem(str(D_R)))

 

 - 원치 않는 종목을 실수로 입력하면 삭제를 해야 합니다.

 - selectedIndexes()를 통해 현재 클릭한 행의 위치의 행과 열을 x에 넣습니다.

 - 그리고 x의 행[0]의 위치를 파악 후 removeRow 함수를 사용해 삭제합니다.

def deltecode(self):
    x = self.buylast.selectedIndexes()  # 리스트로 선택된 행번호와 열번호가 x에 입력된다.
    self.buylast.removeRow(x[0].row())

 위의 모든 과정을 거치면 아래와 같이 Table Widget에 우리가 원하는 갑을 넣을 수 있습니다.

GUI 구성

3. 소스 공유 : Main Script

반응형

import sys                        # system specific parameters and functions : 파이썬 스크립트 관리
from PyQt5.QtWidgets import *     # GUI의 그래픽적 요소를 제어       하단의 terminal 선택, activate py37_32,  pip install pyqt5,   전부다 y
from PyQt5 import uic             # ui 파일을 가져오기위한 함수
from PyQt5.QtCore import *        # eventloop/스레드를 사용 할 수 있는 함수 가져옴.

################# 부가 기능 수행(일꾼) #####################################
from kiwoom import Kiwoom          # 키움증권 함수/공용 방 (싱글턴)
from Qthread_1 import Thread1      # 계좌평가잔고내역 가져오기
from Qthread_2 import Thread2      # 계좌 관리

#=================== 프로그램 실행 프로그램 =========================#

form_class = uic.loadUiType("ALBA.ui")[0]             # 만들어 놓은 ui 불러오기

class Login_Machnine(QMainWindow, QWidget, form_class):       # QMainWindow : PyQt5에서 윈도우 생성시 필요한 함수

    def __init__(self, *args, **kwargs):                      # Main class의 self를 초기화 한다.

        print("Login Machine 실행합니다.")
        super(Login_Machnine, self).__init__(*args, **kwargs)
        form_class.__init__(self)                            # 상속 받은 from_class를 실행하기 위한 초기값(초기화)
        self.setUI()                                         # UI 초기값 셋업 반드시 필요

        ### 초기 셋팅
        self.label_11.setText(str("총매입금액"))
        self.label_12.setText(str("총평가금액"))
        self.label_13.setText(str("추정예탁자산"))
        self.label_14.setText(str("총평가손익금액"))
        self.label_15.setText(str("총수익률(%)"))


        #### 기타 함수
        self.login_event_loop = QEventLoop()  # 이때 QEventLoop()는 block 기능을 가지고 있다.

        ####키움증권 로그인 하기
        self.k = Kiwoom()                     # Kiwoom()을 실행하며 상속 받는다. Kiwoom()은 전지적인 아이다.
        self.set_signal_slot()                # 키움로그인을 위한 명령어 전송 시 받는 공간을 미리 생성한다.
        self.signal_login_commConnect()

        #####이벤트 생성 및 진행
        self.call_account.clicked.connect(self.c_acc)         # 계좌정보가져오기
        self.acc_manage.clicked.connect(self.a_manage)         # 계좌정보가져오기

        ################# 부가기능 1 : 종목선택하기 새로운 종목 추가 및 삭제
        self.k.kiwoom.OnReceiveTrData.connect(self.trdata_slot)           # 키움서버 데이터 받는 곳
        self.additmelast.clicked.connect(self.searchItem2)                # 종목 추가
        self.Deletcode.clicked.connect(self.deltecode)                        # 종목 삭제
        ####################

    def deltecode(self):
        x = self.buylast.selectedIndexes()  # 리스트로 선택된 행번호와 열번호가 x에 입력된다.
        self.buylast.removeRow(x[0].row())


    def searchItem2(self):            # 종목추가시 사용됨.
        itemName = self.searchItemTextEdit2.toPlainText()
        if itemName != "":
            for code in self.k.All_Stock_Code.keys():  # 포트폴리오에 저장된 코드들을 실시간 등록
                # 주식체결 정보 가져오기(틱 데이터) : 현재가, 전일대비, 등락률, 매도호가, 매수호가, 거래량, 누적거래량, 고가, 시가, 저가
                if itemName == self.k.All_Stock_Code[code]['종목명']:
                    self.new_code = code

        column_head = ["종목코드", "종목명", "현재가", "신용비율"]
        colCount = len(column_head)
        row_count = self.buylast.rowCount()

        self.buylast.setColumnCount(colCount)  # 행 갯수
        self.buylast.setRowCount(row_count+1)  # colum_haed가 한 행을 잡아 먹는다. 실제 입력 되는 값은 1행 부터이다.
        self.buylast.setHorizontalHeaderLabels(column_head)  # 행의 이름 삽입

        self.buylast.setItem(row_count, 0, QTableWidgetItem(str(self.new_code))) # 실제 입력값은 1행부터이나 0부터 들어가야 된다.
        self.buylast.setItem(row_count, 1, QTableWidgetItem(str(itemName)))

        self.getItemInfo(self.new_code)



    def getItemInfo(self, new_code):
        self.k.kiwoom.dynamicCall("SetInputValue(QString, QString)", "종목코드", new_code)
        self.k.kiwoom.dynamicCall("CommRqData(QString, QString, int, QString)", "주식기본정보요청", "opt10001", 0, "100")

    def setUI(self):
        self.setupUi(self)                # UI 초기값 셋업

    def set_signal_slot(self):
        self.k.kiwoom.OnEventConnect.connect(self.login_slot)  # 내가 알고 있는 login_slot에다가 특정 값을 던져 준다.

    def signal_login_commConnect(self):
        self.k.kiwoom.dynamicCall("CommConnect()")  # 네트워크적 서버 응용프로그램에 데이터를 전송할 수 있게 만든 함수
        self.login_event_loop.exec_()  # 로그인이 완료될 때까지 계속 반복됨. 꺼지지 않음.

    def login_slot(self, errCode):
        if errCode == 0:
            print("로그인 성공")
            self.statusbar.showMessage("로그인 성공")
            self.get_account_info()                    # 로그인시 계좌정보 가져오기

        elif errCode == 100:
            print("사용자 정보교환 실패")
        elif errCode == 101:
            print("서버접속 실패")
        elif errCode == 102:
            print("버전처리 실패")
        self.login_event_loop.exit()  # 로그인이 완료되면 로그인 창을 닫는다.

    def get_account_info(self):
        account_list = self.k.kiwoom.dynamicCall("GetLoginInfo(String)", "ACCNO")

        for n in account_list.split(';'):
            self.accComboBox.addItem(n)

    def c_acc(self):
        print("선택 계좌 정보 가져오기")
        ##### 1번 일꾼 실행
        h1 = Thread1(self)
        h1.start()

    def a_manage(self):
        print("계좌 관리")
        h2 = Thread2(self)
        h2.start()

    def trdata_slot(self, sScrNo, sRQName, sTrCode, sRecordName, sPrevNext):

        if sTrCode == "opt10001":
            if sRQName == "주식기본정보요청":
                currentPrice = abs(int(self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "현재가")))
                D_R = (self.k.kiwoom.dynamicCall("GetCommData(QString, QString, int, QString)", sTrCode, sRQName, 0, "신용비율")).strip()
                row_count = self.buylast.rowCount()
                self.buylast.setItem(row_count - 1, 2, QTableWidgetItem(str(currentPrice)))
                self.buylast.setItem(row_count - 1, 3, QTableWidgetItem(str(D_R)))



if __name__=='__main__':             # import된 것들을 실행시키지 않고 __main__에서 실행하는 것만 실행 시킨다.
                                     # 즉 import된 다른 함수의 코드를 이 화면에서 실행시키지 않겠다는 의미이다.

    app = QApplication(sys.argv)     # PyQt5로 실행할 파일명을 자동으로 설정, PyQt5에서 자동으로 프로그램 실행
    CH = Login_Machnine()            # Main 클래스 myApp으로 인스턴스화
    CH.show()                        # myApp에 있는 ui를 실행한다.
    app.exec_()                      # 이벤트 루프