Tidy up autoloading in places (#1832)

* Only try loading from stocks and crypto contexts

* im terrible with git
This commit is contained in:
James Maslek
2022-05-17 18:50:18 -04:00
committed by GitHub
parent 2815f42a3c
commit 26781cb153
3 changed files with 7 additions and 3 deletions

View File

@@ -80,6 +80,7 @@ class ETFController(BaseController):
self.etf_name = ""
self.etf_data = ""
self.etf_holdings: List = list()
self.TRY_RELOAD = True
if session and obbff.USE_PROMPT_TOOLKIT:
choices: dict = {c: {} for c in self.controller_choices}

View File

@@ -76,6 +76,7 @@ class FundController(BaseController):
self.data = pd.DataFrame()
self.fund_name = ""
self.fund_symbol = ""
self.TRY_RELOAD = True
if session and obbff.USE_PROMPT_TOOLKIT:
choices: dict = {c: {} for c in self.controller_choices}

View File

@@ -76,7 +76,7 @@ class BaseController(metaclass=ABCMeta):
SUPPORT_CHOICES: Dict = {}
COMMAND_SEPARATOR = "/"
KEYS_MENU = "keys" + COMMAND_SEPARATOR
TRY_RELOAD = False
PATH: str = ""
FILE_PATH: str = ""
@@ -163,7 +163,7 @@ class BaseController(metaclass=ABCMeta):
@abstractmethod
def print_help(self) -> None:
raise NotImplementedError("Must override print_help")
raise NotImplementedError("Must override print_help.")
def contains_keys(self, string_to_check: str) -> bool:
if self.KEYS_MENU in string_to_check or self.KEYS_MENU in self.PATH:
@@ -462,7 +462,7 @@ class BaseController(metaclass=ABCMeta):
console.print(f" Replacing by '{an_input}'.")
self.queue.insert(0, an_input)
else:
if "load" in self.controller_choices:
if self.TRY_RELOAD:
console.print(f"Trying `load {an_input}`")
self.queue.insert(0, "load " + an_input)
console.print("")
@@ -480,6 +480,7 @@ class StockBaseController(BaseController, metaclass=ABCMeta):
self.start = ""
self.suffix = "" # To hold suffix for Yahoo Finance
self.add_info = stocks_helper.additional_info_about_ticker("")
self.TRY_RELOAD = True
def call_load(self, other_args: List[str]):
"""Process load command"""
@@ -670,6 +671,7 @@ class CryptoBaseController(BaseController, metaclass=ABCMeta):
self.price_str = ""
self.resolution = "1D"
self.coin = ""
self.TRY_RELOAD = True
def call_load(self, other_args):
"""Process load command"""