cli.py

import click
from defy.Utilities import Utilities
from defy.Wallet import Wallet
from defy.PriceFinder import PriceFinder
from defy.exchanges.Binance import Binance
from defy.exchanges.BinanceFutures import BinanceFutures
from defy.platforms.ValueDefi import ValueDefi
from defy.platforms.Fulcrum import Fulcrum
from defy.platforms.AutoFarm import AutoFarm

__author__ = "Parin Kobboon"


priceFinder = PriceFinder()

defyWallet = Wallet(priceFinder)
defyBinance = Binance(priceFinder)
defyBinanceFutures = BinanceFutures(priceFinder)
defyValueDefi = ValueDefi(priceFinder)
defyFulcrum = Fulcrum(priceFinder)
defyAutoFarm = AutoFarm(priceFinder)

CONTEXT_SETTINGS = dict(help_option_names=["-h", "--help"])


@click.group(context_settings=CONTEXT_SETTINGS)
def cli():
    pass


@cli.command(
    "all", short_help="Lookup to all available wallet/platforms/exchanges balance"
)
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def all(address, hideSmallBal):
    walletBal = defyWallet.getWallet(address, hideSmallBal)
    binanceBal = defyBinance.getWallet(hideSmallBal)
    binanceFuturesBal = defyBinanceFutures.getWallet(hideSmallBal)
    valueDefiBal = defyValueDefi.getWallet(address, hideSmallBal)
    fulcrumBal = defyFulcrum.getWallet(address, hideSmallBal)
    autoFarmBal = defyAutoFarm.getWallet(address, hideSmallBal)

    total = Utilities.getTotal(
        [
            walletBal,
            binanceBal,
            binanceFuturesBal,
            valueDefiBal,
            fulcrumBal,
            autoFarmBal,
        ]
    )

    defyWallet.displayWallet(walletBal)
    defyBinance.displayWallet(binanceBal)
    defyBinanceFutures.displayWallet(binanceFuturesBal)
    defyAutoFarm.displayWallet(autoFarmBal)
    defyFulcrum.displayWallet(fulcrumBal)
    defyValueDefi.displayWallet(valueDefiBal)

    Utilities.displayTotal(total)


@cli.command("wallet", short_help="Lookup to wallet balance")
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def wallet(address, hideSmallBal):
    walletBal = defyWallet.getWallet(address, hideSmallBal)

    total = Utilities.getTotal([walletBal])

    defyWallet.displayWallet(walletBal)

    Utilities.displayTotal(total)


@cli.command("platform", short_help="Lookup to platforms balance")
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def platform(address, hideSmallBal):
    autoFarmBal = defyAutoFarm.getWallet(address, hideSmallBal)
    fulcrumBal = defyFulcrum.getWallet(address, hideSmallBal)
    valueDefiBal = defyValueDefi.getWallet(address, hideSmallBal)

    total = Utilities.getTotal([valueDefiBal, fulcrumBal, autoFarmBal])

    defyAutoFarm.displayWallet(autoFarmBal)
    defyFulcrum.displayWallet(fulcrumBal)
    defyValueDefi.displayWallet(valueDefiBal)

    Utilities.displayTotal(total)


@cli.command("valuedefi", short_help="Lookup to ValueDefi balance")
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def valuedefi(address, hideSmallBal):
    valueDefiBal = defyValueDefi.getWallet(address, hideSmallBal)

    total = Utilities.getTotal([valueDefiBal])

    defyValueDefi.displayWallet(valueDefiBal)

    Utilities.displayTotal(total)


@cli.command("fulcrum", short_help="Lookup to Fulcrum balance")
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def fulcrum(address, hideSmallBal):
    fulcrumBal = defyFulcrum.getWallet(address, hideSmallBal)

    total = Utilities.getTotal([fulcrumBal])

    defyFulcrum.displayWallet(fulcrumBal)

    Utilities.displayTotal(total)


@cli.command("autofarm", short_help="Lookup to Fulcrum balance")
@click.argument("address")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def autofarm(address, hideSmallBal):
    autoFarmBal = defyAutoFarm.getWallet(address, hideSmallBal)

    total = Utilities.getTotal([autoFarmBal])

    defyAutoFarm.displayWallet(autoFarmBal)

    Utilities.displayTotal(total)


@cli.command("exchange", short_help="Lookup to exchanges balance")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def exchange(hideSmallBal):
    binanceBal = defyBinance.getWallet(hideSmallBal)
    binanceFuturesBal = defyBinanceFutures.getWallet(hideSmallBal)

    total = Utilities.getTotal([binanceBal, binanceFuturesBal])

    defyBinance.displayWallet(binanceBal)
    defyBinanceFutures.displayWallet(binanceFuturesBal)

    Utilities.displayTotal(total)


@cli.command("binance", short_help="Lookup to exchanges balance")
@click.option(
    "-hsb",
    "--hide-small-bal",
    "hideSmallBal",
    is_flag=True,
    default=False,
    help="`True` to hide small balance in wallet, default=false",
)
def binance(hideSmallBal):
    binanceBal = defyBinance.getWallet(hideSmallBal)
    binanceFuturesBal = defyBinanceFutures.getWallet(hideSmallBal)

    total = Utilities.getTotal([binanceBal, binanceFuturesBal])

    defyBinance.displayWallet(binanceBal)
    defyBinanceFutures.displayWallet(binanceFuturesBal)

    Utilities.displayTotal(total)