import click import db import wownero import commands @click.group() def cli(): pass @click.command() def debug(): pass @click.command() def get_users(): users = db.User.select() wallet = wownero.Wallet() for u in users: click.echo(f'{u.telegram_user} ({u.telegram_id}) - wallet {u.account_index}') @click.command() @click.argument('account_index') def get_address(account_index): address = wownero.Wallet().addresses(account=int(account_index))[0] click.echo(address) @click.command() def get_balances(): wallet = wownero.Wallet() accounts = wallet.accounts() for acc in accounts: balances = wallet.balances(account=acc) click.echo(f'wallet {acc} - {float(balances[0])} locked, {float(balances[1])} unlocked') @click.command() def generate_bot_help(): for cmd in commands.all_commands: c = commands.all_commands[cmd] if not 'admin' in c: click.echo(f'{cmd} - {commands.all_commands[cmd]["help"]}') cli.add_command(debug) cli.add_command(get_users) cli.add_command(get_address) cli.add_command(generate_bot_help) cli.add_command(get_balances) if __name__ == '__main__': cli()