initial commit
This commit is contained in:
@@ -0,0 +1,111 @@
|
||||
"""
|
||||
Created on 20250115
|
||||
"""
|
||||
|
||||
import sys
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
from pension_inquire_present_balance import pension_inquire_present_balance
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내주식] 주문/계좌 > 퇴직연금 체결기준잔고[v1_국내주식-032]
|
||||
##############################################################################################
|
||||
|
||||
COLUMN_MAPPING = {
|
||||
'cblc_dvsn': '잔고구분',
|
||||
'cblc_dvsn_name': '잔고구분명',
|
||||
'pdno': '상품번호',
|
||||
'prdt_name': '상품명',
|
||||
'hldg_qty': '보유수량',
|
||||
'slpsb_qty': '매도가능수량',
|
||||
'pchs_avg_pric': '매입평균가격',
|
||||
'evlu_pfls_amt': '평가손익금액',
|
||||
'evlu_pfls_rt': '평가손익율',
|
||||
'prpr': '현재가',
|
||||
'evlu_amt': '평가금액',
|
||||
'pchs_amt': '매입금액',
|
||||
'cblc_weit': '잔고비중',
|
||||
'pchs_amt_smtl_amt': '매입금액합계금액',
|
||||
'evlu_amt_smtl_amt': '평가금액합계금액',
|
||||
'evlu_pfls_smtl_amt': '평가손익합계금액',
|
||||
'trad_pfls_smtl': '매매손익합계',
|
||||
'thdt_tot_pfls_amt': '당일총손익금액',
|
||||
'pftrt': '수익률'
|
||||
}
|
||||
|
||||
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()
|
||||
|
||||
trenv = ka.getTREnv()
|
||||
|
||||
# Case 1: 퇴직연금 체결기준잔고 조회
|
||||
logging.info("=== 퇴직연금 체결기준잔고 조회 ===")
|
||||
try:
|
||||
result1, result2 = pension_inquire_present_balance(cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod,
|
||||
user_dvsn_cd="00"
|
||||
)
|
||||
except ValueError as e:
|
||||
logging.error("에러 발생: %s" % str(e))
|
||||
return
|
||||
|
||||
# output1 처리
|
||||
logging.info("=== output1 (보유종목 정보) ===")
|
||||
logging.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)
|
||||
|
||||
logging.info("결과:")
|
||||
print(result1)
|
||||
|
||||
# output2 처리
|
||||
logging.info("=== output2 (계좌 요약 정보) ===")
|
||||
logging.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)
|
||||
|
||||
logging.info("결과:")
|
||||
print(result2)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,86 @@
|
||||
"""
|
||||
Created on 20250115
|
||||
"""
|
||||
|
||||
import sys
|
||||
from typing import Tuple
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내주식] 주문/계좌 > 퇴직연금 체결기준잔고[v1_국내주식-032]
|
||||
##############################################################################################
|
||||
|
||||
# 상수 정의
|
||||
API_URL = "/uapi/domestic-stock/v1/trading/pension/inquire-present-balance"
|
||||
|
||||
|
||||
def pension_inquire_present_balance(
|
||||
cano: str, # 종합계좌번호
|
||||
acnt_prdt_cd: str, # 계좌상품코드
|
||||
user_dvsn_cd: str, # 상품번호
|
||||
FK100: str = "", # 연속조회검색조건100
|
||||
NK100: str = "" # 연속조회키100
|
||||
) -> Tuple[pd.DataFrame, pd.DataFrame]:
|
||||
"""
|
||||
[국내주식] 주문/계좌 > 퇴직연금 체결기준잔고[v1_국내주식-032]
|
||||
|
||||
※ 55번 계좌(DC가입자계좌)의 경우 해당 API 이용이 불가합니다.
|
||||
KIS Developers API의 경우 HTS ID에 반드시 연결되어있어야만 API 신청 및 앱정보 발급이 가능한 서비스로 개발되어서 실물계좌가 아닌 55번 계좌는 API 이용이 불가능한 점 양해 부탁드립니다.
|
||||
|
||||
Args:
|
||||
cano (str): [필수] 종합계좌번호 (ex. '12345678')
|
||||
acnt_prdt_cd (str): [필수] 계좌상품코드 (ex. '29')
|
||||
user_dvsn_cd (str): [필수] 상품번호 (ex. '00')
|
||||
FK100 (str): 연속조회검색조건100
|
||||
NK100 (str): 연속조회키100
|
||||
|
||||
Returns:
|
||||
Tuple[pd.DataFrame, pd.DataFrame]: (output1, output2) 데이터프레임 튜플
|
||||
|
||||
Example:
|
||||
>>> df1, df2 = pension_inquire_present_balance(cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, user_dvsn_cd="00")
|
||||
>>> print(df1)
|
||||
>>> print(df2)
|
||||
"""
|
||||
|
||||
# 필수 파라미터 검증
|
||||
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 user_dvsn_cd == "":
|
||||
raise ValueError("user_dvsn_cd is required (e.g. '00')")
|
||||
|
||||
tr_id = "TTTC2202R" # 퇴직연금 체결기준잔고
|
||||
|
||||
params = {
|
||||
"CANO": cano, # 종합계좌번호
|
||||
"ACNT_PRDT_CD": acnt_prdt_cd, # 계좌상품코드
|
||||
"USER_DVSN_CD": user_dvsn_cd, # 상품번호
|
||||
"CTX_AREA_FK100": FK100, # 연속조회검색조건100
|
||||
"CTX_AREA_NK100": NK100 # 연속조회키100
|
||||
}
|
||||
|
||||
res = ka._url_fetch(API_URL, tr_id, "", params)
|
||||
|
||||
if res.isOK():
|
||||
# output1 (array) - 보유종목 정보
|
||||
output1_data = pd.DataFrame(res.getBody().output1)
|
||||
|
||||
# output2 (array) - 계좌 요약 정보
|
||||
output2_data = pd.DataFrame(res.getBody().output2)
|
||||
|
||||
return output1_data, output2_data
|
||||
else:
|
||||
res.printError(url=API_URL)
|
||||
return pd.DataFrame(), pd.DataFrame()
|
||||
Reference in New Issue
Block a user