코스피(KOSPI), 코스닥(KOSDAQ)이 서로 다른 시장으로 분류되기 때문에 종목 정보를 받아 오는 데 필요한 구분 값도 나누어져 있습니다. 구분 값은 코스피(0), 코스닥(10)이며, 이 값을 전달하여 코스피, 코스닥에 상장된 모든 종목의 코드를 얻어 오는 함수를 만들어 보겠습니다. 앞서 작성한 Kiwoom.py 파일의 get_account_number 함수에 이어서 다음 코드를 작성합니다.

Kiwoom.py

(...)
def get_code_list_by_market(self, market_type):
    code_list = self.dynamicCall("GetCodeListByMarket(QString)", market_type)
    code_list = code_list.split(';')[:-1]
    return code_list

매개변수로 전달되는 market_type은 어떤 시장에 해당하는 종목을 얻어 올지 의미하는 구분 값입니다. market_type 값에 0(코스피 구분 값)을 전달하여 함수를 실행하면 코스피 종목 코드들이 code_list에 바로 저장됩니다. 먼저 main.py 파일에서 이 함수를 호출하여 결과를 확인해 보겠습니다.

main.py

from api.Kiwoom import *
import sys

app = QApplication(sys.argv)
kiwoom = Kiwoom()

kospi_code_list = kiwoom.get_code_list_by_market("0")
print(kospi_code_list)

app.exec_()