initial commit
This commit is contained in:
@@ -0,0 +1,116 @@
|
||||
"""
|
||||
Created on 20250601
|
||||
"""
|
||||
|
||||
import sys
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
from period_rights import period_rights
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내주식] 주문/계좌 > 기간별계좌권리현황조회 [국내주식-211]
|
||||
##############################################################################################
|
||||
|
||||
COLUMN_MAPPING = {
|
||||
'acno10': '계좌번호10',
|
||||
'rght_type_cd': '권리유형코드',
|
||||
'bass_dt': '기준일자',
|
||||
'rght_cblc_type_cd': '권리잔고유형코드',
|
||||
'rptt_pdno': '대표상품번호',
|
||||
'pdno': '상품번호',
|
||||
'prdt_type_cd': '상품유형코드',
|
||||
'shtn_pdno': '단축상품번호',
|
||||
'prdt_name': '상품명',
|
||||
'cblc_qty': '잔고수량',
|
||||
'last_alct_qty': '최종배정수량',
|
||||
'excs_alct_qty': '초과배정수량',
|
||||
'tot_alct_qty': '총배정수량',
|
||||
'last_ftsk_qty': '최종단수주수량',
|
||||
'last_alct_amt': '최종배정금액',
|
||||
'last_ftsk_chgs': '최종단수주대금',
|
||||
'rdpt_prca': '상환원금',
|
||||
'dlay_int_amt': '지연이자금액',
|
||||
'lstg_dt': '상장일자',
|
||||
'sbsc_end_dt': '청약종료일자',
|
||||
'cash_dfrm_dt': '현금지급일자',
|
||||
'rqst_qty': '신청수량',
|
||||
'rqst_amt': '신청금액',
|
||||
'rqst_dt': '신청일자',
|
||||
'rfnd_dt': '환불일자',
|
||||
'rfnd_amt': '환불금액',
|
||||
'lstg_stqt': '상장주수',
|
||||
'tax_amt': '세금금액',
|
||||
'sbsc_unpr': '청약단가'
|
||||
}
|
||||
|
||||
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()
|
||||
|
||||
# 기간별계좌권리현황조회
|
||||
logging.info("=== 기간별계좌권리현황조회 ===")
|
||||
try:
|
||||
result = period_rights(inqr_dvsn="03", cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, inqr_strt_dt="20250101",
|
||||
inqr_end_dt="20250103")
|
||||
except ValueError as e:
|
||||
logging.error("에러 발생: %s" % str(e))
|
||||
return
|
||||
|
||||
logging.info("사용 가능한 컬럼: %s", 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)
|
||||
|
||||
logging.info("결과:")
|
||||
print(result)
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
@@ -0,0 +1,140 @@
|
||||
"""
|
||||
Created on 20250601
|
||||
"""
|
||||
|
||||
|
||||
import sys
|
||||
import time
|
||||
from typing import Optional
|
||||
import logging
|
||||
|
||||
import pandas as pd
|
||||
|
||||
sys.path.extend(['../..', '.'])
|
||||
import kis_auth as ka
|
||||
|
||||
# 로깅 설정
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
##############################################################################################
|
||||
# [국내주식] 주문/계좌 > 기간별계좌권리현황조회 [국내주식-211]
|
||||
##############################################################################################
|
||||
|
||||
# 상수 정의
|
||||
API_URL = "/uapi/domestic-stock/v1/trading/period-rights"
|
||||
|
||||
def period_rights(
|
||||
inqr_dvsn: str, # [필수] 조회구분 (ex. 03)
|
||||
cano: str, # [필수] 종합계좌번호 (ex. 12345678)
|
||||
acnt_prdt_cd: str, # [필수] 계좌상품코드 (ex. 01 or 22)
|
||||
inqr_strt_dt: str, # [필수] 조회시작일자 (ex. 20250101)
|
||||
inqr_end_dt: str, # [필수] 조회종료일자 (ex. 20250103)
|
||||
cust_rncno25: str = "", # 고객실명확인번호25
|
||||
hmid: str = "", # 홈넷ID
|
||||
rght_type_cd: str = "", # 권리유형코드
|
||||
pdno: str = "", # 상품번호
|
||||
prdt_type_cd: str = "", # 상품유형코드
|
||||
NK100: str = "", # 연속조회키100
|
||||
FK100: str = "", # 연속조회검색조건100
|
||||
tr_cont: str = "", # 연속 거래 여부
|
||||
dataframe: Optional[pd.DataFrame] = None, # 누적 데이터프레임
|
||||
depth: int = 0, # 내부 재귀깊이 (자동관리)
|
||||
max_depth: int = 10 # 최대 재귀 횟수 제한
|
||||
) -> pd.DataFrame:
|
||||
"""
|
||||
기간별계좌권리현황조회 API입니다.
|
||||
한국투자 HTS(eFriend Plus) > [7344] 권리유형별 현황조회 화면을 API로 개발한 사항으로, 해당 화면을 참고하시면 기능을 이해하기 쉽습니다.
|
||||
|
||||
Args:
|
||||
inqr_dvsn (str): [필수] 조회구분 (ex. 03)
|
||||
cano (str): [필수] 종합계좌번호 (ex. 12345678)
|
||||
acnt_prdt_cd (str): [필수] 계좌상품코드 (ex. 01 or 22)
|
||||
inqr_strt_dt (str): [필수] 조회시작일자 (ex. 20250101)
|
||||
inqr_end_dt (str): [필수] 조회종료일자 (ex. 20250103)
|
||||
cust_rncno25 (str): 고객실명확인번호25
|
||||
hmid (str): 홈넷ID
|
||||
rght_type_cd (str): 권리유형코드
|
||||
pdno (str): 상품번호
|
||||
prdt_type_cd (str): 상품유형코드
|
||||
NK100 (str): 연속조회키100
|
||||
FK100 (str): 연속조회검색조건100
|
||||
tr_cont (str): 연속 거래 여부
|
||||
dataframe (Optional[pd.DataFrame]): 누적 데이터프레임
|
||||
depth (int): 내부 재귀깊이 (자동관리)
|
||||
max_depth (int): 최대 재귀 횟수 제한
|
||||
|
||||
Returns:
|
||||
pd.DataFrame: 기간별계좌권리현황 데이터
|
||||
|
||||
Example:
|
||||
>>> df = period_rights(inqr_dvsn="03", cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, inqr_strt_dt="20250101", inqr_end_dt="20250103")
|
||||
>>> print(df)
|
||||
"""
|
||||
|
||||
if inqr_dvsn == "":
|
||||
raise ValueError("inqr_dvsn is required (e.g. '03')")
|
||||
|
||||
if cano == "":
|
||||
raise ValueError("cano is required (e.g. '12345678')")
|
||||
|
||||
if acnt_prdt_cd == "":
|
||||
raise ValueError("acnt_prdt_cd is required (e.g. '01' or '22')")
|
||||
|
||||
if inqr_strt_dt == "":
|
||||
raise ValueError("inqr_strt_dt is required (e.g. '20250101')")
|
||||
|
||||
if inqr_end_dt == "":
|
||||
raise ValueError("inqr_end_dt is required (e.g. '20250103')")
|
||||
|
||||
if depth > max_depth:
|
||||
logging.warning("Max recursive depth reached.")
|
||||
if dataframe is None:
|
||||
return pd.DataFrame()
|
||||
else:
|
||||
return dataframe
|
||||
|
||||
tr_id = "CTRGA011R" # 기간별계좌권리현황조회
|
||||
|
||||
params = {
|
||||
"INQR_DVSN": inqr_dvsn,
|
||||
"CANO": cano,
|
||||
"ACNT_PRDT_CD": acnt_prdt_cd,
|
||||
"INQR_STRT_DT": inqr_strt_dt,
|
||||
"INQR_END_DT": inqr_end_dt,
|
||||
"CUST_RNCNO25": cust_rncno25,
|
||||
"HMID": hmid,
|
||||
"RGHT_TYPE_CD": rght_type_cd,
|
||||
"PDNO": pdno,
|
||||
"PRDT_TYPE_CD": prdt_type_cd,
|
||||
"CTX_AREA_NK100": NK100,
|
||||
"CTX_AREA_FK100": FK100
|
||||
}
|
||||
|
||||
res = ka._url_fetch(API_URL, tr_id, tr_cont, params)
|
||||
|
||||
if res.isOK():
|
||||
current_data = pd.DataFrame(res.getBody().output)
|
||||
|
||||
if dataframe is not None:
|
||||
dataframe = pd.concat([dataframe, current_data], ignore_index=True)
|
||||
else:
|
||||
dataframe = current_data
|
||||
|
||||
tr_cont = res.getHeader().tr_cont
|
||||
FK100 = res.getBody().ctx_area_fk100
|
||||
NK100 = res.getBody().ctx_area_nk100
|
||||
|
||||
if tr_cont in ["M", "F"]: # 다음 페이지 존재
|
||||
logging.info("Call Next page...")
|
||||
ka.smart_sleep() # 시스템 안정적 운영을 위한 지연
|
||||
return period_rights(
|
||||
inqr_dvsn, cano, acnt_prdt_cd, inqr_strt_dt, inqr_end_dt,
|
||||
cust_rncno25, hmid, rght_type_cd, pdno, prdt_type_cd,
|
||||
NK100, FK100, "N", dataframe, depth + 1, max_depth
|
||||
)
|
||||
else:
|
||||
logging.info("Data fetch complete.")
|
||||
return dataframe
|
||||
else:
|
||||
res.printError(url=API_URL)
|
||||
return pd.DataFrame()
|
||||
Reference in New Issue
Block a user