반응형
import win32com.client
import pandas as pd

# 연결 여부 체크
objCpCybos = win32com.client.Dispatch("CpUtil.CpCybos")
bConnect = objCpCybos.IsConnect
if (bConnect == 0):
    print("PLUS가 정상적으로 연결되지 않음.  ")
    exit()


# 전체 종목정보 가져오기

instCpCodeMgr = win32com.client.Dispatch("CpUtil.CpCodeMgr")
codeList = instCpCodeMgr.GetStockListByMarket(1)
rows = []
CPE_MARKET_KIND = {'KOSPI':1, 'KOSDAQ':2}

for key, value in CPE_MARKET_KIND.items():
    codeList = instCpCodeMgr.GetStockListByMarket(value)
    for code in codeList:
        name = instCpCodeMgr.CodeToName(code)
        sectionCode = instCpCodeMgr.GetStockSectionKind(code)
        rows.append([code, sectionCode, name, key])

stockitems = pd.DataFrame(data= rows, columns=['code','sectionKind','name','section'])
stockitems.loc[stockitems['sectionKind'] == 10, 'section'] = 'ETF'
stockitems = stockitems.drop('sectionKind',axis=1)
print(stockitems)
stockitems.to_csv('stockitems.csv', index=False)

출력결과

 

반응형

'공부 > 시스템트레이딩' 카테고리의 다른 글

3. 데이터 불러오기  (0) 2021.10.04
1. 개발환경설정  (0) 2021.10.04

+ Recent posts