import keyring

keyring.set_password("system", "username", "password")
keyring.get_password("system", "username")

import keyring
from typing import Optional
import pprint
import requests
from io import BytesIO
import zipfile
import xmltodict
import json
import pandas as pd

DART_API_KEY = keyring.get_password("open_dart_api", "gimmaru")

def get_corp_code(DART_API_KEY: str) -> pd.DataFrame:
    corp_code_url = f"https://opendart.fss.or.kr/api/corpCode.xml?crtfc_key={DART_API_KEY}"
    corp_code_byte = requests.get(corp_code_url)
    assert corp_code_byte.status_code == 200, f"{corp_code_byte.status_code} HTTP Error\n\n See the following sites: https://httpstatusdogs.com/"

    corp_code_zip = BytesIO(corp_code_byte.content)
    
    if zipfile.is_zipfile(corp_code_zip):
        corp_code_unzip = zipfile.ZipFile(corp_code_zip)
        corp_code_xml = corp_code_unzip.read('CORPCODE.xml').decode('utf-8')
        corp_code_dict = xmltodict.parse(corp_code_xml)
        return pd.DataFrame(corp_code_dict['result']['list'])
    else:
        status_code = xmltodict.parse(corp_code_byte.content)['result']['status']
        assert status_code == '000', f'{status_code} Error\n\nCheck Message Explanation: https://opendart.fss.or.kr/guide/detail.do?apiGrpCd=DS001&apiId=2019018'

df_code = get_corp_code(DART_API_KEY)
df_code.info()

df_code.sample(5)

df_code = (
    df_code
    .dropna()
    .reset_index(drop=True)
)

df_code

print( df_code.loc[df_code['corp_name'] == '삼성제약', 'corp_code'].values[0] )

def get_financial_statements(
    DART_API_KEY: str, 
    CORP_CODE: str, 
    BSNS_YEAR: str, 
    REPRT_CODE: str, 
    FS_DIV: str
) -> pd.DataFrame:

    fs_url = f"https://opendart.fss.or.kr/api/fnlttSinglAcntAll.json?crtfc_key={DART_API_KEY}&corp_code={CORP_CODE}&bsns_year={BSNS_YEAR}&reprt_code={REPRT_CODE}&fs_div={FS_DIV}"
    fs_raw = requests.get(fs_url)
    assert fs_raw.status_code == 200, f"{fs_raw.status_code} HTTP Error\n\n See the following sites: https://httpstatusdogs.com/"

    fs_dict = fs_raw.json()
    assert fs_dict['status'] == '000', f"{fs_dict['status']} Error\n\n {fs_dict['message']}"

    return pd.DataFrame(fs_dict['list'])

df_fs = get_financial_statements(
    DART_API_KEY=DART_API_KEY,
    CORP_CODE = df_code.loc[df_code['corp_name'] == '삼성제약', 'corp_code'].values[0],
    BSNS_YEAR = '2015',
    REPRT_CODE = '11011',
    FS_DIV = 'CFS',
)

df_fs

df_fs_2018 = get_financial_statements(
    DART_API_KEY=DART_API_KEY,
    CORP_CODE = df_code.loc[df_code['corp_name'] == '삼성제약', 'corp_code'].values[0],
    BSNS_YEAR = '2018',
    REPRT_CODE = '11011',
    FS_DIV = 'CFS',
)

df_fs_2018

df_fs_3q = get_financial_statements(
    DART_API_KEY=DART_API_KEY,
    CORP_CODE = df_code.loc[df_code['corp_name'] == '삼성제약', 'corp_code'].values[0],
    BSNS_YEAR = '2016',
    REPRT_CODE = '11014',
    FS_DIV = 'CFS',
)

df_fs_3q