From c1b21f777264455a1096c78d5e5b7868b8c53a75 Mon Sep 17 00:00:00 2001 From: I-Al-Istannen Date: Sun, 10 May 2020 12:28:30 +0200 Subject: [PATCH] Only remove a progress task when we added it --- PFERD/progress.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/PFERD/progress.py b/PFERD/progress.py index add6458..92f23d7 100644 --- a/PFERD/progress.py +++ b/PFERD/progress.py @@ -103,8 +103,10 @@ class ProgressContextManager: traceback: Optional[TracebackType], ) -> Optional[bool]: """Context manager exit function. Removes the task.""" - if self._task_id is not None: - _progress.remove_task(self._task_id) + if self._task_id is None: + return None + + _progress.remove_task(self._task_id) if len(_progress.task_ids) == 0: # We need to clean up after ourselves, as we were the last one @@ -112,8 +114,7 @@ class ProgressContextManager: _progress.refresh() # And we existed, so remove the line above (remove_task leaves one behind) - if self._task_id is not None: - Console().print(_OneLineUp()) + Console().print(_OneLineUp()) return None