bugfix: https://discourse.julialang.org/t/an-experiment-for-python-style-but-unidirectional-generators-for-julia/76451/2?u=complyue

Author: complyue

src/PyStyleButUnidirGenerators.jl

@@ -53,10 +53,15 @@ macro generator(fd)
       $gt!var = $GenTask(current_task(), Task(() -> begin
         try
           @assert current_task() === $gt!var.iteratee
-          $body
-          yieldto($gt!var.iterater, $StopIteration())
+          try
+            $body
+          finally
+            yieldto($gt!var.iterater, $StopIteration())
+          end
         catch e
-          Base.throwto($gt!var.iterater, e)
+          if !istaskdone($gt!var.iterater)
+            Base.throwto($gt!var.iterater, e)
+          end
         end
       end))
       return $gt!var

test/loops.jl

@@ -0,0 +1,19 @@
+
+using PyStyleButUnidirGenerators
+
+##.
+@generator function organpipe(n::Integer)
+  i = 0
+  while i != n
+    i += 1
+    @yield i
+  end
+  while true
+    i -= 1
+    i == 0 && return
+    @yield i
+  end
+end
+
+##.
+collect(organpipe(2))