#!/usr/bin/env pyston3

import argparse
import logging
import os
import sys

sys.path.insert(0, os.path.abspath(
    os.path.join(os.path.dirname(__file__), "..")))
import mango  # nopep8

parser = argparse.ArgumentParser(description="Shows all Wrapped SOL accounts for the wallet.")
mango.Context.add_command_line_parameters(parser)
mango.Wallet.add_command_line_parameters(parser)
parser.add_argument("--symbol", type=str, required=True,
                    help="symbol of the token to look up, e.g. 'ETH'")
args = parser.parse_args()

logging.getLogger().setLevel(args.log_level)
logging.warning(mango.WARNING_DISCLAIMER_TEXT)

context = mango.Context.from_command_line_parameters(args)
wallet = mango.Wallet.from_command_line_parameters_or_raise(args)

token = context.token_lookup.find_by_symbol_or_raise(args.symbol)
token_accounts = mango.TokenAccount.fetch_all_for_owner_and_token(context, wallet.address, token)

if len(token_accounts) == 0:
    print("No wrapped SOL accounts.")
else:
    print(f"{token.name}:")
    for account in token_accounts:
        print(f"    {account.address}: {account.value}")