mirrors/pferd
1
0
Fork 0
mirror of https://github.com/Garmelon/PFERD.git synced 2025-07-15 15:32:36 +02:00
Code Issues Packages Projects Releases Wiki Activity

Only remove a progress task when we added it

Browse Source
This commit is contained in:
I-Al-Istannen
2020-05-10 12:28:30 +02:00
parent 9850ab1d73
commit c1b21f7772
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
PFERD/progress.py
View File

@@ -103,7 +103,9 @@ class ProgressContextManager:
traceback: Optional[TracebackType], traceback: Optional[TracebackType],
) -> Optional[bool]: ) -> Optional[bool]:
"""Context manager exit function. Removes the task.""" """Context manager exit function. Removes the task."""
if self._task_id is not None: if self._task_id is None:
return None
_progress.remove_task(self._task_id) _progress.remove_task(self._task_id)
if len(_progress.task_ids) == 0: if len(_progress.task_ids) == 0:
@@ -112,7 +114,6 @@ class ProgressContextManager:
_progress.refresh() _progress.refresh()
# And we existed, so remove the line above (remove_task leaves one behind) # 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 return None