whallets/texts_cli.py

"""A module with all the display info for whallets-cli."""
from tabulate import tabulate


def _welcome_message():
    """welcoming message to the cli"""
    msg = \
        "\n=========================================\n"\
        "WELCOME TO WHALLETS CLI"\
        "\n========================================="
    return msg

def _main_menu():
    """Displays the CLI main menu"""
    line_0 = ("**","Chose an operation from the menu:")
    line = ("---","---------------------------------",)
    line_1 = ("1 -","Add a wallet to the your list")
    line_2 = ("2 -","Remove a wallet from your list")
    line_3 = ("3 -","Display your wallet list")
    line_4 = ("4 -","Display recent whale TXs on your EVM list")
    line_5 = ("5 -","Display recent whale TXs on your SPL list")
    line_q = ("q -","Quit!")

    table = [
        line_0,
        line,
        line_1,
        line_2,
        line_3,
        line_4,
        line_5,
        line_q
    ]
    return table

def _menu_choice():
    """Sentence asking user for the next step"""
    msg = "\nEnter number and press Enter: "
    return msg

def _missing_operation():
    """Inform about non-existing users choice"""
    msg = "Sorry, but you had either choosen an unexisting option, "\
    "or this operation has not been developped yet."
    return msg

def _ask_new_choice():
    """Message offering a new choice or quit"""
    msg = \
        "\n\n=========================================\n"\
        "Do you have another choice?\n1 - YES\n2 - NO (quit)\n"
    return msg



def _choose_network():
    """Gives a choice of networks to work with"""
    line_0 = ("**","Please chose the network for the wallet address:")
    line = ("---", "---------------------------------",)
    line_1 = ("1 -","EVM; Etherum main net and any forks (BSC, Poly..)")
    line_2 = ("2 -","SPL; Solana Program Platform")
    line_3 = ("3 -","BTC; Bitcoin")
    table = [ line_0, line, line_1, line_2, line_3]
    return table

def _prompt_new_info(item):
    """Ask or the wallet address"""
    msg = f"\nEnter the wallet {item}:\n(if unknown press Enter)\n"
    return msg

def _ask_more_wallets():
    """ask user if an account has more wallet addresses"""
    msg_0 = \
        "\nDo you want to add another wallet to this username?" \
        "\n1 - YES (add more wallets)\n2 - NO (continue)\n"
    msg_1 = "\nEnter another wallet address:\n"

    return msg_0, msg_1



def _prompt_twitter():
    """Ask or the wallet address"""
    msg = "\nEnter the twitter link (full address):\n"
    return msg

def _prompt_info():
    """Ask or the wallet address"""
    msg = "\nEnter short info about the user:\n"
    return msg

def _prompt_name():
    """Ask or the wallet address"""
    msg = "\nEnter user/wallet name/tag:\n"
    return msg

def _prompt_ens():
    """Ask or the wallet address"""
    msg = "\nEnter user's ENS:\n"
    return msg


def _display_wallet_check_result(z):
    """displays answer based on added data"""
    msg_0 = \
        f"\nThis {z} already exists in your wallet dictionary."\
        f"\nDo you want to continue?\n1 - YES (keep it duplicate)"\
        f"\n2 - NO (change the item)"\
        f"{_menu_choice()}"

    msg_1 = \
        f"This is your the wallet info:\n(Press Enter to save, 'q' to cancell)"
    return (msg_0, msg_1)


def _enter_new_info(y):
    """Asks user for a correct information"""
    msg = f"\nPlease write a correct {y}:\n\n"
    return msg

def _save_wallet_confirm(addresses, new_wallet):
    """print wallet and ask user if to save it"""
    network = new_wallet["Network"]
    username = new_wallet["username"]
    twtr = new_wallet["twitter address"]
    ens = new_wallet["ENS"]
    info = new_wallet["info/note"]
    # addr = list(addresses)

    line_0 = ("****","This wallet will be saved to your dictionary:")
    line = ("-----","----------------------------------------------")
    line_1 = (f"Network:",f"{network}")
    line_2 = (f"Name:",f"{username}")
    line_3 = (f"Twitter:",f"{twtr}")
    line_4 = (f"ENS:",f"{ens}")
    line_5 = (f"Info:", f"{info}")


    table = [
        line_0,
        line,
        line_1,
        line_2,
        line_3,
        line_4,
        line_5,
        ]

    for i, address in enumerate(addresses):
        line = (f"Address_{i}:", f"{address}")
        table.append(line)
    return table

def _confirm_entry():
    """Asks user to confirm the wallet saving or repeat"""
    msg_0 = \
        "\nDo you want to save the wallet to the database?"\
        "\n1 - YES\n2 - NO\n"
    msg_1 = "\nThe wallet was saved to your database."
    msg_2 = \
        "\nDo you want to add another wallet?"\
        "\n1 - YES\n2 - NO\n"

    return msg_0, msg_1, msg_2


# def _display_wallets(chain, user):
#     """Text template for table of wallet display function"""
#     line_0 = ("*****",f"{chain.upper} WALLETS")
#     line = ("=====================================================",)
#     line_1 = ("Username:",f"{user}")


def _return_network(ntw_i):
    """Return network based on index"""
    if ntw_i == 0:
        ntw = "Cosmoss/EVM (ETH and forks)"
    elif ntw_i == 1:
        ntw = "Solana Program Library"
    elif ntw_i == 2:
        ntw = "Bitcoin"
    else:
        ntw = None

    return ntw