Shofel2_T124_python/venv/lib/python3.10/site-packages/pyfx/cli_utils.py

43 lines
1.2 KiB
Python

"""
Utility libraries for CLI.
"""
import functools
import select
import sys
import click
from loguru import logger
def exit_on_exception(func):
"""
A decorator which exit the current click application when there's
unexpected error and print the error message to the stderr.
"""
# noinspection PyBroadException
@functools.wraps(func)
def wrapper(*args, **kwargs):
try:
return func(*args, **kwargs)
except Exception as e:
if not isinstance(e.__str__(), str):
e = Exception(
f"Unknown error {type(e)}. Please consider create an issue "
f"at https://github.com/cielong/pyfx/issues."
)
logger.opt(exception=True).error(e)
raise click.ClickException(e.__str__())
return wrapper
# noinspection PyBroadException
def is_stdin_readable():
try:
ready, _, _ = select.select([sys.stdin.fileno()], [],
[sys.stdin.fileno()], 0)
return ready
except Exception:
# Ignore any potential error when trying to wait on stdin
# This may happen when sys.stdin has been replaced, e.g. in E2E test.
return False