mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-11-05 08:51:20 +00:00
ee34a1f278
* Fixed isort * Fixed issues * FIxed linters --------- Co-authored-by: James Maslek <jmaslek11@gmail.com>
124 lines
3.9 KiB
Python
124 lines
3.9 KiB
Python
"""Discovery Controller Module"""
|
|
__docformat__ = "numpy"
|
|
|
|
import argparse
|
|
import logging
|
|
from typing import List
|
|
|
|
from openbb_terminal import feature_flags as obbff
|
|
from openbb_terminal.custom_prompt_toolkit import NestedCompleter
|
|
from openbb_terminal.decorators import log_start_end
|
|
from openbb_terminal.etf.discovery import wsj_view
|
|
from openbb_terminal.helper_funcs import (
|
|
EXPORT_ONLY_RAW_DATA_ALLOWED,
|
|
)
|
|
from openbb_terminal.menu import session
|
|
from openbb_terminal.parent_classes import BaseController
|
|
from openbb_terminal.rich_config import MenuText, console
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DiscoveryController(BaseController):
|
|
"""Discovery Controller class"""
|
|
|
|
CHOICES_COMMANDS = [
|
|
"gainers",
|
|
"decliners",
|
|
"active",
|
|
]
|
|
PATH = "/etf/disc/"
|
|
CHOICES_GENERATION = True
|
|
|
|
def __init__(self, queue: List[str] = None):
|
|
"""Constructor"""
|
|
super().__init__(queue)
|
|
|
|
if session and obbff.USE_PROMPT_TOOLKIT:
|
|
choices: dict = self.choices_default
|
|
|
|
self.completer = NestedCompleter.from_nested_dict(choices)
|
|
|
|
def print_help(self):
|
|
"""Print help"""
|
|
mt = MenuText("etf/disc/", 60)
|
|
mt.add_cmd("gainers")
|
|
mt.add_cmd("decliners")
|
|
mt.add_cmd("active")
|
|
console.print(text=mt.menu_text, menu="ETF - Discovery")
|
|
|
|
@log_start_end(log=logger)
|
|
def call_gainers(self, other_args):
|
|
"""Process gainers command"""
|
|
parser = argparse.ArgumentParser(
|
|
prog="gainers",
|
|
description="Displays top ETF/Mutual fund gainers from wsj.com/market-data",
|
|
add_help=False,
|
|
)
|
|
if other_args and "-" not in other_args[0][0]:
|
|
other_args.insert(0, "-l")
|
|
ns_parser = self.parse_known_args_and_warn(
|
|
parser, other_args, export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED, limit=10
|
|
)
|
|
if ns_parser:
|
|
wsj_view.show_top_mover(
|
|
"gainers",
|
|
ns_parser.limit,
|
|
ns_parser.export,
|
|
sheet_name=" ".join(ns_parser.sheet_name)
|
|
if ns_parser.sheet_name
|
|
else None,
|
|
)
|
|
|
|
@log_start_end(log=logger)
|
|
def call_decliners(self, other_args):
|
|
"""Process decliners command"""
|
|
parser = argparse.ArgumentParser(
|
|
prog="decliners",
|
|
description="Displays top ETF/Mutual fund decliners from wsj.com/market-data",
|
|
add_help=False,
|
|
)
|
|
if other_args and "-" not in other_args[0][0]:
|
|
other_args.insert(0, "-l")
|
|
ns_parser = self.parse_known_args_and_warn(
|
|
parser,
|
|
other_args,
|
|
export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED,
|
|
limit=10,
|
|
)
|
|
if ns_parser:
|
|
wsj_view.show_top_mover(
|
|
"decliners",
|
|
ns_parser.limit,
|
|
ns_parser.export,
|
|
sheet_name=" ".join(ns_parser.sheet_name)
|
|
if ns_parser.sheet_name
|
|
else None,
|
|
)
|
|
|
|
@log_start_end(log=logger)
|
|
def call_active(self, other_args):
|
|
"""Process gainers command"""
|
|
parser = argparse.ArgumentParser(
|
|
prog="active",
|
|
description="Displays most active ETF/Mutual funds from wsj.com/market-data",
|
|
add_help=False,
|
|
)
|
|
if other_args and "-" not in other_args[0][0]:
|
|
other_args.insert(0, "-l")
|
|
ns_parser = self.parse_known_args_and_warn(
|
|
parser,
|
|
other_args,
|
|
export_allowed=EXPORT_ONLY_RAW_DATA_ALLOWED,
|
|
limit=10,
|
|
)
|
|
if ns_parser:
|
|
wsj_view.show_top_mover(
|
|
"active",
|
|
ns_parser.limit,
|
|
ns_parser.export,
|
|
sheet_name=" ".join(ns_parser.sheet_name)
|
|
if ns_parser.sheet_name
|
|
else None,
|
|
)
|