fix: read ret when no exception

Grapycal · Apr 23, 2024 · ad077ad · ad077ad
1 parent b00e936
commit ad077ad
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/backend/src/grapycal/core/background_runner.py b/backend/src/grapycal/core/background_runner.py
@@ -115,9 +115,10 @@ def run(self):
                         ret = task()
                     except Exception as e:
                         on_exception(e, exception_callback)
-                    # if ret is a generator, push it to stack
-                    if isinstance(ret, Iterator):
-                        self._stack.append(TaskInfo(iter(ret), exception_callback))
+                    else:
+                        # if ret is a generator, push it to stack
+                        if isinstance(ret, Iterator):
+                            self._stack.append(TaskInfo(iter(ret), exception_callback))
 
             except RunnerInterrupt:
                 logger.info("Runner interrupted")