GoogleAIBuilder/KisStock
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

initial commit

Browse Source
This commit is contained in:
tindevil
2026-01-31 22:34:57 +09:00
commit f1301de543
875 changed files with 196598 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

223
한국투자증권(API)/examples_llm/overseas_stock/inquire_present_balance/chk_inquire_present_balance.py Normal file
View File

@@ -0,0 +1,223 @@
# -*- coding: utf-8 -*-
"""
Created on 2025-06-30
"""
import sys
import logging
import pandas as pd
sys.path.extend(['../..', '.']) # kis_auth 파일 경로 추가
import kis_auth as ka
from inquire_present_balance import inquire_present_balance
# 로깅 설정
logging.basicConfig(level=logging.INFO, format='%(levelname)s - %(message)s')
logger = logging.getLogger(__name__)
##############################################################################################
# [해외주식] 주문/계좌 > 해외주식 체결기준현재잔고 [v1_해외주식-008]
##############################################################################################
# 컬럼명 매핑 (한글 변환용)
COLUMN_MAPPING = {
'cblc_qty13': '잔고수량13',
'thdt_buy_ccld_qty1': '당일매수체결수량1',
'thdt_sll_ccld_qty1': '당일매도체결수량1',
'ccld_qty_smtl1': '체결수량합계1',
'ord_psbl_qty1': '주문가능수량1',
'frcr_pchs_amt': '외화매입금액',
'frcr_evlu_amt2': '외화평가금액2',
'evlu_pfls_amt2': '평가손익금액2',
'evlu_pfls_rt1': '평가손익율1',
'pdno': '상품번호',
'bass_exrt': '기준환율',
'buy_crcy_cd': '매수통화코드',
'ovrs_now_pric1': '해외현재가격1',
'avg_unpr3': '평균단가3',
'tr_mket_name': '거래시장명',
'natn_kor_name': '국가한글명',
'pchs_rmnd_wcrc_amt': '매입잔액원화금액',
'thdt_buy_ccld_frcr_amt': '당일매수체결외화금액',
'thdt_sll_ccld_frcr_amt': '당일매도체결외화금액',
'unit_amt': '단위금액',
'std_pdno': '표준상품번호',
'prdt_type_cd': '상품유형코드',
'loan_rmnd': '대출잔액',
'loan_dt': '대출일자',
'loan_expd_dt': '대출만기일자',
'ovrs_excg_cd': '해외거래소코드',
'item_lnkg_excg_cd': '종목연동거래소코드',
'crcy_cd': '통화코드',
'frcr_buy_amt_smtl': '외화매수금액합계',
'frcr_sll_amt_smtl': '외화매도금액합계',
'frcr_dncl_amt_2': '외화예수금액2',
'frst_bltn_exrt': '최초고시환율',
'frcr_buy_mgn_amt': '외화매수증거금액',
'frcr_etc_mgna': '외화기타증거금',
'frcr_drwg_psbl_amt_1': '외화출금가능금액1',
'frcr_evlu_amt2': '출금가능원화금액',
'acpl_cstd_crcy_yn': '현지보관통화여부',
'nxdy_frcr_drwg_psbl_amt': '익일외화출금가능금액',
'output3': '응답상세3',
'pchs_amt_smtl': '매입금액합계',
'evlu_amt_smtl': '평가금액합계',
'evlu_pfls_amt_smtl': '평가손익금액합계',
'dncl_amt': '예수금액',
'cma_evlu_amt': 'CMA평가금액',
'tot_dncl_amt': '총예수금액',
'etc_mgna': '기타증거금',
'wdrw_psbl_tot_amt': '인출가능총금액',
'frcr_evlu_tota': '외화평가총액',
'evlu_erng_rt1': '평가수익율1',
'pchs_amt_smtl_amt': '매입금액합계금액',
'evlu_amt_smtl_amt': '평가금액합계금액',
'tot_evlu_pfls_amt': '총평가손익금액',
'tot_asst_amt': '총자산금액',
'buy_mgn_amt': '매수증거금액',
'mgna_tota': '증거금총액',
'frcr_use_psbl_amt': '외화사용가능금액',
'ustl_sll_amt_smtl': '미결제매도금액합계',
'ustl_buy_amt_smtl': '미결제매수금액합계',
'tot_frcr_cblc_smtl': '총외화잔고합계',
'tot_loan_amt': '총대출금액'
}
# 숫자형 컬럼 정의 (소수점 처리용)
NUMERIC_COLUMNS = [
'cblc_qty13', 'thdt_buy_ccld_qty1', 'thdt_sll_ccld_qty1', 'ccld_qty_smtl1', 'ord_psbl_qty1',
'frcr_pchs_amt', 'frcr_evlu_amt2', 'evlu_pfls_amt2', 'evlu_pfls_rt1', 'bass_exrt',
'ovrs_now_pric1', 'avg_unpr3', 'pchs_rmnd_wcrc_amt', 'thdt_buy_ccld_frcr_amt', 'thdt_sll_ccld_frcr_amt',
'unit_amt', 'loan_rmnd', 'frcr_buy_amt_smtl', 'frcr_sll_amt_smtl', 'frcr_dncl_amt_2',
'frst_bltn_exrt', 'frcr_buy_mgn_amt', 'frcr_etc_mgna', 'frcr_drwg_psbl_amt_1', 'frcr_evlu_amt2',
'nxdy_frcr_drwg_psbl_amt', 'pchs_amt_smtl', 'evlu_amt_smtl', 'evlu_pfls_amt_smtl', 'dncl_amt',
'cma_evlu_amt', 'tot_dncl_amt', 'etc_mgna', 'wdrw_psbl_tot_amt', 'frcr_evlu_tota',
'evlu_erng_rt1', 'pchs_amt_smtl_amt', 'evlu_amt_smtl_amt', 'tot_evlu_pfls_amt', 'tot_asst_amt',
'buy_mgn_amt', 'mgna_tota', 'frcr_use_psbl_amt', 'ustl_sll_amt_smtl', 'ustl_buy_amt_smtl',
'tot_frcr_cblc_smtl', 'tot_loan_amt'
]
def main():
"""
[해외주식] 주문/계좌
해외주식 체결기준현재잔고[v1_해외주식-008]
해외주식 체결기준현재잔고 테스트 함수
Parameters:
- cano (str): 종합계좌번호 (계좌번호 체계(8-2)의 앞 8자리)
- acnt_prdt_cd (str): 계좌상품코드 (계좌번호 체계(8-2)의 뒤 2자리)
- wcrc_frcr_dvsn_cd (str): 원화외화구분코드 (01 : 원화 02 : 외화)
- natn_cd (str): 국가코드 (000 전체 840 미국 344 홍콩 156 중국 392 일본 704 베트남)
- tr_mket_cd (str): 거래시장코드 ([Request body NATN_CD 000 설정] 00 : 전체 [Request body NATN_CD 840 설정] 00 : 전체 01 : 나스닥(NASD) 02 : 뉴욕거래소(NYSE) 03 : 미국(PINK SHEETS) 04 : 미국(OTCBB) 05 : 아멕스(AMEX) [Request body NATN_CD 156 설정] 00 : 전체 01 : 상해B 02 : 심천B 03 : 상해A 04 : 심천A [Request body NATN_CD 392 설정] 01 : 일본 [Request body NATN_CD 704 설정] 01 : 하노이거래 02 : 호치민거래소 [Request body NATN_CD 344 설정] 01 : 홍콩 02 : 홍콩CNY 03 : 홍콩USD)
- inqr_dvsn_cd (str): 조회구분코드 (00 : 전체 01 : 일반해외주식 02 : 미니스탁)
- env_dv (str): 실전모의구분 (real:실전, demo:모의)
Returns:
- DataFrame: 해외주식 체결기준현재잔고 결과
Example:
>>> df1, df2, df3 = inquire_present_balance(cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, wcrc_frcr_dvsn_cd="02", natn_cd="000", tr_mket_cd="00", inqr_dvsn_cd="00", 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) # 모든 행 표시
# 실전/모의투자 선택 (모의투자 지원 로직)
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("토큰 발급 완료")
trenv = ka.getTREnv()
# API 호출
logger.info("API 호출")
result1, result2, result3 = inquire_present_balance(
cano=trenv.my_acct, # 종합계좌번호
acnt_prdt_cd=trenv.my_prod, # 계좌상품코드
wcrc_frcr_dvsn_cd="02", # 원화외화구분코드
natn_cd="000", # 국가코드
tr_mket_cd="00", # 거래시장코드
inqr_dvsn_cd="00", # 조회구분코드
env_dv=env_dv, # 실전모의구분
)
# 결과 확인
results = [result1, result2, result3]
if all(result is None or result.empty for result in results):
logger.warning("조회된 데이터가 없습니다.")
return
# output1 결과 처리
logger.info("=== output1 조회 ===")
if not result1.empty:
logger.info("사용 가능한 컬럼: %s", result1.columns.tolist())
# 통합 컬럼명 한글 변환 (필요한 컬럼만 자동 매핑됨)
result1 = result1.rename(columns=COLUMN_MAPPING)
# 숫자형 컬럼 처리
for col in NUMERIC_COLUMNS:
if col in result1.columns:
result1[col] = pd.to_numeric(result1[col], errors='coerce').round(2)
logger.info("output1 결과:")
print(result1)
else:
logger.info("output1 데이터가 없습니다.")
# output2 결과 처리
logger.info("=== output2 조회 ===")
if not result2.empty:
logger.info("사용 가능한 컬럼: %s", result2.columns.tolist())
# 통합 컬럼명 한글 변환 (필요한 컬럼만 자동 매핑됨)
result2 = result2.rename(columns=COLUMN_MAPPING)
# 숫자형 컬럼 처리
for col in NUMERIC_COLUMNS:
if col in result2.columns:
result2[col] = pd.to_numeric(result2[col], errors='coerce').round(2)
logger.info("output2 결과:")
print(result2)
else:
logger.info("output2 데이터가 없습니다.")
# output3 결과 처리
logger.info("=== output3 조회 ===")
if not result3.empty:
logger.info("사용 가능한 컬럼: %s", result3.columns.tolist())
# 통합 컬럼명 한글 변환 (필요한 컬럼만 자동 매핑됨)
result3 = result3.rename(columns=COLUMN_MAPPING)
# 숫자형 컬럼 처리
for col in NUMERIC_COLUMNS:
if col in result3.columns:
result3[col] = pd.to_numeric(result3[col], errors='coerce').round(2)
logger.info("output3 결과:")
print(result3)
else:
logger.info("output3 데이터가 없습니다.")
except Exception as e:
logger.error("에러 발생: %s", str(e))
raise
if __name__ == "__main__":
main()

215
한국투자증권(API)/examples_llm/overseas_stock/inquire_present_balance/inquire_present_balance.py Normal file
View File

@@ -0,0 +1,215 @@
# -*- coding: utf-8 -*-
"""
Created on 2025-06-30
"""
import logging
import time
from typing import Optional, Tuple
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_해외주식-008]
##############################################################################################
# 상수 정의
API_URL = "/uapi/overseas-stock/v1/trading/inquire-present-balance"
def inquire_present_balance(
cano: str, # 종합계좌번호
acnt_prdt_cd: str, # 계좌상품코드
wcrc_frcr_dvsn_cd: str, # 원화외화구분코드
natn_cd: str, # 국가코드
tr_mket_cd: str, # 거래시장코드
inqr_dvsn_cd: str, # 조회구분코드
env_dv: str = "real", # 실전모의구분
dataframe1: Optional[pd.DataFrame] = None, # 누적 데이터프레임 (output1)
dataframe2: Optional[pd.DataFrame] = None, # 누적 데이터프레임 (output2)
dataframe3: Optional[pd.DataFrame] = None, # 누적 데이터프레임 (output3)
tr_cont: str = "",
depth: int = 0,
max_depth: int = 10
) -> Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]:
"""
[해외주식] 주문/계좌
해외주식 체결기준현재잔고[v1_해외주식-008]
해외주식 체결기준현재잔고 API를 호출하여 DataFrame으로 반환합니다.
Args:
cano (str): 계좌번호 체계(8-2)의 앞 8자리
acnt_prdt_cd (str): 계좌번호 체계(8-2)의 뒤 2자리
wcrc_frcr_dvsn_cd (str): 01 : 원화 02 : 외화
natn_cd (str): 000 전체 840 미국 344 홍콩 156 중국 392 일본 704 베트남
tr_mket_cd (str): [Request body NATN_CD 000 설정] 00 : 전체 [Request body NATN_CD 840 설정] 00 : 전체 01 : 나스닥(NASD) 02 : 뉴욕거래소(NYSE) 03 : 미국(PINK SHEETS) 04 : 미국(OTCBB) 05 : 아멕스(AMEX) [Request body NATN_CD 156 설정] 00 : 전체 01 : 상해B 02 : 심천B 03 : 상해A 04 : 심천A [Request body NATN_CD 392 설정] 01 : 일본 [Request body NATN_CD 704 설정] 01 : 하노이거래 02 : 호치민거래소 [Request body NATN_CD 344 설정] 01 : 홍콩 02 : 홍콩CNY 03 : 홍콩USD
inqr_dvsn_cd (str): 00 : 전체 01 : 일반해외주식 02 : 미니스탁
env_dv (str): 실전모의구분 (real:실전, demo:모의)
dataframe1 (Optional[pd.DataFrame]): 누적 데이터프레임 (output1)
dataframe2 (Optional[pd.DataFrame]): 누적 데이터프레임 (output2)
dataframe3 (Optional[pd.DataFrame]): 누적 데이터프레임 (output3)
tr_cont (str): 연속 거래 여부
depth (int): 현재 재귀 깊이
max_depth (int): 최대 재귀 깊이 (기본값: 10)
Returns:
Tuple[pd.DataFrame, pd.DataFrame, pd.DataFrame]: 해외주식 체결기준현재잔고 데이터
Example:
>>> df1, df2, df3 = inquire_present_balance(
... cano=trenv.my_acct,
... acnt_prdt_cd=trenv.my_prod,
... wcrc_frcr_dvsn_cd="01",
... natn_cd="000",
... tr_mket_cd="00",
... inqr_dvsn_cd="00"
... )
>>> print(df1)
>>> print(df2)
>>> print(df3)
"""
# [필수 파라미터 검증]
if not cano:
logger.error("cano is required. (e.g. '810XXXXX')")
raise ValueError("cano is required. (e.g. '810XXXXX')")
if not acnt_prdt_cd:
logger.error("acnt_prdt_cd is required. (e.g. '01')")
raise ValueError("acnt_prdt_cd is required. (e.g. '01')")
if not wcrc_frcr_dvsn_cd:
logger.error("wcrc_frcr_dvsn_cd is required. (e.g. '01')")
raise ValueError("wcrc_frcr_dvsn_cd is required. (e.g. '01')")
if not natn_cd:
logger.error("natn_cd is required. (e.g. '000')")
raise ValueError("natn_cd is required. (e.g. '000')")
if not tr_mket_cd:
logger.error("tr_mket_cd is required. (e.g. '00')")
raise ValueError("tr_mket_cd is required. (e.g. '00')")
if not inqr_dvsn_cd:
logger.error("inqr_dvsn_cd is required. (e.g. '00')")
raise ValueError("inqr_dvsn_cd is required. (e.g. '00')")
# 최대 재귀 깊이 체크
if depth >= max_depth:
logger.warning("Maximum recursion depth (%d) reached. Stopping further requests.", max_depth)
return dataframe1 if dataframe1 is not None else pd.DataFrame(), dataframe2 if dataframe2 is not None else pd.DataFrame(), dataframe3 if dataframe3 is not None else pd.DataFrame()
# TR ID 설정 (모의투자 지원 로직)
if env_dv == "real":
tr_id = "CTRP6504R" # 실전투자용 TR ID
elif env_dv == "demo":
tr_id = "VTRP6504R" # 모의투자용 TR ID
else:
raise ValueError("env_dv can only be 'real' or 'demo'")
params = {
"CANO": cano,
"ACNT_PRDT_CD": acnt_prdt_cd,
"WCRC_FRCR_DVSN_CD": wcrc_frcr_dvsn_cd,
"NATN_CD": natn_cd,
"TR_MKET_CD": tr_mket_cd,
"INQR_DVSN_CD": inqr_dvsn_cd,
}
res = ka._url_fetch(api_url=API_URL, ptr_id=tr_id, tr_cont=tr_cont, params=params)
if res.isOK():
# output1 처리
if hasattr(res.getBody(), 'output1'):
output_data = res.getBody().output1
if output_data:
if isinstance(output_data, list):
current_data1 = pd.DataFrame(output_data)
else:
current_data1 = pd.DataFrame([output_data])
if dataframe1 is not None:
dataframe1 = pd.concat([dataframe1, current_data1], ignore_index=True)
else:
dataframe1 = current_data1
else:
if dataframe1 is None:
dataframe1 = pd.DataFrame()
else:
if dataframe1 is None:
dataframe1 = pd.DataFrame()
# output2 처리
if hasattr(res.getBody(), 'output2'):
output_data = res.getBody().output2
if output_data:
if isinstance(output_data, list):
current_data2 = pd.DataFrame(output_data)
else:
current_data2 = pd.DataFrame([output_data])
if dataframe2 is not None:
dataframe2 = pd.concat([dataframe2, current_data2], ignore_index=True)
else:
dataframe2 = current_data2
else:
if dataframe2 is None:
dataframe2 = pd.DataFrame()
else:
if dataframe2 is None:
dataframe2 = pd.DataFrame()
# output3 처리
if hasattr(res.getBody(), 'output3'):
output_data = res.getBody().output3
if output_data:
if isinstance(output_data, list):
current_data3 = pd.DataFrame(output_data)
else:
current_data3 = pd.DataFrame([output_data])
if dataframe3 is not None:
dataframe3 = pd.concat([dataframe3, current_data3], ignore_index=True)
else:
dataframe3 = current_data3
else:
if dataframe3 is None:
dataframe3 = pd.DataFrame()
else:
if dataframe3 is None:
dataframe3 = pd.DataFrame()
tr_cont = res.getHeader().tr_cont
if tr_cont in ["M", "F"]:
logger.info("Calling next page...")
ka.smart_sleep()
return inquire_present_balance(
cano=cano,
acnt_prdt_cd=acnt_prdt_cd,
wcrc_frcr_dvsn_cd=wcrc_frcr_dvsn_cd,
natn_cd=natn_cd,
tr_mket_cd=tr_mket_cd,
inqr_dvsn_cd=inqr_dvsn_cd,
env_dv=env_dv,
dataframe1=dataframe1,
dataframe2=dataframe2,
dataframe3=dataframe3,
tr_cont="N",
depth=depth + 1,
max_depth=max_depth
)
else:
logger.info("Data fetch complete.")
return dataframe1, dataframe2, dataframe3
else:
logger.error("API call failed: %s - %s", res.getErrorCode(), res.getErrorMessage())
res.printError(API_URL)
# 이미 수집된 데이터가 있으면 그것을 반환, 없으면 빈 DataFrame 반환
if dataframe1 is not None and not dataframe1.empty:
logger.info("Returning already collected data due to API error.")
return dataframe1, dataframe2 if dataframe2 is not None else pd.DataFrame(), dataframe3 if dataframe3 is not None else pd.DataFrame()
else:
return pd.DataFrame(), pd.DataFrame(), pd.DataFrame()