이렇게 get_code_list_by_market에 전달하는 값을 “0” 또는 “10”으로 나누어서 호출하면 각각 코스피, 코스닥에 속한 전체 종목 코드를 kospi_code_list와 kosdaq_code_list 변수에 저장합니다. 그러면 얻어 온 종목 코드를 바탕으로 종목명을 확인해 보겠습니다. 종목명을 확인하는 get_master_code_name 함수는 하나의 종목 코드를 받아 종목명을 반환하기 때문에 전체 종목명을 확인하려면 종목 코드의 개수만큼 호출해야 합니다. 저장된 종목 코드는 리스트이므로 다음과 같이 for 문을 사용하면 쉽게 종목명을 얻을 수 있습니다.
for code in kospi_code_list:
code_name = kiwoom.get_master_code_name(code)
print(code, code_name)
이 코드를 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)
for code in kospi_code_list:
code_name = kiwoom.get_master_code_name(code)
print(code, code_name)
kosdaq_code_list = kiwoom.get_code_list_by_market("10")
print(kosdaq_code_list)
for code in kosdaq_code_list:
code_name = kiwoom.get_master_code_name(code)
print(code, code_name)
app.exec_()