initial commit

한국투자증권(API)/examples_llm/domestic_stock/chk_holiday/chk_chk_holiday.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 chk_holiday import chk_holiday
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+##############################################################################################
+# [국내주식] 업종/기타 > 국내휴장일조회[국내주식-040]
+##############################################################################################
+
+COLUMN_MAPPING = {
+    'bass_dt': '기준일자',
+    'wday_dvsn_cd': '요일구분코드',
+    'bzdy_yn': '영업일여부',
+    'tr_day_yn': '거래일여부',
+    'opnd_yn': '개장일여부',
+    'sttl_day_yn': '결제일여부'
+}
+
+NUMERIC_COLUMNS = []
+
+
+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()
+
+    # case1 조회
+    logging.info("=== case1 조회 ===")
+    try:
+        result = chk_holiday(bass_dt="20250630")
+    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/chk_holiday/chk_holiday.py

@@ -0,0 +1,109 @@
+"""
+Created on 20250601 
+"""
+
+import sys
+import time
+from typing import Optional
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+##############################################################################################
+# [국내주식] 업종/기타 > 국내휴장일조회[국내주식-040]
+##############################################################################################
+
+# 상수 정의
+API_URL = "/uapi/domestic-stock/v1/quotations/chk-holiday"
+
+
+def chk_holiday(
+        bass_dt: str,  # 기준일자 (YYYYMMDD)
+        NK100: str = "",  # 연속조회키
+        FK100: str = "",  # 연속조회검색조건
+        tr_cont: str = "",  # 연속거래여부
+        dataframe: Optional[pd.DataFrame] = None,  # 누적 데이터프레임
+        depth: int = 0,  # 내부 재귀깊이 (자동관리)
+        max_depth: int = 10  # 최대 재귀 횟수 제한
+) -> pd.DataFrame:
+    """
+    (★중요) 국내휴장일조회(TCA0903R) 서비스는 당사 원장서비스와 연관되어 있어 
+    단시간 내 다수 호출시 서비스에 영향을 줄 수 있어 가급적 1일 1회 호출 부탁드립니다.
+
+    국내휴장일조회 API입니다.
+    영업일, 거래일, 개장일, 결제일 여부를 조회할 수 있습니다.
+    주문을 넣을 수 있는지 확인하고자 하실 경우 개장일여부(opnd_yn)을 사용하시면 됩니다.
+    
+    Args:
+        bass_dt (str): [필수] 기준일자 (ex. YYYYMMDD)
+        NK100 (str): 연속조회키
+        FK100 (str): 연속조회검색조건
+        tr_cont (str): 연속거래여부
+        dataframe (Optional[pd.DataFrame]): 누적 데이터프레임
+        depth (int): 내부 재귀깊이 (자동관리)
+        max_depth (int): 최대 재귀 횟수 제한
+
+    Returns:
+        pd.DataFrame: 국내휴장일조회 데이터
+        
+    Example:
+        >>> df = chk_holiday(bass_dt="20250630")
+        >>> print(df)
+    """
+
+    if bass_dt == "":
+        raise ValueError("bass_dt is required (e.g. 'YYYYMMDD')")
+
+    if depth > max_depth:
+        logging.warning("Max recursive depth reached.")
+        if dataframe is None:
+            return pd.DataFrame()
+        else:
+            return dataframe
+
+    tr_id = "CTCA0903R"  # 국내휴장일조회
+
+    params = {
+        "BASS_DT": bass_dt,
+        "CTX_AREA_FK": FK100,
+        "CTX_AREA_NK": NK100
+    }
+
+    res = ka._url_fetch(API_URL, tr_id, tr_cont, params)
+
+    if res.isOK():
+        if hasattr(res.getBody(), 'output'):
+            output_data = res.getBody().output
+            if not isinstance(output_data, list):
+                output_data = [output_data]
+            current_data = pd.DataFrame(output_data)
+        else:
+            current_data = pd.DataFrame()
+
+        if dataframe is not None:
+            dataframe = pd.concat([dataframe, current_data], ignore_index=True)
+        else:
+            dataframe = current_data
+
+        tr_cont = res.getHeader().tr_cont
+        FK100 = res.getBody().ctx_area_fk
+        NK100 = res.getBody().ctx_area_nk
+
+        if tr_cont in ["M", "F"]:  # 다음 페이지 존재
+            logging.info("Call Next page...")
+            ka.smart_sleep()  # 시스템 안정적 운영을 위한 지연
+            return chk_holiday(
+                bass_dt, NK100, FK100, "N", dataframe, depth + 1, max_depth
+            )
+        else:
+            logging.info("Data fetch complete.")
+            return dataframe
+    else:
+        res.printError(url=API_URL)
+        return pd.DataFrame()