From e1c876817e56cf4a6d2df62b8c0faa4355dfc950 Mon Sep 17 00:00:00 2001 From: Sven Augustin Date: Mon, 11 May 2020 17:27:22 +0000 Subject: [PATCH] added better repr for DAQTask --- slic/core/task/daqtask.py | 10 ++++++++++ slic/utils/printing.py | 7 +++++++ 2 files changed, 17 insertions(+) diff --git a/slic/core/task/daqtask.py b/slic/core/task/daqtask.py index f7884216..ee51e6b0 100644 --- a/slic/core/task/daqtask.py +++ b/slic/core/task/daqtask.py @@ -1,3 +1,5 @@ +from slic.utils.printing import itemize + from .task import Task @@ -7,6 +9,14 @@ class DAQTask(Task): super().__init__(*args, **kwargs) self.filenames = homogenized_list(filename, filenames) + def __repr__(self): + res = super().__repr__() + if self.filenames: + items = itemize(self.filenames) + header = "Output files:" + res = "\n".join((res, header, items)) + return res + def homogenized_list(item, items): diff --git a/slic/utils/printing.py b/slic/utils/printing.py index 67f906c2..62904215 100644 --- a/slic/utils/printing.py +++ b/slic/utils/printing.py @@ -26,4 +26,11 @@ def format_header(msg, line_char="-"): return msg +def itemize(iterable, bullet="-"): + if not bullet.endswith(" "): + bullet += " " + lines = (bullet + str(i) for i in iterable) + return "\n".join(lines) + +