initial commit
This commit is contained in:
@@ -0,0 +1,144 @@
|
||||
"""
|
||||
Created on 20250601
|
||||
"""
|
||||
|
||||
import sys
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
from inquire_ngt_balance import inquire_ngt_balance
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내선물옵션] 주문/계좌 > (야간)선물옵션 잔고현황 [국내선물-010]
|
||||
##############################################################################################
|
||||
|
||||
COLUMN_MAPPING = {
|
||||
'cano': '종합계좌번호',
|
||||
'acnt_prdt_cd': '계좌상품코드',
|
||||
'pdno': '상품번호',
|
||||
'prdt_type_cd': '상품유형코드',
|
||||
'shtn_pdno': '단축상품번호',
|
||||
'prdt_name': '상품명',
|
||||
'sll_buy_dvsn_name': '매도매수구분명',
|
||||
'sll_buy_dvsn_cd': '매도매수구분코드',
|
||||
'trad_dvsn_name': '매매구분명',
|
||||
'cblc_qty': '잔고수량',
|
||||
'excc_unpr': '정산단가',
|
||||
'ccld_avg_unpr1': '체결평균단가1',
|
||||
'idx_clpr': '지수종가',
|
||||
'pchs_amt': '매입금액',
|
||||
'evlu_amt': '평가금액',
|
||||
'evlu_pfls_amt': '평가손익금액',
|
||||
'trad_pfls_amt': '매매손익금액',
|
||||
'lqd_psbl_qty': '청산가능수량',
|
||||
'dnca_cash': '예수금현금',
|
||||
'frcr_dncl_amt': '외화예수금액',
|
||||
'dnca_sbst': '예수금대용',
|
||||
'tot_dncl_amt': '총예수금액',
|
||||
'cash_mgna': '현금증거금',
|
||||
'sbst_mgna': '대용증거금',
|
||||
'mgna_tota': '증거금총액',
|
||||
'opt_dfpa': '옵션차금',
|
||||
'thdt_dfpa': '당일차금',
|
||||
'rnwl_dfpa': '갱신차금',
|
||||
'fee': '수수료',
|
||||
'nxdy_dnca': '익일예수금',
|
||||
'nxdy_dncl_amt': '익일예수금액',
|
||||
'prsm_dpast': '추정예탁자산',
|
||||
'pprt_ord_psbl_cash': '적정주문가능현금',
|
||||
'add_mgna_cash': '추가증거금현금',
|
||||
'add_mgna_tota': '추가증거금총액',
|
||||
'futr_trad_pfls_amt': '선물매매손익금액',
|
||||
'opt_trad_pfls_amt': '옵션매매손익금액',
|
||||
'futr_evlu_pfls_amt': '선물평가손익금액',
|
||||
'opt_evlu_pfls_amt': '옵션평가손익금액',
|
||||
'trad_pfls_amt_smtl': '매매손익금액합계',
|
||||
'evlu_pfls_amt_smtl': '평가손익금액합계',
|
||||
'wdrw_psbl_tot_amt': '인출가능총금액',
|
||||
'ord_psbl_cash': '주문가능현금',
|
||||
'ord_psbl_sbst': '주문가능대용',
|
||||
'ord_psbl_tota': '주문가능총액',
|
||||
'mmga_tot_amt': '유지증거금총금액',
|
||||
'mmga_cash_amt': '유지증거금현금금액',
|
||||
'mtnc_rt': '유지비율',
|
||||
'isfc_amt': '부족금액',
|
||||
'pchs_amt_smtl': '매입금액합계',
|
||||
'evlu_amt_smtl': '평가금액합계'
|
||||
}
|
||||
|
||||
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("=== Case 1 조회 ===")
|
||||
try:
|
||||
result1, result2 = inquire_ngt_balance(
|
||||
cano=trenv.my_acct,
|
||||
acnt_prdt_cd=trenv.my_prod,
|
||||
mgna_dvsn="01",
|
||||
excc_stat_cd="1"
|
||||
)
|
||||
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,131 @@
|
||||
"""
|
||||
Created on 20250601
|
||||
"""
|
||||
|
||||
|
||||
import sys
|
||||
import time
|
||||
from typing import Optional, Tuple
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내선물옵션] 주문/계좌 > (야간)선물옵션 잔고현황 [국내선물-010]
|
||||
##############################################################################################
|
||||
|
||||
# 상수 정의
|
||||
API_URL = "/uapi/domestic-futureoption/v1/trading/inquire-ngt-balance"
|
||||
|
||||
def inquire_ngt_balance(
|
||||
cano: str, # 종합계좌번호
|
||||
acnt_prdt_cd: str, # 계좌상품코드
|
||||
mgna_dvsn: str, # 증거금구분
|
||||
excc_stat_cd: str, # 정산상태코드
|
||||
acnt_pwd: str = "", # 계좌비밀번호
|
||||
FK200: str = "", # 연속조회검색조건200
|
||||
NK200: str = "", # 연속조회키200
|
||||
tr_cont: str = "", # 연속거래여부
|
||||
dataframe1: Optional[pd.DataFrame] = None, # 누적 데이터프레임1
|
||||
dataframe2: Optional[pd.DataFrame] = None, # 누적 데이터프레임2
|
||||
depth: int = 0, # 내부 재귀깊이 (자동관리)
|
||||
max_depth: int = 10 # 최대 재귀 횟수 제한
|
||||
) -> Tuple[pd.DataFrame, pd.DataFrame]:
|
||||
"""
|
||||
(야간)선물옵션 잔고현황 API입니다.
|
||||
|
||||
Args:
|
||||
cano (str): [필수] 종합계좌번호 (ex. 계좌번호 체계(8-2)의 앞 8자리)
|
||||
acnt_prdt_cd (str): [필수] 계좌상품코드 (ex. 계좌번호 체계(8-2)의 뒤 2자리)
|
||||
mgna_dvsn (str): [필수] 증거금구분 (ex. 01:개시, 02:유지)
|
||||
excc_stat_cd (str): [필수] 정산상태코드 (ex. 1:정산, 2:본정산)
|
||||
acnt_pwd (str): 계좌비밀번호
|
||||
FK200 (str): 연속조회검색조건200
|
||||
NK200 (str): 연속조회키200
|
||||
tr_cont (str): 연속거래여부
|
||||
dataframe1 (Optional[pd.DataFrame]): 누적 데이터프레임1
|
||||
dataframe2 (Optional[pd.DataFrame]): 누적 데이터프레임2
|
||||
depth (int): 내부 재귀깊이 (자동관리)
|
||||
max_depth (int): 최대 재귀 횟수 제한
|
||||
|
||||
Returns:
|
||||
Tuple[pd.DataFrame, pd.DataFrame]: (output1 데이터, output2 데이터)
|
||||
|
||||
Example:
|
||||
>>> df1, df2 = inquire_ngt_balance("12345678", "01", "01", "1")
|
||||
>>> print(df1, df2)
|
||||
"""
|
||||
|
||||
# 필수 파라미터 검증
|
||||
if cano == "":
|
||||
raise ValueError("cano is required (e.g. '계좌번호 체계(8-2)의 앞 8자리')")
|
||||
|
||||
if acnt_prdt_cd == "":
|
||||
raise ValueError("acnt_prdt_cd is required (e.g. '계좌번호 체계(8-2)의 뒤 2자리')")
|
||||
|
||||
if mgna_dvsn == "":
|
||||
raise ValueError("mgna_dvsn is required (e.g. '01:개시, 02:유지')")
|
||||
|
||||
if excc_stat_cd == "":
|
||||
raise ValueError("excc_stat_cd is required (e.g. '1:정산, 2:본정산')")
|
||||
|
||||
# 재귀 깊이 제한 확인
|
||||
if depth > max_depth:
|
||||
logging.warning("Max recursive depth reached.")
|
||||
return (
|
||||
dataframe1 if dataframe1 is not None else pd.DataFrame(),
|
||||
dataframe2 if dataframe2 is not None else pd.DataFrame()
|
||||
)
|
||||
|
||||
tr_id = "CTFN6118R"
|
||||
|
||||
params = {
|
||||
"CANO": cano,
|
||||
"ACNT_PRDT_CD": acnt_prdt_cd,
|
||||
"MGNA_DVSN": mgna_dvsn,
|
||||
"EXCC_STAT_CD": excc_stat_cd,
|
||||
"ACNT_PWD": acnt_pwd,
|
||||
"CTX_AREA_FK200": FK200,
|
||||
"CTX_AREA_NK200": NK200
|
||||
}
|
||||
|
||||
res = ka._url_fetch(API_URL, tr_id, tr_cont, params)
|
||||
|
||||
if res.isOK():
|
||||
# output1 처리 (array)
|
||||
current_data1 = pd.DataFrame(res.getBody().output1)
|
||||
if dataframe1 is not None:
|
||||
dataframe1 = pd.concat([dataframe1, current_data1], ignore_index=True)
|
||||
else:
|
||||
dataframe1 = current_data1
|
||||
|
||||
# output2 처리 (object)
|
||||
current_data2 = pd.DataFrame(res.getBody().output2, index=[0])
|
||||
if dataframe2 is not None:
|
||||
dataframe2 = pd.concat([dataframe2, current_data2], ignore_index=True)
|
||||
else:
|
||||
dataframe2 = current_data2
|
||||
|
||||
tr_cont = res.getHeader().tr_cont
|
||||
FK200 = res.getBody().ctx_area_fk200
|
||||
NK200 = res.getBody().ctx_area_nk200
|
||||
|
||||
if tr_cont in ["M", "F"]: # 다음 페이지 존재
|
||||
logging.info("Call Next page...")
|
||||
ka.smart_sleep() # 시스템 안정적 운영을 위한 지연
|
||||
return inquire_ngt_balance(
|
||||
cano, acnt_prdt_cd, mgna_dvsn, excc_stat_cd, acnt_pwd,
|
||||
FK200, NK200, "N", dataframe1, dataframe2, depth + 1, max_depth
|
||||
)
|
||||
else:
|
||||
logging.info("Data fetch complete.")
|
||||
return (dataframe1, dataframe2)
|
||||
else:
|
||||
res.printError(url=API_URL)
|
||||
return (pd.DataFrame(), pd.DataFrame())
|
||||
Reference in New Issue
Block a user