Finance Balance Sheet Generator

import requests, json

API_FINANCE = "https://api.finance.com/v1/balance"

def fetch_accounts():
    resp = requests.get(API_FINANCE)
    return resp.json().get("accounts", [])

def calculate_equity(assets, liabilities):
    return assets - liabilities

def generate_balance_sheet(accounts):
    assets = sum(a["amount"] for a in accounts if a["type"] == "asset")
    liabilities = sum(a["amount"] for a in accounts if a["type"] == "liability")
    equity = calculate_equity(assets, liabilities)
    wc = sum(a["amount"] for a in accounts if a["type"] == "current_asset") - sum(a["amount"] for a in accounts if a["type"] == "current_liability")
    return {"assets": assets, "liabilities": liabilities, "equity": equity, "working_capital": wc}

bs = generate_balance_sheet(fetch_accounts())
print(json.dumps(bs, indent=2))