get_balance.py

"""기본 잔고 조회 예제.

config.yaml의 인증 정보를 사용해 계좌 잔고를 조회합니다.
"""
import yaml
from pykis import PyKis, KisAuth


def load_config(path: str = "config.yaml", profile: str | None = None) -> dict:
    import os

    profile = profile or os.environ.get("PYKIS_PROFILE")
    with open(path, "r", encoding="utf-8") as f:
        cfg = yaml.safe_load(f)

    if isinstance(cfg, dict) and "configs" in cfg:
        sel = profile or cfg.get("default") or "virtual"
        selected = cfg["configs"].get(sel)
        if not selected:
            raise ValueError(f"Profile '{sel}' not found in {path}")
        return selected

    return cfg


def main() -> None:
    import argparse

    parser = argparse.ArgumentParser()
    parser.add_argument("--config", default="config.yaml", help="path to config file")
    parser.add_argument("--profile", help="config profile name (virtual|real)")
    args = parser.parse_args()

    cfg = load_config(path=args.config, profile=args.profile)

    auth = KisAuth(
        id=cfg["id"],
        account=cfg["account"],
        appkey=cfg["appkey"],
        secretkey=cfg["secretkey"],
        virtual=cfg.get("virtual", False),
    )

    kis = PyKis(auth, keep_token=True)

    account = kis.account()
    balance = account.balance()
    print(balance)


if __name__ == "__main__":
    main()