initial commit

한국투자증권(API)/stocks_info/domestic_stock_future_code.py

@@ -0,0 +1,39 @@
+'''선물옵션 종목코드(fo_stk_code_mts.mst) 정제 파이썬 파일'''
+
+import pandas as pd
+import urllib.request
+import ssl
+import zipfile
+import os
+
+base_dir = os.getcwd()
+
+def get_domestic_stk_future_master_dataframe(base_dir):
+    
+    # download file
+    print("Downloading...")
+
+    ssl._create_default_https_context = ssl._create_unverified_context
+    urllib.request.urlretrieve("https://new.real.download.dws.co.kr/common/master/fo_stk_code_mts.mst.zip", base_dir + "\\fo_stk_code_mts.mst.zip")
+    os.chdir(base_dir)
+
+    fo_stk_code_zip = zipfile.ZipFile('fo_stk_code_mts.mst.zip')
+    fo_stk_code_zip.extractall()
+    fo_stk_code_zip.close()
+    file_name = base_dir + "\\fo_stk_code_mts.mst"
+
+    fo_stk_code_zip = zipfile.ZipFile('fo_stk_code_mts.mst.zip')
+    fo_stk_code_zip.extractall()
+    fo_stk_code_zip.close()
+    file_name = base_dir + "\\fo_stk_code_mts.mst"
+
+    columns = ['상품종류','단축코드','표준코드',' 한글종목명',' ATM구분',
+               ' 행사가',' 월물구분코드',' 기초자산 단축코드',' 기초자산 명']
+    df=pd.read_table(file_name, sep='|',encoding='cp949',header=None)
+    df.columns = columns
+    df.to_excel('fo_stk_code_mts.xlsx',index=False)  # 현재 위치에 엑셀파일로 저장
+
+    return df
+    
+df = get_domestic_stk_future_master_dataframe(base_dir)
+print("Done")