initial commit

한국투자증권(API)/examples_llm/domestic_stock/inquire_psbl_order/chk_inquire_psbl_order.py

@@ -0,0 +1,84 @@
+"""
+Created on 20250601 
+"""
+
+import sys
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+from inquire_psbl_order import inquire_psbl_order
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+
+##############################################################################################
+# [국내주식] 주문/계좌 > 매수가능조회[v1_국내주식-007]
+##############################################################################################
+
+COLUMN_MAPPING = {
+    'ord_psbl_cash': '주문가능현금',
+    'ord_psbl_sbst': '주문가능대용',
+    'ruse_psbl_amt': '재사용가능금액',
+    'fund_rpch_chgs': '펀드환매대금',
+    'psbl_qty_calc_unpr': '가능수량계산단가',
+    'nrcvb_buy_amt': '미수없는매수금액',
+    'nrcvb_buy_qty': '미수없는매수수량',
+    'max_buy_amt': '최대매수금액',
+    'max_buy_qty': '최대매수수량',
+    'cma_evlu_amt': 'CMA평가금액',
+    'ovrs_re_use_amt_wcrc': '해외재사용금액원화',
+    'ord_psbl_frcr_amt_wcrc': '주문가능외화금액원화'
+}
+
+NUMERIC_COLUMNS = []
+
+def main():
+    """
+    매수가능조회 테스트 함수
+    
+    이 함수는 매수가능조회 API를 호출하여 결과를 출력합니다.
+    테스트 데이터로 삼성전자(005930)를 사용합니다.
+    
+    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()
+
+    # case1 조회
+    logging.info("=== case1 조회 ===")
+    try:
+        result = inquire_psbl_order(env_dv="real",cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, pdno="005930", ord_unpr="55000",
+                                    ord_dvsn="01", cma_evlu_amt_icld_yn="N", ovrs_icld_yn="N")
+    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()

한국투자증권(API)/examples_llm/domestic_stock/inquire_psbl_order/inquire_psbl_order.py

@@ -0,0 +1,118 @@
+"""
+Created on 20250601 
+"""
+
+
+import sys
+import logging
+
+import pandas as pd
+
+sys.path.extend(['../..', '.'])
+import kis_auth as ka
+
+# 로깅 설정
+logging.basicConfig(level=logging.INFO)
+
+##############################################################################################
+# [국내주식] 주문/계좌 > 매수가능조회[v1_국내주식-007]
+##############################################################################################
+
+# 상수 정의
+API_URL = "/uapi/domestic-stock/v1/trading/inquire-psbl-order"
+
+def inquire_psbl_order(
+    env_dv: str,  # 실전모의구분
+    cano: str,    # 종합계좌번호
+    acnt_prdt_cd: str,  # 계좌상품코드
+    pdno: str,    # 상품번호
+    ord_unpr: str,  # 주문단가
+    ord_dvsn: str,  # 주문구분
+    cma_evlu_amt_icld_yn: str,  # CMA평가금액포함여부
+    ovrs_icld_yn: str  # 해외포함여부
+) -> pd.DataFrame:
+    """
+    매수가능 조회 API입니다. 
+    실전계좌/모의계좌의 경우, 한 번의 호출에 최대 1건까지 확인 가능합니다.
+
+    1) 매수가능금액 확인
+    . 미수 사용 X: nrcvb_buy_amt(미수없는매수금액) 확인
+    . 미수 사용 O: max_buy_amt(최대매수금액) 확인
+
+    2) 매수가능수량 확인
+    . 특정 종목 전량매수 시 가능수량을 확인하실 경우 ORD_DVSN:00(지정가)는 종목증거금율이 반영되지 않습니다. 
+    따라서 "반드시" ORD_DVSN:01(시장가)로 지정하여 종목증거금율이 반영된 가능수량을 확인하시기 바랍니다. 
+
+    (다만, 조건부지정가 등 특정 주문구분(ex.IOC)으로 주문 시 가능수량을 확인할 경우 주문 시와 동일한 주문구분(ex.IOC) 입력하여 가능수량 확인)
+
+    . 미수 사용 X: ORD_DVSN:01(시장가) or 특정 주문구분(ex.IOC)로 지정하여 nrcvb_buy_qty(미수없는매수수량) 확인
+    . 미수 사용 O: ORD_DVSN:01(시장가) or 특정 주문구분(ex.IOC)로 지정하여 max_buy_qty(최대매수수량) 확인
+    
+    Args:
+        env_dv (str): [필수] 실전모의구분 (ex. real:실전, demo:모의)
+        cano (str): [필수] 종합계좌번호 (ex. 계좌번호 체계(8-2)의 앞 8자리)
+        acnt_prdt_cd (str): [필수] 계좌상품코드 (ex. 계좌번호 체계(8-2)의 뒤 2자리)
+        pdno (str): [필수] 상품번호 (ex. 종목번호(6자리))
+        ord_unpr (str): [필수] 주문단가 (ex. 1주당 가격)
+        ord_dvsn (str): [필수] 주문구분 (ex. 01 : 시장가)
+        cma_evlu_amt_icld_yn (str): [필수] CMA평가금액포함여부 (ex. Y)
+        ovrs_icld_yn (str): [필수] 해외포함여부 (ex. N)
+
+    Returns:
+        pd.DataFrame: 매수가능조회 데이터
+        
+    Example:
+        >>> df = inquire_psbl_order(env_dv="real", cano=trenv.my_acct, acnt_prdt_cd=trenv.my_prod, pdno="005930", ord_unpr="55000", ord_dvsn="01", cma_evlu_amt_icld_yn="N", ovrs_icld_yn="N")
+        >>> print(df)
+    """
+
+    # 필수 파라미터 검증
+    if env_dv == "":
+        raise ValueError("env_dv is required (e.g. 'real:실전, demo:모의')")
+    
+    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 pdno == "":
+        raise ValueError("pdno is required (e.g. '종목번호(6자리)')")
+    
+    if ord_unpr == "":
+        raise ValueError("ord_unpr is required (e.g. '1주당 가격')")
+    
+    if ord_dvsn == "":
+        raise ValueError("ord_dvsn is required (e.g. '01 : 시장가')")
+    
+    if cma_evlu_amt_icld_yn == "":
+        raise ValueError("cma_evlu_amt_icld_yn is required (e.g. 'Y')")
+    
+    if ovrs_icld_yn == "":
+        raise ValueError("ovrs_icld_yn is required (e.g. 'N')")
+
+    # tr_id 설정
+    if env_dv == "real":
+        tr_id = "TTTC8908R"
+    elif env_dv == "demo":
+        tr_id = "VTTC8908R"
+    else:
+        raise ValueError("env_dv can only be 'real' or 'demo'")
+
+    params = {
+        "CANO": cano,
+        "ACNT_PRDT_CD": acnt_prdt_cd,
+        "PDNO": pdno,
+        "ORD_UNPR": ord_unpr,
+        "ORD_DVSN": ord_dvsn,
+        "CMA_EVLU_AMT_ICLD_YN": cma_evlu_amt_icld_yn,
+        "OVRS_ICLD_YN": ovrs_icld_yn
+    }
+    
+    res = ka._url_fetch(API_URL, tr_id, "", params)
+    
+    if res.isOK():
+        return pd.DataFrame(res.getBody().output, index=[0])
+    else:
+        res.printError(url=API_URL)
+        return pd.DataFrame()