mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 04:52:07 +00:00
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
import inspect
|
|
from typing import Dict, Type
|
|
|
|
from argparse_translator.argparse_translator import ArgparseTranslator
|
|
|
|
|
|
class ArgparseClassProcessor:
|
|
"""
|
|
Process a target class to create ArgparseTranslators for its methods.
|
|
"""
|
|
|
|
def __init__(self, target_class: Type, menu_designation: str, add_help: bool):
|
|
"""
|
|
Initialize the ArgparseClassProcessor.
|
|
|
|
Parameters
|
|
----------
|
|
target_class : Type
|
|
The target class whose methods will be processed.
|
|
menu_designation : str
|
|
The designation for the menu level associated with the target class.
|
|
"""
|
|
self._target_class: Type = target_class
|
|
self._menu_designation: str = menu_designation
|
|
self._add_help: bool = add_help
|
|
self._translators: Dict[str, Dict[str, ArgparseTranslator]] = {
|
|
self._menu_designation: {}
|
|
}
|
|
|
|
self._process_methods()
|
|
|
|
def _process_methods(self):
|
|
"""
|
|
Process the methods of the target class to create ArgparseTranslators.
|
|
"""
|
|
for name, method in inspect.getmembers(self._target_class, inspect.ismethod):
|
|
if name.startswith("__") or name.startswith("_"):
|
|
continue
|
|
|
|
self._translators[self._menu_designation][name] = ArgparseTranslator(
|
|
func=method, add_help=self._add_help
|
|
)
|
|
|
|
def get_translator(self, menu: str, command: str) -> ArgparseTranslator:
|
|
"""
|
|
Retrieve the ArgparseTranslator object associated with a specific menu and command.
|
|
|
|
Parameters
|
|
----------
|
|
menu : str
|
|
The menu designation.
|
|
command : str
|
|
The command associated with the ArgparseTranslator.
|
|
|
|
Returns
|
|
-------
|
|
ArgparseTranslator
|
|
The ArgparseTranslator associated with the specified menu and command.
|
|
"""
|
|
return self._translators[menu][command]
|