omlish 0.0.0.dev410__py3-none-any.whl → 0.0.0.dev411__py3-none-any.whl

omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev410'
-__revision__ = 'fa58128e15cab568117709c80765dc8d633cb4e0'
+__version__ = '0.0.0.dev411'
+__revision__ = 'ab397888a161157384b42de969a8c743c5a2c4b6'
 
 
 #

omlish/lite/maysyncs.py

@@ -1,4 +1,4 @@
-# ruff: noqa: UP006 UP043 UP045 UP046 UP047
+# ruff: noqa: UP006 UP007 UP043 UP045 UP046 UP047
 # @omlish-lite
 """
 A system for writing a python function once which can then be effectively used in both sync and async contexts -
@@ -35,6 +35,8 @@ TODO:
  - for debug, mask any running eventloop while running maysync code
  - `[CO_ASYNC_GENERATOR] = {k for k, v in dis.COMPILER_FLAG_NAMES.items() if v == 'ASYNC_GENERATOR'}` ? inspect is big..
   - works down to 3.8
+ - make_maysync_from_sync can run with asyncio.run_in_thread
+ - make_maysync_from_async can run with asyncio.run_soon
 """
 import abc
 import functools
@@ -177,14 +179,18 @@ class _SyncMaysyncContext(_MaysyncContext):
         prev = _MaysyncThreadLocal.context
         _MaysyncThreadLocal.context = self
 
-        ph = sys.get_asyncgen_hooks()
-        sys.set_asyncgen_hooks(firstiter=None, finalizer=None)
+        ph: ta.Any = sys.get_asyncgen_hooks()
+        if ph.firstiter is not None or ph.finalizer is not None:
+            sys.set_asyncgen_hooks(firstiter=None, finalizer=None)
+        else:
+            ph = None
 
         try:
             return fn(*args, **kwargs)
 
         finally:
-            sys.set_asyncgen_hooks(*ph)
+            if ph is not None:
+                sys.set_asyncgen_hooks(*ph)
 
             _MaysyncThreadLocal.context = prev
 
@@ -300,7 +306,7 @@ class _FpMaywaitable(
 
     def a(self) -> ta.Awaitable[T]:
         if _MaysyncContext.current() is not None:
-            return _MaysyncFuture(self._x, self._args, self._kwargs)
+            return _MaysyncFuture(functools.partial(self._x, *self._args, **self._kwargs))  # noqa
 
         return self._x._a(*self._args, **self._kwargs)  # noqa
 
@@ -411,11 +417,6 @@ class _MgMaysyncFnLike(
     abc.ABC,
     ta.Generic[T],
 ):
-    """
-    A maysync object backed by an underlying generator yielding _MaysyncOp's. The _MgDriver and _MgGeneratorDriver
-    classes produce such generators.
-    """
-
     def __init__(
             self,
             mg: ta.Callable[..., T],
@@ -452,13 +453,13 @@ class _MgMaysyncFn(
 
 @ta.final
 class _MgMaysyncDriver:
-    def __init__(self, ctx, mg):
+    def __init__(self, ctx: _MaysyncContext, mg: ta.Any) -> None:
         self.ctx = ctx
         self.mg = mg
 
     value: ta.Any
 
-    def __iter__(self):
+    def __iter__(self) -> ta.Generator['_MaysyncFuture', None, None]:
         try:
             a = self.mg.__await__()
             try:
@@ -478,7 +479,8 @@ class _MgMaysyncDriver:
                         del f
 
                 finally:
-                    self.ctx.run(g.close)
+                    if g is not a:
+                        self.ctx.run(g.close)
 
             finally:
                 self.ctx.run(a.close)
@@ -527,11 +529,21 @@ class _MgMaysyncGeneratorFn(
 
 @ta.final
 class _MgMaysyncGeneratorDriver:
-    def __init__(self, ctx, ag):
+    def __init__(self, ctx: _MaysyncContext, ag: ta.Any) -> None:
         self.ctx = ctx
         self.ag = ag
 
-    def __iter__(self):
+    def __iter__(self) -> ta.Generator[
+        ta.Union[
+            ta.Tuple[ta.Literal['f'], '_MaysyncFuture'],
+            ta.Tuple[ta.Literal['o'], ta.Any],
+        ],
+        ta.Union[
+            ta.Tuple[ta.Any, BaseException],
+            None,
+        ],
+        None,
+    ]:
         try:
             ai = self.ag.__aiter__()
             try:
@@ -553,11 +565,12 @@ class _MgMaysyncGeneratorDriver:
 
                         del f
 
-                    i, e = yield ('o', drv.value)
+                    i, e = yield ('o', drv.value)  # type: ignore[misc]
 
             finally:
-                for f in _MgMaysyncDriver(self.ctx, ai.aclose()):
-                    yield ('f', f)
+                if ai is not self.ag:
+                    for f in _MgMaysyncDriver(self.ctx, ai.aclose()):
+                        yield ('f', f)
 
         finally:
             for f in _MgMaysyncDriver(self.ctx, self.ag.aclose()):
@@ -572,8 +585,7 @@ class _MgMaysyncGenerator(
         return _MgMaysyncGeneratorDriver(ctx, self._x._mg(*self._args, **self._kwargs))  # noqa
 
     def s(self) -> ta.Generator[O, I, None]:
-        drv = self._driver(_SyncMaysyncContext())
-        di = iter(drv)
+        di = iter(self._driver(_SyncMaysyncContext()))
 
         ie: ta.Any = None
 
@@ -592,8 +604,7 @@ class _MgMaysyncGenerator(
 
             elif t == 'o':
                 try:
-                    y = yield x
-                    ie = (y, None)
+                    ie = ((yield x), None)  # type: ignore[misc]
                 except BaseException as ex:  # noqa
                     ie = (None, ex)
 
@@ -607,8 +618,7 @@ class _MgMaysyncGenerator(
             return self._x._mg(*self._args, **self._kwargs)  # noqa
 
         async def inner():
-            drv = self._driver(_AsyncMaysyncContext())
-            di = iter(drv)
+            di = iter(self._driver(_AsyncMaysyncContext()))
 
             ie: ta.Any = None
 
@@ -627,8 +637,7 @@ class _MgMaysyncGenerator(
 
                 elif t == 'o':
                     try:
-                        y = yield x
-                        ie = (y, None)
+                        ie = ((yield x), None)
                     except BaseException as ex:  # noqa
                         ie = (None, ex)
 
@@ -694,16 +703,12 @@ class _MaysyncFutureNotAwaitedError(RuntimeError):
 class _MaysyncFuture(ta.Generic[T]):
     def __init__(
             self,
-            x: ta.Any,
-            args: ta.Tuple[ta.Any, ...],
-            kwargs: ta.Mapping[str, ta.Any],
+            x: ta.Callable[[], Maywaitable[T]],
     ) -> None:
-        self.x = x
-        self.args = args
-        self.kwargs = kwargs
+        self._x = x
 
     def __repr__(self) -> str:
-        return f'{self.__class__.__name__}@{id(self):x}({self.x!r}, done={self.done!r})'
+        return f'{self.__class__.__name__}@{id(self):x}({self._x!r}, done={self.done!r})'
 
     done: bool = False
     result: T
@@ -724,7 +729,7 @@ class _MaysyncFuture(ta.Generic[T]):
             return
 
         try:
-            self.result = self.x(*self.args, **self.kwargs).s()
+            self.result = self._x().s()
         except BaseException as ex:  # noqa
             self.error = ex
         self.done = True
@@ -734,7 +739,7 @@ class _MaysyncFuture(ta.Generic[T]):
             return
 
         try:
-            self.result = await self.x(*self.args, **self.kwargs).a()
+            self.result = await self._x().a()
         except BaseException as ex:  # noqa
             self.error = ex
         self.done = True

{omlish-0.0.0.dev410.dist-info → omlish-0.0.0.dev411.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omlish
-Version: 0.0.0.dev410
+Version: 0.0.0.dev411
 Summary: omlish
 Author: wrmsr
 License-Expression: BSD-3-Clause

{omlish-0.0.0.dev410.dist-info → omlish-0.0.0.dev411.dist-info}/RECORD

@@ -1,5 +1,5 @@
 omlish/.manifests.json,sha256=FLw7xkPiSXuImZgqSP8BwrEib2R1doSzUPLUkc-QUIA,8410
-omlish/__about__.py,sha256=4TpD9f2NLOxe8wPIx2jkab-TAgd8pW7229kNrBPdko4,3601
+omlish/__about__.py,sha256=qRafTkrQl5EPJTSgaE1ncsq5b84uBIoagA3OZKLm0p4,3601
 omlish/__init__.py,sha256=SsyiITTuK0v74XpKV8dqNaCmjOlan1JZKrHQv5rWKPA,253
 omlish/c3.py,sha256=rer-TPOFDU6fYq_AWio_AmA-ckZ8JDY5shIzQ_yXfzA,8414
 omlish/cached.py,sha256=MLap_p0rdGoDIMVhXVHm1tsbcWobJF0OanoodV03Ju8,542
@@ -490,7 +490,7 @@ omlish/lite/json.py,sha256=m0Ce9eqUZG23-H7-oOp8n1sf4fzno5vtK4AK_4Vc-Mg,706
 omlish/lite/logs.py,sha256=CWFG0NKGhqNeEgryF5atN2gkPYbUdTINEw_s1phbINM,51
 omlish/lite/marshal.py,sha256=K_wnZwfC8cftGILyE3RlmzQEYuZOfzkMLKey41zuwtM,20296
 omlish/lite/maybes.py,sha256=0p_fzb6yiOjEpvMKaQ53Q6CH1VPW1or7v7Lt1JIKcgM,4359
-omlish/lite/maysyncs.py,sha256=XZ-zbK3bES8Al6--lYcvGGhYKqfVg1zoHJ6GCwLbiv4,19663
+omlish/lite/maysyncs.py,sha256=QqXV0thRHwevPFlkyELv4tHmXBM9cJ6uMjx0pZ0LbHc,20046
 omlish/lite/pycharm.py,sha256=FRHGcCDo42UzZXqNwW_DkhI-6kb_CmJKPiQ8F6mYkLA,1174
 omlish/lite/reflect.py,sha256=gI-Qlws9V-jND7kvCQFaIhBFrndVpDsikTQ7C6U2z3w,2434
 omlish/lite/reprs.py,sha256=2Bc7ukhKvYNTKmxPIuv9glZIph13C37y_W4fg9pBnu8,2006
@@ -909,9 +909,9 @@ omlish/typedvalues/marshal.py,sha256=AtBz7Jq-BfW8vwM7HSxSpR85JAXmxK2T0xDblmm1HI0
 omlish/typedvalues/of_.py,sha256=UXkxSj504WI2UrFlqdZJbu2hyDwBhL7XVrc2qdR02GQ,1309
 omlish/typedvalues/reflect.py,sha256=PAvKW6T4cW7u--iX80w3HWwZUS3SmIZ2_lQjT65uAyk,1026
 omlish/typedvalues/values.py,sha256=ym46I-q2QJ_6l4UlERqv3yj87R-kp8nCKMRph0xQ3UA,1307
-omlish-0.0.0.dev410.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
-omlish-0.0.0.dev410.dist-info/METADATA,sha256=tqNKk_ovprTJtbpkIsi_jZRS44y4hTlBreu5pTxzTRs,18881
-omlish-0.0.0.dev410.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-omlish-0.0.0.dev410.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
-omlish-0.0.0.dev410.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
-omlish-0.0.0.dev410.dist-info/RECORD,,
+omlish-0.0.0.dev411.dist-info/licenses/LICENSE,sha256=B_hVtavaA8zCYDW99DYdcpDLKz1n3BBRjZrcbv8uG8c,1451
+omlish-0.0.0.dev411.dist-info/METADATA,sha256=3MPeR2SuWXjWtGnh7xZs-LvdohVQarQEXwYO7Oh01ao,18881
+omlish-0.0.0.dev411.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+omlish-0.0.0.dev411.dist-info/entry_points.txt,sha256=Lt84WvRZJskWCAS7xnQGZIeVWksprtUHj0llrvVmod8,35
+omlish-0.0.0.dev411.dist-info/top_level.txt,sha256=pePsKdLu7DvtUiecdYXJ78iO80uDNmBlqe-8hOzOmfs,7
+omlish-0.0.0.dev411.dist-info/RECORD,,