initial commit

한국투자증권(API)/examples_llm/domestic_stock/pension_inquire_deposit/chk_pension_inquire_deposit.py

@@ -0,0 +1,74 @@
+"""
+Created on 20250601 
+"""
+
+import sys
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+from pension_inquire_deposit import pension_inquire_deposit
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+##############################################################################################
+# [국내주식] 주문/계좌 > 퇴직연금 예수금조회[v1_국내주식-035]
+##############################################################################################
+
+COLUMN_MAPPING = {
+    'dnca_tota': '예수금총액',
+    'nxdy_excc_amt': '익일정산액',
+    'nxdy_sttl_amt': '익일결제금액',
+    'nx2_day_sttl_amt': '2익일결제금액'
+}
+
+NUMERIC_COLUMNS = ['예수금총액', '익일정산액', '익일결제금액', '2익일결제금액']
+
+
+def main():
+    """
+    퇴직연금 예수금조회 테스트 함수
+    
+    이 함수는 퇴직연금 예수금조회 API를 호출하여 결과를 출력합니다.
+    
+    Returns:
+        None
+    """
+
+    # pandas 출력 옵션 설정
+    pd.set_option('display.max_columns', None)  # 모든 컬럼 표시
+    pd.set_option('display.width', None)  # 출력 너비 제한 해제
+    pd.set_option('display.max_rows', None)  # 모든 행 표시
+
+    # 인증 토큰 발급
+    ka.auth()
+
+    trenv = ka.getTREnv()
+
+    # case1 조회
+    logging.info("=== case1 조회 ===")
+    try:
+        result = pension_inquire_deposit(cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, acca_dvsn_cd="00")
+    except ValueError as e:
+        logging.error("에러 발생: %s" % str(e))
+        return
+
+    logging.info("사용 가능한 컬럼: %s", result.columns.tolist())
+
+    # 컬럼명 한글 변환 및 데이터 출력
+    result = result.rename(columns=COLUMN_MAPPING)
+
+    # 숫자형 컬럼 소수점 둘째자리까지 표시
+    for col in NUMERIC_COLUMNS:
+        if col in result.columns:
+            result[col] = pd.to_numeric(result[col], errors='coerce').round(2)
+
+    logging.info("결과:")
+    print(result)
+
+
+if __name__ == "__main__":
+    main()

한국투자증권(API)/examples_llm/domestic_stock/pension_inquire_deposit/pension_inquire_deposit.py

@@ -0,0 +1,70 @@
+"""
+Created on 20250601 
+"""
+
+
+import sys
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+##############################################################################################
+# [국내주식] 주문/계좌 > 퇴직연금 예수금조회[v1_국내주식-035]
+##############################################################################################
+
+# 상수 정의
+API_URL = "/uapi/domestic-stock/v1/trading/pension/inquire-deposit"
+
+def pension_inquire_deposit(
+    cano: str,              # 종합계좌번호 (12345678)
+    acnt_prdt_cd: str,      # 계좌상품코드 (29)
+    acca_dvsn_cd: str       # 적립금구분코드 (00)
+) -> pd.DataFrame:
+    """
+    ​※ 55번 계좌(DC가입자계좌)의 경우 해당 API 이용이 불가합니다.
+    KIS Developers API의 경우 HTS ID에 반드시 연결되어있어야만 API 신청 및 앱정보 발급이 가능한 서비스로 개발되어서 실물계좌가 아닌 55번 계좌는 API 이용이 불가능한 점 양해 부탁드립니다.
+    
+    Args:
+        cano (str): [필수] 종합계좌번호 (ex. 12345678)
+        acnt_prdt_cd (str): [필수] 계좌상품코드 (ex. 29)
+        acca_dvsn_cd (str): [필수] 적립금구분코드 (ex. 00)
+
+    Returns:
+        pd.DataFrame: 퇴직연금 예수금 데이터
+        
+    Example:
+        >>> df = pension_inquire_deposit(cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, acca_dvsn_cd="00")
+        >>> print(df)
+    """
+
+    if cano == "":
+        raise ValueError("cano is required (e.g. '12345678')")
+    
+    if acnt_prdt_cd == "":
+        raise ValueError("acnt_prdt_cd is required (e.g. '29')")
+    
+    if acca_dvsn_cd == "":
+        raise ValueError("acca_dvsn_cd is required (e.g. '00')")
+
+    tr_id = "TTTC0506R"  # 퇴직연금 예수금조회
+
+    params = {
+        "CANO": cano,                    # 종합계좌번호
+        "ACNT_PRDT_CD": acnt_prdt_cd,    # 계좌상품코드
+        "ACCA_DVSN_CD": acca_dvsn_cd     # 적립금구분코드
+    }
+    
+    res = ka._url_fetch(API_URL, tr_id, "", params)
+    
+    if res.isOK():
+        current_data = pd.DataFrame([res.getBody().output])
+        return current_data
+    else:
+        res.printError(url=API_URL)
+        return pd.DataFrame()