OpenBB/gamestonk_terminal/technical_analysis/ta_menu.py

import argparse

import matplotlib.pyplot as plt
from gamestonk_terminal import feature_flags as gtff
from gamestonk_terminal.helper_funcs import get_flair
from gamestonk_terminal.menu import session
from gamestonk_terminal.technical_analysis import momentum as ta_momentum
from gamestonk_terminal.technical_analysis import overlap as ta_overlap
from gamestonk_terminal.technical_analysis import trend_indicators as ta_trend
from gamestonk_terminal.technical_analysis import volatility as ta_volatility
from gamestonk_terminal.technical_analysis import volume as ta_volume
from prompt_toolkit.completion import NestedCompleter


def print_technical_analysis(s_ticker, s_start, s_interval):
    """ Print help """

    s_intraday = (f"Intraday {s_interval}", "Daily")[s_interval == "1440min"]

    if s_start:
        print(f"\n{s_intraday} Stock: {s_ticker} (from {s_start.strftime('%Y-%m-%d')})")
    else:
        print(f"\n{s_intraday} Stock: {s_ticker}")

    print("\nTechnical Analysis:")  # https://github.com/twopirllc/pandas-ta
    print("   help        show this technical analysis menu again")
    print("   q           quit this menu, and shows back to main menu")
    print("   quit        quit to abandon program")
    print("overlap:")
    print("   ema         exponential moving average")
    print("   sma         simple moving average")
    print("   vwap        volume weighted average price")
    print("momentum:")
    print("   cci         commodity channel index")
    print("   macd        moving average convergence/divergence")
    print("   rsi         relative strength index")
    print("   stoch       stochastic oscillator")
    print("trend:")
    print("   adx         average directional movement index")
    print("   aroon       aroon indicator")
    print("volatility:")
    print("   bbands      bollinger bands")
    print("volume:")
    print("   ad          chaikin accumulation/distribution line values")
    print("   obv         on balance volume")
    print("")


def ta_menu(df_stock, s_ticker, s_start, s_interval):

    # Add list of arguments that the technical analysis parser accepts
    ta_parser = argparse.ArgumentParser(prog="ta", add_help=False)
    choices = [
        "help",
        "q",
        "quit",
        "ema",
        "sma",
        "vwap",
        "cci",
        "macd",
        "rsi",
        "stoch",
        "adx",
        "aroon",
        "bbands",
        "ad",
        "obv",
    ]
    ta_parser.add_argument("cmd", choices=choices)
    completer = NestedCompleter.from_nested_dict({c: None for c in choices})

    print_technical_analysis(s_ticker, s_start, s_interval)

    # Loop forever and ever
    while True:
        # Get input command from user
        if session and gtff.USE_PROMPT_TOOLKIT:
            as_input = session.prompt(
                f"{get_flair()} (ta)> ",
                completer=completer,
            )
        else:
            as_input = input(f"{get_flair()} (ta)> ")

        # Images are non blocking - allows to close them if we type other command
        plt.close()

        # Parse fundamental analysis command of the list of possible commands
        try:
            (ns_known_args, l_args) = ta_parser.parse_known_args(as_input.split())

        except SystemExit:
            print("The command selected doesn't exist\n")
            continue

        if ns_known_args.cmd == "help":
            print_technical_analysis(s_ticker, s_start, s_interval)

        elif ns_known_args.cmd == "q":
            # Just leave the FA menu
            return False

        elif ns_known_args.cmd == "quit":
            # Abandon the program
            return True

        # OVERLAP
        elif ns_known_args.cmd == "ema":
            ta_overlap.ema(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "sma":
            ta_overlap.sma(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "vwap":
            ta_overlap.vwap(l_args, s_ticker, s_interval, df_stock)

        # MOMENTUM
        elif ns_known_args.cmd == "cci":
            ta_momentum.cci(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "macd":
            ta_momentum.macd(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "rsi":
            ta_momentum.rsi(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "stoch":
            ta_momentum.stoch(l_args, s_ticker, s_interval, df_stock)

        # TREND
        elif ns_known_args.cmd == "adx":
            ta_trend.adx(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "aroon":
            ta_trend.aroon(l_args, s_ticker, s_interval, df_stock)

        # VOLATILITY
        elif ns_known_args.cmd == "bbands":
            ta_volatility.bbands(l_args, s_ticker, s_interval, df_stock)

        # VOLUME
        elif ns_known_args.cmd == "ad":
            ta_volume.ad(l_args, s_ticker, s_interval, df_stock)

        elif ns_known_args.cmd == "obv":
            ta_volume.obv(l_args, s_ticker, s_interval, df_stock)

        else:
            print("Command not recognized!")