Files
KisStock/한국투자증권(API)/examples_llm/auth/auth_token/chk_auth_token.py
2026-01-31 22:34:57 +09:00

105 lines
3.6 KiB
Python

# -*- coding: utf-8 -*-
"""
Created on 2025-06-19
"""
import sys
import logging
import pandas as pd
sys.path.extend(['../..', '.']) # kis_auth 파일 경로 추가
import kis_auth as ka
from auth_token import auth_token
# 로깅 설정
logging.basicConfig(level=logging.INFO, format='%(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
##############################################################################################
# [인증] OAuth 접근토큰 발급 테스트
##############################################################################################
# 통합 컬럼 매핑
COLUMN_MAPPING = {
'access_token': '접근토큰',
'token_type': '접근토큰유형',
'expires_in': '접근토큰유효기간_초',
'access_token_token_expired': '접근토큰유효기간_일시표시'
}
def main():
"""
OAuth 접근토큰 발급 테스트 함수
Parameters:
- grant_type (str): 권한부여 Type (client_credentials)
- appkey (str): 앱키 (한국투자증권 홈페이지에서 발급받은 appkey)
- appsecret (str): 앱시크릿키 (한국투자증권 홈페이지에서 발급받은 appsecret)
- env_dv (str): 환경구분 (real: 실전, demo: 모의)
Returns:
- pd.DataFrame: OAuth 토큰 발급 결과
Response Fields:
- access_token: 접근토큰 (OAuth 토큰이 필요한 API 경우 발급한 Access token)
- token_type: 접근토큰유형 ("Bearer")
- expires_in: 접근토큰 유효기간(초)
- access_token_token_expired: 접근토큰 유효기간(일시표시)
Example:
>>> df = auth_token(grant_type="client_credentials", appkey=trenv.my_app, appsecret=trenv.my_sec, env_dv="real")
"""
try:
# pandas 출력 옵션 설정
pd.set_option('display.max_columns', None) # 모든 컬럼 표시
pd.set_option('display.width', None) # 출력 너비 제한 해제
pd.set_option('display.max_rows', None) # 모든 행 표시
# 환경 설정에서 앱키와 앱시크릿 가져오기
config = ka.getEnv()
# 실전투자용 앱키/앱시크릿 사용 (모의투자의 경우 paper_app, paper_sec 사용)
appkey = config.get("my_app", "")
appsecret = config.get("my_sec", "")
# 앱키와 앱시크릿이 설정되어 있는지 확인
if not appkey or not appsecret:
logger.error("앱키 또는 앱시크릿이 설정되지 않았습니다. kis_devlp.yaml 파일을 확인해주세요.")
logger.info("필요한 설정: my_app (앱키), my_sec (앱시크릿)")
return
# API 호출
logger.info("OAuth 접근토큰 발급 API 호출 시작")
result = auth_token(
grant_type="client_credentials",
appkey=appkey,
appsecret=appsecret,
env_dv="real" # 실전 환경으로 설정 (필요시 "demo"로 변경)
)
# 결과 확인
if result.empty:
logger.warning("조회된 데이터가 없습니다.")
return
# 결과 처리
logger.info("=== OAuth 접근토큰 발급 결과 ===")
logger.info("사용 가능한 컬럼: %s", result.columns.tolist())
# 통합 컬럼명 한글 변환 (필요한 컬럼만 자동 매핑됨)
result = result.rename(columns=COLUMN_MAPPING)
logger.info("결과:")
print(result)
except Exception as e:
logger.error("에러 발생: %s", str(e))
raise
if __name__ == "__main__":
main()