"""Feature Flags Controller Module""" __docformat__ = "numpy" # IMPORTATION STANDARD import logging from typing import List # IMPORTATION THIRDPARTY from dotenv import set_key # IMPORTATION INTERNAL from openbb_terminal import feature_flags as obbff from openbb_terminal.core.config.paths import USER_ENV_FILE from openbb_terminal.custom_prompt_toolkit import NestedCompleter from openbb_terminal.decorators import log_start_end from openbb_terminal.menu import session from openbb_terminal.parent_classes import BaseController from openbb_terminal.rich_config import console, MenuText # pylint: disable=too-many-lines,no-member,too-many-public-methods,C0302 # pylint:disable=import-outside-toplevel logger = logging.getLogger(__name__) class FeatureFlagsController(BaseController): """Feature Flags Controller class""" CHOICES_COMMANDS: List[str] = [ "retryload", "tab", "cls", "color", "ion", "watermark", "cmdloc", "promptkit", "thoughts", "reporthtml", "exithelp", "rcontext", "rich", "richpanel", "tbhint", "overwrite", ] PATH = "/featflags/" def __init__(self, queue: List[str] = None): """Constructor""" super().__init__(queue) if session and obbff.USE_PROMPT_TOOLKIT: choices: dict = {c: {} for c in self.controller_choices} self.completer = NestedCompleter.from_nested_dict(choices) def print_help(self): """Print help""" mt = MenuText("featflags/") mt.add_info("_info_") mt.add_raw("\n") mt.add_setting("retryload", obbff.RETRY_WITH_LOAD) mt.add_setting("tab", obbff.USE_TABULATE_DF) mt.add_setting("cls", obbff.USE_CLEAR_AFTER_CMD) mt.add_setting("color", obbff.USE_COLOR) mt.add_setting("promptkit", obbff.USE_PROMPT_TOOLKIT) mt.add_setting("thoughts", obbff.ENABLE_THOUGHTS_DAY) mt.add_setting("reporthtml", obbff.OPEN_REPORT_AS_HTML) mt.add_setting("exithelp", obbff.ENABLE_EXIT_AUTO_HELP) mt.add_setting("rcontext", obbff.REMEMBER_CONTEXTS) mt.add_setting("rich", obbff.ENABLE_RICH) mt.add_setting("richpanel", obbff.ENABLE_RICH_PANEL) mt.add_setting("ion", obbff.USE_ION) mt.add_setting("watermark", obbff.USE_WATERMARK) mt.add_setting("cmdloc", obbff.USE_CMD_LOCATION_FIGURE) mt.add_setting("tbhint", obbff.TOOLBAR_HINT) mt.add_setting("overwrite", obbff.FILE_OVERWITE) console.print(text=mt.menu_text, menu="Feature Flags") def call_overwrite(self, _): """Process overwrite command""" obbff.FILE_OVERWITE = not obbff.FILE_OVERWITE set_key(USER_ENV_FILE, "OPENBB_FILE_OVERWITE", str(obbff.FILE_OVERWITE)) def call_retryload(self, _): """Process retryload command""" obbff.RETRY_WITH_LOAD = not obbff.RETRY_WITH_LOAD set_key(USER_ENV_FILE, "OPENBB_RETRY_WITH_LOAD", str(obbff.RETRY_WITH_LOAD)) @log_start_end(log=logger) def call_tab(self, _): """Process tab command""" obbff.USE_TABULATE_DF = not obbff.USE_TABULATE_DF set_key(USER_ENV_FILE, "OPENBB_USE_TABULATE_DF", str(obbff.USE_TABULATE_DF)) @log_start_end(log=logger) def call_cls(self, _): """Process cls command""" obbff.USE_CLEAR_AFTER_CMD = not obbff.USE_CLEAR_AFTER_CMD set_key( USER_ENV_FILE, "OPENBB_USE_CLEAR_AFTER_CMD", str(obbff.USE_CLEAR_AFTER_CMD), ) @log_start_end(log=logger) def call_color(self, _): """Process color command""" obbff.USE_COLOR = not obbff.USE_COLOR set_key(USER_ENV_FILE, "OPENBB_USE_COLOR", str(obbff.USE_COLOR)) @log_start_end(log=logger) def call_promptkit(self, _): """Process promptkit command""" obbff.USE_PROMPT_TOOLKIT = not obbff.USE_PROMPT_TOOLKIT set_key( USER_ENV_FILE, "OPENBB_USE_PROMPT_TOOLKIT", str(obbff.USE_PROMPT_TOOLKIT), ) @log_start_end(log=logger) def call_thoughts(self, _): """Process thoughts command""" obbff.ENABLE_THOUGHTS_DAY = not obbff.ENABLE_THOUGHTS_DAY set_key( USER_ENV_FILE, "OPENBB_ENABLE_THOUGHTS_DAY", str(obbff.ENABLE_THOUGHTS_DAY), ) @log_start_end(log=logger) def call_reporthtml(self, _): """Process reporthtml command""" obbff.OPEN_REPORT_AS_HTML = not obbff.OPEN_REPORT_AS_HTML set_key( USER_ENV_FILE, "OPENBB_OPEN_REPORT_AS_HTML", str(obbff.OPEN_REPORT_AS_HTML), ) @log_start_end(log=logger) def call_exithelp(self, _): """Process exithelp command""" obbff.ENABLE_EXIT_AUTO_HELP = not obbff.ENABLE_EXIT_AUTO_HELP set_key( USER_ENV_FILE, "OPENBB_ENABLE_EXIT_AUTO_HELP", str(obbff.ENABLE_EXIT_AUTO_HELP), ) @log_start_end(log=logger) def call_rcontext(self, _): """Process rcontext command""" obbff.REMEMBER_CONTEXTS = not obbff.REMEMBER_CONTEXTS set_key( USER_ENV_FILE, "OPENBB_REMEMBER_CONTEXTS", str(obbff.REMEMBER_CONTEXTS), ) @log_start_end(log=logger) def call_dt(self, _): """Process dt command""" obbff.USE_DATETIME = not obbff.USE_DATETIME set_key(USER_ENV_FILE, "OPENBB_USE_DATETIME", str(obbff.USE_DATETIME)) @log_start_end(log=logger) def call_rich(self, _): """Process rich command""" obbff.ENABLE_RICH = not obbff.ENABLE_RICH set_key(USER_ENV_FILE, "OPENBB_ENABLE_RICH", str(obbff.ENABLE_RICH)) @log_start_end(log=logger) def call_richpanel(self, _): """Process richpanel command""" obbff.ENABLE_RICH_PANEL = not obbff.ENABLE_RICH_PANEL set_key( USER_ENV_FILE, "OPENBB_ENABLE_RICH_PANEL", str(obbff.ENABLE_RICH_PANEL), ) @log_start_end(log=logger) def call_ion(self, _): """Process ion command""" obbff.USE_ION = not obbff.USE_ION set_key(USER_ENV_FILE, "OPENBB_USE_ION", str(obbff.USE_ION)) @log_start_end(log=logger) def call_watermark(self, _): """Process watermark command""" obbff.USE_WATERMARK = not obbff.USE_WATERMARK set_key(USER_ENV_FILE, "OPENBB_USE_WATERMARK", str(obbff.USE_WATERMARK)) @log_start_end(log=logger) def call_cmdloc(self, _): """Process cmdloc command""" obbff.USE_CMD_LOCATION_FIGURE = not obbff.USE_CMD_LOCATION_FIGURE set_key( USER_ENV_FILE, "OPENBB_USE_CMD_LOCATION_FIGURE", str(obbff.USE_CMD_LOCATION_FIGURE), ) @log_start_end(log=logger) def call_tbhint(self, _): """Process tbhint command""" if obbff.TOOLBAR_HINT: console.print("Will take effect when running terminal next.") obbff.TOOLBAR_HINT = not obbff.TOOLBAR_HINT set_key( USER_ENV_FILE, "OPENBB_TOOLBAR_HINT", str(obbff.TOOLBAR_HINT), )