43 lines
1.2 KiB
Python
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
|