import requests
import base58
import base64
from pprint import pprint
from tronpy import Tron
from tronpy.keys import PrivateKey
import requests
import json

#https://betterprogramming.pub/how-to-send-tron-cryptocurrency-with-python-5d842bc8f9cd

def get_balance_tronscan_usdt(address):
  url = "https://apilist.tronscan.org/api/account"
  payload = {
      "address": address,
  }
  res = requests.get(url, params=payload).json()
  trc20token_balances = res["trc20token_balances"]

  token_balance = next((item for item in trc20token_balances if item["tokenAbbr"] == 'USDT'), None)
  #print(token_balance)
  if token_balance == None:
    return 0
  else:
    return int(token_balance["balance"])/1000000

def get_balance_tronscan_tron(address):
  url = "https://apilist.tronscan.org/api/account"
  payload = {
      "address": address,
  }
  res = requests.get(url, params=payload).json()
  tron_balances = int(res["balance"])/1000000
  return tron_balances

  # token_balance = next((item for item in trc20token_balances if item["tokenAbbr"] == 'USDT'), None)
  # #print(token_balance)
  # if token_balance == None:
  #   return 0
  # else:

#
# bl = get_balance_tronscan('TFTYGZTWXx8KTf1k5qeD9Dh9ckLQuWVnhA')
# print('GEN WALLET', bl)
wallets = ['TLJjdgJMGDrVn22a37MZAiH3L9YPe3hYkc',
           'TX4RsY9T7YzrXjHgH9bPXEfKHngx2P7W5E',
           'TNgwTtrDWEGM71ehRAbEvsoYMwFctEW6oM',
           'TNTt88d4Nr4PZTtFm5ZMVKqJxPpSp5bhWd',
           'TFTYGZTWXx8KTf1k5qeD9Dh9ckLQuWVnhA']

for i in wallets:
    print(f'------------{i}------------')
    bl = get_balance_tronscan_tron(i)
    print('Ballance TRON', bl)
    bl = get_balance_tronscan_usdt(i)
    print('Ballance USDT', bl)
