mirror of
https://github.com/OpenBB-finance/OpenBB.git
synced 2026-10-27 20:42:26 +00:00
3d243b361a
* keep session alive on reset * remove print * silence routines * ruff * skip login if is papermill * ruff * move is_papermill to avoid prompt_toolkit warning fd is not a terminal * move is_papermill back and check for atty instead * only download routines and update sources if terminal * only starts prompt toolkit if isatty * black * avoid duplicate files after reset/login * remove hub routines on reset * comment * fix unittest * forgot this debugging prints
67 lines
1.7 KiB
Python
67 lines
1.7 KiB
Python
import logging
|
|
import sys
|
|
from typing import Optional
|
|
|
|
from matplotlib import pyplot
|
|
from prompt_toolkit import PromptSession
|
|
from prompt_toolkit.eventloop.inputhook import set_eventloop_with_inputhook
|
|
from prompt_toolkit.history import FileHistory
|
|
|
|
from openbb_terminal.core.config.paths import HIST_FILE_PATH
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
# pylint: disable=W0104
|
|
def is_jupyter() -> bool:
|
|
try:
|
|
__IPYTHON__ # type: ignore
|
|
return True
|
|
except NameError:
|
|
return False
|
|
|
|
|
|
def is_papermill() -> bool:
|
|
"""Check if session is being launched with args '-m ipykernel_launcher',
|
|
thus coming from papermill Popen. See 'ipykernel_launcher' in reports_model
|
|
for more detail.
|
|
"""
|
|
|
|
return all(
|
|
i in sys.argv
|
|
for i in [
|
|
"-m",
|
|
"ipykernel_launcher",
|
|
"-f",
|
|
"--HistoryManager.hist_file=:memory:",
|
|
]
|
|
)
|
|
|
|
|
|
def inputhook(inputhook_context):
|
|
while not inputhook_context.input_is_ready():
|
|
try:
|
|
pyplot.pause(0.1)
|
|
except Exception as exp:
|
|
logger.exception("%s", type(exp).__name__)
|
|
continue
|
|
return False
|
|
|
|
|
|
try:
|
|
if sys.stdin.isatty():
|
|
session: Optional[PromptSession] = PromptSession(
|
|
history=FileHistory(str(HIST_FILE_PATH))
|
|
)
|
|
set_eventloop_with_inputhook(inputhook)
|
|
else:
|
|
session = None
|
|
# pylint: disable=unused-variable
|
|
except Exception as e: # noqa: F841
|
|
logger.exception("%s", type(e).__name__)
|
|
print(
|
|
"WARNING: Prompt toolkit is turned on but did not initialize successfully."
|
|
" Falling back to input()..."
|
|
)
|
|
session = None # type: ignore
|