From 350784eb70843c9640ba1a6d50a537ee26aa665d Mon Sep 17 00:00:00 2001 From: Mattias Jernberg Date: Sat, 9 Jun 2018 22:19:38 +0200 Subject: [PATCH] Print an error message when assertion fails --- lib/solaar/cli/__init__.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/lib/solaar/cli/__init__.py b/lib/solaar/cli/__init__.py index b94029a7..25001d8e 100644 --- a/lib/solaar/cli/__init__.py +++ b/lib/solaar/cli/__init__.py @@ -155,5 +155,9 @@ def run(cli_args=None, hidraw_path=None): from importlib import import_module m = import_module('.' + action, package=__name__) m.run(c, args, _find_receiver, _find_device) + except AssertionError as e: + from traceback import extract_tb + tb_last = extract_tb(_sys.exc_info()[2])[-1] + _sys.exit('%s: assertion failed: %s line %d' % (NAME.lower(), tb_last[0], tb_last[1])) except Exception as e: _sys.exit('%s: error: %s' % (NAME.lower(), e))