# check_balance.py
import os
import sys
from decimal import Decimal

from dotenv import load_dotenv
from t_tech.invest import Client
from t_tech.invest.utils import quotation_to_decimal


def dec_money(mv) -> Decimal:
    return quotation_to_decimal(mv) if mv is not None else Decimal(0)


load_dotenv()
token = os.environ.get("TINKOFF_TOKEN")
account_id = os.environ.get("TINKOFF_ACCOUNT_ID")

if not token:
    print("ERROR: TINKOFF_TOKEN is not set")
    sys.exit(2)

with Client(token) as client:
    # Если account_id не задан, берём первый реальный счёт
    if not account_id:
        accs = client.users.get_accounts().accounts
        if not accs:
            print(
                "Нет реальных счетов. Проверь права токена: accounts:read и наличие счёта."
            )
            sys.exit(3)
        account_id = accs[0].id

    pos = client.operations.get_positions(account_id=account_id)

    rub_money = next((m for m in pos.money if m.currency.lower() == "rub"), None)
    rub_blocked = next((m for m in pos.blocked if m.currency.lower() == "rub"), None)

    rub_free = dec_money(rub_money) if rub_money else Decimal(0)
    rub_frozen = dec_money(rub_blocked) if rub_blocked else Decimal(0)
    rub_available = rub_free - rub_frozen

    print("ACCOUNT_ID:", account_id)
    print(f"RUB free:      {rub_free:.2f}")
    print(f"RUB blocked:   {rub_frozen:.2f}")
    print(f"RUB available: {rub_available:.2f}")

    # Дополнительно: оценка портфеля (если нужно)
    pf = client.operations.get_portfolio(account_id=account_id)
    total_currencies = (
        dec_money(pf.total_amount_currencies)
        if hasattr(pf, "total_amount_currencies")
        else None
    )
    if total_currencies is not None:
        print(f"Portfolio total_amount_currencies: {total_currencies:.2f}")