From 4a047ca26018492b6b02217f673050c05c6c45ec Mon Sep 17 00:00:00 2001 From: Jan Lindemann Date: Thu, 23 Apr 2020 10:32:05 +0200 Subject: [PATCH] log.remove_from_prefix(): Add support for str argument In addition to an integer, a string can now be passed, which allows prefix = ' | ' append_to_prefix(prefix) slog(INFO, "something") remove_from_prefix(prefx) Signed-off-by: Jan Lindemann --- tools/python/jwutils/log.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/tools/python/jwutils/log.py b/tools/python/jwutils/log.py index 3ade23c..2d3b7f7 100644 --- a/tools/python/jwutils/log.py +++ b/tools/python/jwutils/log.py @@ -220,7 +220,9 @@ def append_to_prefix(prefix: str) -> str: # export _log_prefix += prefix return r -def remove_from_prefix(count: int) -> str: # export +def remove_from_prefix(count) -> str: # export + if isinstance(count, str): + count = len(count) global _log_prefix r = _log_prefix _log_prefix = _log_prefix[:-count]