mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-28 13:01:38 +00:00
721a7d560c
* Fixes * Update stocks_helper.py * update git-actions set-output to new format * Update stocks_helper.py * Update terminal_helper.py * removed LineAnnotateDrawer from qa_view * lint * Update stocks_helper.py Co-authored-by: Colin Delahunty <72827203+colin99d@users.noreply.github.com> Co-authored-by: James Maslek <jmaslek11@gmail.com>
88 lines
2.3 KiB
Python
88 lines
2.3 KiB
Python
# This is for helpers that do NOT import any OpenBB Modules
|
|
from typing import Callable, Any, Literal
|
|
import os
|
|
|
|
from rich.console import Console
|
|
|
|
console = Console()
|
|
|
|
menus = Literal["", "featflags", "settings"]
|
|
|
|
|
|
def handle_error(name: str, default: Any, menu: menus = ""):
|
|
"""Handles the error by returning the default value and printing an
|
|
informative error message.
|
|
|
|
Parameters
|
|
----------
|
|
name: str
|
|
The name of the environment variable
|
|
default: Any
|
|
The value to return if the converter fails
|
|
menu: menus
|
|
If provided, will tell the user where to fix the setting
|
|
|
|
Returns
|
|
----------
|
|
Any
|
|
The default value
|
|
|
|
"""
|
|
base = f"[red]Invalid variable provided for variable '{name}'."
|
|
if menu:
|
|
base += f" Please change the setting in the `{menu}` menu."
|
|
base += "[/red]\n"
|
|
console.print(base)
|
|
return default
|
|
|
|
|
|
def load_env_vars(
|
|
name: str, converter: Callable, default: Any, menu: menus = ""
|
|
) -> Any:
|
|
"""Loads an environment variable and attempts to convert it to the correct data type.
|
|
Will return the provided default if it fails
|
|
|
|
Parameters
|
|
----------
|
|
name: str
|
|
The name of the environment variable
|
|
converter: Callable
|
|
The function to convert the env variable to the desired format
|
|
default: Any
|
|
The value to return if the converter fails
|
|
menu: menus
|
|
If provided, will tell the user where to fix the setting
|
|
|
|
Returns
|
|
----------
|
|
Any
|
|
The value or the default
|
|
"""
|
|
raw_var = os.getenv(name, str(default))
|
|
try:
|
|
return converter(raw_var)
|
|
except ValueError:
|
|
return handle_error(name, default, menu)
|
|
except AttributeError:
|
|
return handle_error(name, default, menu)
|
|
except TypeError:
|
|
return handle_error(name, default, menu)
|
|
|
|
|
|
def strtobool(val):
|
|
"""Convert a string representation of truth to true (1) or false (0).
|
|
|
|
True values are 'y', 'yes', 't', 'true', 'on', and '1'; false values
|
|
are 'n', 'no', 'f', 'false', 'off', and '0'. Raises ValueError if
|
|
'val' is anything else.
|
|
"""
|
|
val = str(val).lower()
|
|
if val in ("y", "yes", "t", "true", "on", "1"):
|
|
output = 1
|
|
elif val in ("n", "no", "f", "false", "off", "0"):
|
|
output = 0
|
|
else:
|
|
raise ValueError(f"invalid truth value {val}")
|
|
|
|
return output
|