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 <jan@janware.com>
This commit is contained in:
Jan Lindemann 2020-04-23 10:32:05 +02:00
commit 4a047ca260

View file

@ -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]