python-iterutils 0.0.4__tar.gz → 0.0.4.1__tar.gz

{python_iterutils-0.0.4 → python_iterutils-0.0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.0.4
+Version: 0.0.4.1
 Summary: Python another itertools.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-iterutils
 License: MIT

{python_iterutils-0.0.4 → python_iterutils-0.0.4.1}/iterutils/__init__.py

@@ -178,6 +178,7 @@ def run_gen_step(
         gen = gen_step
         close = None
     send = gen.send
+    throw = gen.throw
     if async_:
         async def process():
             try:
@@ -186,13 +187,20 @@ def run_gen_step(
                 else:
                     func = send(None)
                 while True:
-                    ret = func()
-                    if isawaitable(ret):
-                        ret = await ret
-                    if threaded:
-                        func = await to_thread(send, ret)
+                    try:
+                        ret = func()
+                        if isawaitable(ret):
+                            ret = await ret
+                    except BaseException as e:
+                        if threaded:
+                            func = await to_thread(throw, e)
+                        else:
+                            func = throw(e)
                     else:
-                        func = send(ret)
+                        if threaded:
+                            func = await to_thread(send, ret)
+                        else:
+                            func = send(ret)
             except StopIteration as e:
                 return e.value
             finally:
@@ -206,7 +214,12 @@ def run_gen_step(
         try:
             func = send(None)
             while True:
-                func = send(func())
+                try:
+                    ret = func()
+                except BaseException as e:
+                    func = throw(e)
+                else:
+                    func = send(ret)
         except StopIteration as e:
             return e.value
         finally:

{python_iterutils-0.0.4 → python_iterutils-0.0.4.1}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-iterutils"
-version = "0.0.4"
+version = "0.0.4.1"
 description = "Python another itertools."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"