path: root/gamestonk_terminal/cryptocurrency/crypto_controller.py

"""Cryptocurrency Controller"""
__docformat__ = "numpy"
# pylint: disable=R0904, C0302, R1710, W0622

import argparse
import os
import matplotlib.pyplot as plt
import pandas as pd
from prompt_toolkit.completion import NestedCompleter
from binance.client import Client
from gamestonk_terminal import feature_flags as gtff
from gamestonk_terminal.helper_funcs import (
    get_flair,
    parse_known_args_and_warn,
    check_positive,
)
from gamestonk_terminal.menu import session
from gamestonk_terminal.cryptocurrency.technical_analysis import ta_controller
from gamestonk_terminal.cryptocurrency.overview import overview_controller
from gamestonk_terminal.cryptocurrency.due_diligence import (
    dd_controller,
    coinpaprika_view,
    binance_view,
    pycoingecko_view,
)
from gamestonk_terminal.cryptocurrency.discovery import (
    discovery_controller,
)
from gamestonk_terminal.cryptocurrency.due_diligence import (
    finbrain_crypto_view,
)
from gamestonk_terminal.cryptocurrency.due_diligence.finbrain_crypto_view import COINS

from gamestonk_terminal.cryptocurrency.cryptocurrency_helpers import (
    load,
    find,
    load_ta_data,
    plot_chart,
)
from gamestonk_terminal.cryptocurrency.report import report_controller
from gamestonk_terminal.cryptocurrency.due_diligence.binance_model import (
    show_available_pairs_for_given_symbol,
)
import gamestonk_terminal.config_terminal as cfg


class CryptoController:
    CHOICES = [
        "?",
        "cls",
        "help",
        "q",
        "quit",
    ]

    CHOICES_COMMAND = [
        "finbrain",
        "chart",
        "load",
        "find",
    ]

    CHOICES_MENUS = [
        "ta",
        "dd",
        "ov",
        "disc",
        "report",
    ]

    SOURCES = {
        "bin": "Binance",
        "cg": "CoinGecko",
        "cp": "CoinPaprika",
    }

    DD_VIEWS_MAPPING = {
        "cg": pycoingecko_view,
        "cp": coinpaprika_view,
        "bin": binance_view,
    }

    CHOICES += CHOICES_COMMAND
    CHOICES += CHOICES_MENUS

    def __init__(self):
        """CONSTRUCTOR"""

        self.crypto_parser = argparse.ArgumentParser(add_help=False, prog="crypto")
        self.crypto_parser.add_argument("cmd", choices=self.CHOICES)

        self.current_coin = ""
        self.current_df = pd.DataFrame()
        self.current_currency = ""
        self.source = ""

    def print_help(self):
        """Print help"""
        help_text = """https://github.com/GamestonkTerminal/GamestonkTerminal/tree/main/gamestonk_terminal/cryptocurrency

>> CRYPTO <<

What do you want to do?
    cls         clear screen
    ?/help      show this menu again
    q           quit this menu, and shows back to main menu
    quit        quit to abandon the program
"""
        help_text += (
            f"\nCoin: {self.current_coin}" if self.current_coin != "" else "\nCoin: ?"
        )
        help_text += (
            f"\nSource: {self.SOURCES.get(self.source, '?')}\n"
            if self.source != ""
            else "\nSource: ?\n"
        )
        help_text += """
Note: Some of CoinGecko commands can fail. Team is working on fix.

    load        load a specific cryptocurrency for analysis
    chart       view a candle chart for a specific cryptocurrency
    find        alternate way to search for coins
    finbrain    crypto sentiment from 15+ major news headlines

>   disc        discover trending cryptocurrencies,  e.g.: top gainers, losers, top sentiment
>   ov          overview of the cryptocurrencies,    e.g.: market cap, DeFi, latest news, top exchanges, stables
>   dd          due-diligence for loaded coin,       e.g.: coin information, social media, market stats
>   ta          technical analysis for loaded coin.  e.g.: ema, macd, rsi, adx, bbands, obv
>   report      generate automatic report
"""
        print(help_text)

    def switch(self, an_input: str):
        """Process and dispatch input

        Returns
        -------
        True, False or None
            False - quit the menu
            True - quit the program
            None - continue in the menu
        """

        # Empty command
        if not an_input:
            print("")
            return None

        (known_args, other_args) = self.crypto_parser.parse_known_args(an_input.split())

        # Help menu again
        if known_args.cmd == "?":
            self.print_help()
            return None

        # Clear screen
        if known_args.cmd == "cls":
            os.system("cls||clear")
            return None

        return getattr(
            self, "call_" + known_args.cmd, lambda: "Command not recognized!"
        )(other_args)

    def call_help(self, _):
        """Process Help command"""
        self.print_help()

    def call_q(self, _):
        """Process Q command - quit the menu"""
        return False

    def call_quit(self, _):
        """Process Quit command - quit the program"""
        return True

    def call_load(self, other_args):
        """Process load command"""
        try:
            parser = argparse.ArgumentParser(
                add_help=False,
                formatter_class=argparse.ArgumentDefaultsHelpFormatter,
                prog="load",
                description="Load crypto currency to perform analysis on. "
                "Available data sources are CoinGecko, CoinPaprika, and Binance"
                "By default main source used for analysis is CoinGecko (cg). To change it use --source flag",
            )

            parser.add_argument(
                "-c",
                "--coin",
                help="Coin to get",
                dest="coin",
                type=str,
                required="-h" not in other_args,
            )

            parser.add_argument(
                "-s",
                "--source",
                help="Source of data",
                dest="source",
                choices=("cp", "cg", "bin"),
                default="cg",
                required=False,