#!/usr/bin/env python3
"""Get Kraken Futures Demo balance"""
import os
import json
import sys

# Set keys from env or use defaults
API_KEY = os.environ.get('KRAKEN_DEMO_API_KEY', '+Jw3mx4DaVVca9otxolhxbxti0A+J/+uwZMJclM/mcU9zJWiCRflGW8W')
API_SECRET = os.environ.get('KRAKEN_DEMO_API_SECRET', 'zfH9shvbwfMXfdo4ivuh1qxecLzKmeEI+f1jTR2bZv69hxXnkYML5CbKSrkSWOSbJzPub55VXUzvzVejlhcCTcPs')

try:
    from kraken.futures import User

    user = User(key=API_KEY, secret=API_SECRET, sandbox=True)
    accounts = user.get_wallets()

    if accounts.get('result') == 'success':
        flex = accounts.get('accounts', {}).get('flex', {})
        result = {
            'success': True,
            'portfolioValue': flex.get('portfolioValue', 0),
            'availableMargin': flex.get('availableMargin', 0),
            'collateralValue': flex.get('collateralValue', 0),
            'pnl': flex.get('pnl', 0),
            'currencies': flex.get('currencies', {})
        }
    else:
        result = {'success': False, 'error': accounts.get('error', 'Unknown error')}

    print(json.dumps(result))

except Exception as e:
    print(json.dumps({'success': False, 'error': str(e)}))
    sys.exit(1)