initial commit

한국투자증권(API)/examples_llm/overseas_stock/price/chk_price.py

@@ -0,0 +1,112 @@
+"""
+Created on 2025-06-26
+
+"""
+
+import sys
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])  # kis_auth 파일 경로 추가
+import kis_auth as ka
+from price import price
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO, format='%(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+##############################################################################################
+# [해외주식] 기본시세 > 해외주식 현재체결가[v1_해외주식-009]
+##############################################################################################
+
+COLUMN_MAPPING = {
+    'rsym': '실시간조회종목코드',
+    'zdiv': '소수점자리수',
+    'base': '전일종가',
+    'pvol': '전일거래량',
+    'last': '현재가',
+    'sign': '대비기호',
+    'diff': '대비',
+    'rate': '등락율',
+    'tvol': '거래량',
+    'tamt': '거래대금',
+    'ordy': '매수가능여부'
+}
+
+NUMERIC_COLUMNS = ['소수점자리수', '전일종가', '전일거래량', '현재가', '대비', '등락율', '거래량', '거래대금']
+
+def main():
+    """
+    [해외주식] 기본시세
+    해외주식 현재체결가[v1_해외주식-009]
+
+    해외주식 현재체결가 테스트 함수
+    
+    Parameters:
+        - auth (str): 사용자권한정보 ("" (Null 값 설정))
+        - excd (str): 거래소코드 (HKS : 홍콩 NYS : 뉴욕 NAS : 나스닥 AMS : 아멕스 TSE : 도쿄 SHS : 상해 SZS : 심천 SHI : 상해지수 SZI : 심천지수 HSX : 호치민 HNX : 하노이 BAY : 뉴욕(주간) BAQ : 나스닥(주간) BAA : 아멕스(주간))
+        - symb (str): 종목코드 ()
+        - env_dv (str): 실전모의구분 (real:실전, demo:모의)
+
+    Returns:
+        - DataFrame: 해외주식 현재체결가 결과
+    
+    Example:
+        >>> df = price(auth="", excd="NAS", symb="AAPL", env_dv="real")  # 실전투자
+        >>> df = price(auth="", excd="NAS", symb="AAPL", env_dv="demo")  # 모의투자
+    """
+    try:
+        # pandas 출력 옵션 설정
+        pd.set_option('display.max_columns', None)  # 모든 컬럼 표시
+        pd.set_option('display.width', None)  # 출력 너비 제한 해제
+        pd.set_option('display.max_rows', None)  # 모든 행 표시
+
+        # 실전/모의투자 선택 (모의투자 지원 로직)
+        env_dv = "real"  # "real": 실전투자, "demo": 모의투자
+        logger.info("투자 환경: %s", "실전투자" if env_dv == "real" else "모의투자")
+
+        # 토큰 발급
+        logger.info("토큰 발급 중...")
+        if env_dv == "real":
+            ka.auth(svr='prod')  # 실전투자용 토큰
+        elif env_dv == "demo":
+            ka.auth(svr='vps')   # 모의투자용 토큰
+        logger.info("토큰 발급 완료")
+
+        # API 호출
+        logger.info("API 호출 시작: 해외주식 현재체결가 (%s)", "실전투자" if env_dv == "real" else "모의투자")
+        result = price(
+            auth="",
+            excd="NAS",
+            symb="AAPL",
+            env_dv="real"
+        )
+        
+        if result is None or result.empty:
+            logger.warning("조회된 데이터가 없습니다.")
+            return
+        
+        # 컬럼명 출력
+        logger.info("사용 가능한 컬럼 목록:")
+        logger.info(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)
+        
+        # 결과 출력
+        logger.info("=== 해외주식 현재체결가 결과 (%s) ===", "실전투자" if env_dv == "real" else "모의투자")
+        logger.info("조회된 데이터 건수: %d", len(result))
+        print(result)
+        
+    except Exception as e:
+        logger.error("에러 발생: %s", str(e))
+        raise
+
+if __name__ == "__main__":
+    main()

한국투자증권(API)/examples_llm/overseas_stock/price/price.py

@@ -0,0 +1,127 @@
+# [해외주식] 기본시세 - 해외주식 현재체결가
+# Generated by KIS API Generator (Single API Mode)
+# -*- coding: utf-8 -*-
+"""
+Created on 2025-06-26
+
+"""
+
+import logging
+import time
+from typing import Optional
+import sys
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO, format='%(levelname)s - %(message)s')
+logger = logging.getLogger(__name__)
+
+##############################################################################################
+# [해외주식] 기본시세 > 해외주식 현재체결가[v1_해외주식-009]
+##############################################################################################
+
+# 상수 정의
+API_URL = "/uapi/overseas-price/v1/quotations/price"
+
+def price(
+    auth: str,  # 사용자권한정보
+    excd: str,  # 거래소코드
+    symb: str,  # 종목코드
+    env_dv: str = "real",  # 실전모의구분
+    tr_cont: str = "",
+    dataframe: Optional[pd.DataFrame] = None,
+    depth: int = 0,
+    max_depth: int = 10
+) -> Optional[pd.DataFrame]:
+    """
+    [해외주식] 기본시세 
+    해외주식 현재체결가[v1_해외주식-009]
+    해외주식 현재체결가 API를 호출하여 DataFrame으로 반환합니다.
+    
+    Args:
+        auth (str): 사용자권한정보
+        excd (str): 거래소코드 (예: "NAS")
+        symb (str): 종목코드 (예: "AAPL")
+        env_dv (str): 실전모의구분 (real:실전, demo:모의)
+        tr_cont (str): 연속 거래 여부
+        dataframe (Optional[pd.DataFrame]): 누적 데이터프레임
+        depth (int): 현재 재귀 깊이
+        max_depth (int): 최대 재귀 깊이 (기본값: 10)
+        
+    Returns:
+        Optional[pd.DataFrame]: 해외주식 현재체결가 데이터
+        
+    Example:
+        >>> df = price("", "NAS", "AAPL")
+        >>> print(df)
+    """
+    # 로깅 설정
+    logger = logging.getLogger(__name__)
+
+    # 필수 파라미터 검증
+    if not excd:
+        logger.error("excd is required. (e.g. 'NAS')")
+        raise ValueError("excd is required. (e.g. 'NAS')")
+    
+    if not symb:
+        logger.error("symb is required. (e.g. 'AAPL')")
+        raise ValueError("symb is required. (e.g. 'AAPL')")
+
+    # 최대 재귀 깊이 체크
+    if depth >= max_depth:
+        logger.warning("Maximum recursion depth (%d) reached. Stopping further requests.", max_depth)
+        return dataframe if dataframe is not None else pd.DataFrame()
+    
+    # TR ID 설정 (모의투자 지원 로직)
+    if env_dv == "real" or env_dv == "demo":
+        tr_id = "HHDFS00000300"  # 실전투자, 모의투자 공통 TR ID
+    else:
+        logger.error("Invalid env_dv value: %s. Must be 'real' or 'demo'.", env_dv)
+        raise ValueError("env_dv must be 'real' or 'demo'")
+
+    params = {
+        "AUTH": auth,
+        "EXCD": excd,
+        "SYMB": symb,
+    }
+
+    # API 호출
+    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
+        
+        if tr_cont == "M":
+            logger.info("Calling next page...")
+            ka.smart_sleep()
+            return price(
+                auth,
+                excd,
+                symb,
+                env_dv,
+                "N", dataframe, depth + 1, max_depth
+            )
+        else:
+            logger.info("Data fetch complete.")
+            return dataframe
+    else:
+        logger.error("API call failed: %s - %s", res.getErrorCode(), res.getErrorMessage())
+        res.printError(API_URL)
+        return pd.DataFrame()