import os
import finestock
from finestock import APIProvider

try:
    # 선택 의존성: 설치돼 있으면 .env 파일을 자동으로 읽어 os.environ에 채워준다.
    # pip install python-dotenv
    from dotenv import load_dotenv
    load_dotenv()
except ImportError:
    pass

finestock.print_version_info()

# 실키/토큰은 절대 코드에 하드코딩하지 말 것. 환경변수로 주입한다.
# 예 (PowerShell): $env:APP_KEY="..."; $env:APP_SECRET="..."
app_key = os.environ.get("APP_KEY", "YOUR_APP_KEY")
app_secret = os.environ.get("APP_SECRET", "YOUR_APP_SECRET")
access_token = os.environ.get("ACCESS_TOKEN", "")

# 모의투자로 테스트하려면 APIProvider.NHV를 사용한다. 단, 접근토큰발급(oauth())은
# NH·NHV 모두 항상 운영 도메인에서만 발급된다(모의투자 계좌로 거래하더라도 토큰은 동일).
api = finestock.APIFactory.create_api(APIProvider.NH)
api.set_oauth_info(app_key, app_secret)

login = False
if login:
    api.oauth()
    print(api.access_token)
else:
    api.headers['authorization'] = f"Bearer {access_token}"

# 1) 계좌 목록 조회 — /n2/acctinfo. acct_type이 01/02면 운영(NH) 전용, 03이면 모의투자(NHV) 전용.
accounts = api.get_account_list()
print(accounts)

# 계좌목록에서 사용할 계좌를 골라 세팅한다. act_no는 11자리 단일 값이라 sub는 비워둔다.
account_num = os.environ.get("ACCOUNT_NUM", "YOUR_ACCOUNT_NUM")
#account_num_sub = os.environ.get("ACCOUNT_NUM_SUB", "YOUR_ACCOUNT_NUM")
api.set_account_info(account_num, "")

'''
# 2) 시세 조회
price = api.get_price("005930")
print(price)

ohlcvs = api.get_ohlcv("005930", "20260701", "20260821")
print(len(ohlcvs), ohlcvs[:3] if ohlcvs else ohlcvs)

orderbook = api.get_orderbook("005930")
print(orderbook)
'''
# 3) 계좌/잔고 조회
acc = api.get_balance()
print(acc)
if acc:
    print(len(acc.hold), acc.hold[:3] if acc.hold else acc.hold)

# 4) 주문 (실제 체결되니 운영 도메인에서는 주의 — 테스트는 APIProvider.NHV 권장)
# res = api.do_order("005930", finestock.ORDER_FLAG.BUY, 0, 1)  # price=0 -> 시장가
# print(res)
# if res:
#     cancel = api.do_order_cancel(res.order_num, "005930", 0)  # qty<=0 -> 전량취소
#     print(cancel)