#!/usr/bin/env python3

import sys, argparse, os
from os.path import join, dirname
from shutil import copyfile
from dotenv import load_dotenv
from lib.balancelib import Balance

# Get variables environment
if not os.path.isfile('.env'):
    copyfile(".env.template", ".env")
dotenv_path = join(dirname(__file__), '.env')
load_dotenv(dotenv_path)

dunikey = os.getenv('DUNIKEY')
node = os.getenv('NODE')
if not node:
    sys.stderr.write("Please fill a Duniter node in .env file\n")
    sys.exit(1)

# Parse arguments
parser = argparse.ArgumentParser()
parser.add_argument('-p', '--pubkey', help="Clé publique du compte visé")
parser.add_argument('--mempool', action='store_true', help="Utilise les sources en Mempool")
args = parser.parse_args()

# Create transaction and send it
balance = Balance(dunikey, node, args.pubkey, args.mempool)
result = balance.balance()
print(result)