omlish 0.0.0.dev386__tar.gz → 0.0.0.dev387__tar.gz

{omlish-0.0.0.dev386/omlish.egg-info → omlish-0.0.0.dev387}/PKG-INFO

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

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387}/omlish/__about__.py

@@ -1,5 +1,5 @@
-__version__ = '0.0.0.dev386'
-__revision__ = 'e2f1a23f62c35a30fdb5a509f65af87289ceafef'
+__version__ = '0.0.0.dev387'
+__revision__ = '169f8b630b04ef6cf83c18d134c799c83f033818'
 
 
 #

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387}/omlish/lang/__init__.py

@@ -250,13 +250,12 @@ from .iterables import (  # noqa
     take,
 )
 
-from .maysync_ import (  # noqa
-    maysync_op,
+from .maysyncs import (  # noqa
+    MaysyncableP,
 
-    maysync_yield,
+    make_maysync,
 
-    maysync_wrap,
-    a_maysync_wrap,
+    maysync,
 )
 
 from .objects import (  # noqa
@@ -393,21 +392,13 @@ from ..lite.maybes import (  # noqa
     Maybe,
 )
 
-from ..lite.maysync import (  # noqa
-    MaysyncGen,
-    MaysyncFn,
-
-    MaysyncOp,
-
-    maysync,
-    a_maysync,
-
-    MaysyncRunnable,
-)
-
 empty = Maybe.empty
 just = Maybe.just
 
+from ..lite.maysyncs import (  # noqa
+    Maysyncable,
+)
+
 from ..lite.reprs import (  # noqa
     AttrRepr,
     attr_repr,

omlish-0.0.0.dev387/omlish/lang/maysyncs.py

@@ -0,0 +1,41 @@
+import typing as ta
+
+from ..lite.maysyncs import make_maysync as _make_maysync
+from ..lite.maysyncs import maysync as _maysync
+
+
+T_co = ta.TypeVar('T_co', covariant=True)
+P = ta.ParamSpec('P')
+
+
+##
+
+
+class MaysyncableP(ta.Protocol[P, T_co]):
+    @property
+    def s(self) -> ta.Callable[P, T_co]:
+        ...
+
+    @property
+    def a(self) -> ta.Callable[P, ta.Awaitable[T_co]]:
+        ...
+
+    @property
+    def m(self) -> ta.Callable[P, ta.Awaitable[T_co]]:
+        ...
+
+
+def make_maysync(
+        s: ta.Callable[P, T_co],
+        a: ta.Callable[P, ta.Awaitable[T_co]],
+) -> MaysyncableP[P, T_co]:
+    return ta.cast('MaysyncableP[P, T_co]', _make_maysync(
+        s,
+        a,
+    ))
+
+
+def maysync(m: ta.Callable[P, ta.Awaitable[T_co]]) -> MaysyncableP[P, T_co]:
+    return ta.cast('MaysyncableP[P, T_co]', _maysync(
+        m,
+    ))

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387}/omlish/lite/args.py

@@ -41,3 +41,8 @@ class Args:
             return args(fn)
         else:
             return fn()
+
+    EMPTY: ta.ClassVar['Args']
+
+
+Args.EMPTY = Args()

omlish-0.0.0.dev387/omlish/lite/maysyncs.py

@@ -0,0 +1,184 @@
+# ruff: noqa: UP045
+# @omlish-lite
+import abc
+import dataclasses as dc
+import functools
+import typing as ta
+
+from .args import Args
+from .check import check
+
+
+T = ta.TypeVar('T')
+
+_MaysyncGen = ta.Generator['_MaysyncOp', ta.Any, T]  # ta.TypeAlias
+
+
+##
+
+
+@dc.dataclass(frozen=True, eq=False)
+class Maysyncable(abc.ABC, ta.Generic[T]):
+    @abc.abstractmethod
+    def s(self, *args: ta.Any, **kwargs: ta.Any) -> T:
+        raise NotImplementedError
+
+    @abc.abstractmethod
+    def a(self, *args: ta.Any, **kwargs: ta.Any) -> ta.Awaitable[T]:
+        raise NotImplementedError
+
+    @ta.final
+    def m(self, *args: ta.Any, **kwargs: ta.Any) -> ta.Awaitable[T]:
+        return _MaysyncFuture(_MaysyncOp(self, Args(*args, **kwargs)))
+
+
+##
+
+
+@dc.dataclass(frozen=True, eq=False)
+class _FnMaysyncable(Maysyncable[T]):
+    s: ta.Callable[..., T]
+    a: ta.Callable[..., ta.Awaitable[T]]
+
+    def __post_init__(self) -> None:
+        check.not_none(self.s)
+        check.not_none(self.a)
+
+    def __init_subclass__(cls, **kwargs):
+        raise TypeError
+
+
+_FnMaysyncable.__abstractmethods__ = frozenset()
+
+
+def make_maysync(
+        s: ta.Callable[..., T],
+        a: ta.Callable[..., ta.Awaitable[T]],
+) -> Maysyncable[T]:
+    return _FnMaysyncable(
+        s,
+        a,
+    )
+
+
+##
+
+
+@dc.dataclass(frozen=True, eq=False)
+class _MgMaysyncable(Maysyncable[T]):
+    mg: ta.Callable[..., _MaysyncGen[T]]
+
+    def __init_subclass__(cls, **kwargs):
+        raise TypeError
+
+    def s(self, *args: ta.Any, **kwargs: ta.Any) -> T:
+        g = self.mg(*args, **kwargs)
+
+        i: ta.Any = None
+        e: ta.Any = None
+
+        while True:
+            try:
+                if e is not None:
+                    o = g.throw(e)
+                else:
+                    o = g.send(i)
+            except StopIteration as ex:
+                return ex.value
+
+            i = None
+            e = None
+
+            mo = check.isinstance(o, _MaysyncOp)
+            try:
+                i = mo.a(mo.mx.s)
+            except BaseException as ex:  # noqa
+                e = ex
+
+    async def a(self, *args: ta.Any, **kwargs: ta.Any) -> T:
+        g = self.mg(*args, **kwargs)
+
+        i: ta.Any = None
+        e: ta.Any = None
+
+        while True:
+            try:
+                if e is not None:
+                    o = g.throw(e)
+                else:
+                    o = g.send(i)
+            except StopIteration as ex:
+                return ex.value
+
+            i = None
+            e = None
+
+            mo = check.isinstance(o, _MaysyncOp)
+            try:
+                i = await mo.a(mo.mx.a)
+            except BaseException as ex:  # noqa
+                e = ex
+
+
+def maysync(m: ta.Callable[..., ta.Awaitable[T]]) -> Maysyncable[T]:
+    @functools.wraps(m)
+    def mg_fn(*args, **kwargs):
+        a = m(*args, **kwargs).__await__()
+
+        try:
+            g = iter(a)
+            try:
+                while True:
+                    try:
+                        o = g.send(None)
+                    except StopIteration as e:
+                        return e.value
+
+                    f = check.isinstance(o, _MaysyncFuture)
+                    if not f.done:
+                        try:
+                            f.result = yield f.op
+                        except BaseException as e:  # noqa
+                            f.error = e
+                        f.done = True
+
+            finally:
+                g.close()
+
+        finally:
+            a.close()
+
+    return _MgMaysyncable(mg_fn)
+
+
+##
+
+
+@dc.dataclass(frozen=True, eq=False)
+class _MaysyncOp:
+    mx: Maysyncable
+    a: Args
+
+    def __init_subclass__(cls, **kwargs):
+        raise TypeError
+
+
+##
+
+
+@dc.dataclass(eq=False)
+class _MaysyncFuture(ta.Generic[T]):
+    op: _MaysyncOp
+
+    done: bool = False
+    error: ta.Optional[BaseException] = None
+    result: ta.Optional[T] = None
+
+    def __await__(self):
+        if not self.done:
+            yield self
+        if not self.done:
+            raise RuntimeError("await wasn't used with event future")
+        if self.error is not None:
+            raise self.error
+        return self.result

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387/omlish.egg-info}/PKG-INFO

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

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387}/omlish.egg-info/SOURCES.txt

@@ -445,7 +445,7 @@ omlish/lang/functions.py
 omlish/lang/generators.py
 omlish/lang/imports.py
 omlish/lang/iterables.py
-omlish/lang/maysync_.py
+omlish/lang/maysyncs.py
 omlish/lang/objects.py
 omlish/lang/outcomes.py
 omlish/lang/overrides.py
@@ -488,7 +488,7 @@ omlish/lite/json.py
 omlish/lite/logs.py
 omlish/lite/marshal.py
 omlish/lite/maybes.py
-omlish/lite/maysync.py
+omlish/lite/maysyncs.py
 omlish/lite/pycharm.py
 omlish/lite/reflect.py
 omlish/lite/reprs.py

{omlish-0.0.0.dev386 → omlish-0.0.0.dev387}/pyproject.toml

@@ -13,7 +13,7 @@ urls = {source = 'https://github.com/wrmsr/omlish'}
 license = 'BSD-3-Clause'
 readme = 'README.md'
 requires-python = '>=3.13'
-version = '0.0.0.dev386'
+version = '0.0.0.dev387'
 classifiers = [
     'Development Status :: 2 - Pre-Alpha',
     'Intended Audience :: Developers',

omlish-0.0.0.dev386/omlish/lang/maysync_.py

@@ -1,53 +0,0 @@
-import functools
-import typing as ta
-
-from ..lite.maysync import MaysyncGen
-from ..lite.maysync import MaysyncOp
-from ..lite.maysync import a_maysync
-from ..lite.maysync import maysync
-
-
-T = ta.TypeVar('T')
-P = ta.ParamSpec('P')
-
-
-##
-
-
-def maysync_op(
-        fn: ta.Callable[P, T],
-        a_fn: ta.Callable[P, ta.Awaitable[T]],
-) -> MaysyncOp[T]:
-    return MaysyncOp(
-        fn,
-        a_fn,
-    )
-
-
-##
-
-
-def maysync_yield(
-        fn: ta.Callable[P, T],
-        a_fn: ta.Callable[P, ta.Awaitable[T]],
-) -> ta.Callable[P, ta.Generator[ta.Any, ta.Any, T]]:
-    def inner(*args, **kwargs):
-        return (yield MaysyncOp(fn, a_fn)(*args, **kwargs))
-    return inner
-
-
-##
-
-
-def maysync_wrap(fn: ta.Callable[P, MaysyncGen[T]]) -> ta.Callable[P, T]:
-    @functools.wraps(fn)
-    def inner(*args, **kwargs):
-        return maysync(fn, *args, **kwargs)
-    return inner
-
-
-def a_maysync_wrap(fn: ta.Callable[P, MaysyncGen[T]]) -> ta.Callable[P, ta.Awaitable[T]]:
-    @functools.wraps(fn)
-    async def inner(*args, **kwargs):
-        return await a_maysync(fn, *args, **kwargs)
-    return inner

omlish-0.0.0.dev386/omlish/lite/maysync.py

@@ -1,74 +0,0 @@
-# ruff: noqa: UP045
-import abc
-import dataclasses as dc
-import typing as ta
-
-from .args import Args
-
-
-T = ta.TypeVar('T')
-
-
-MaysyncGen = ta.Generator['MaysyncOp', ta.Any, T]  # ta.TypeAlias
-MaysyncFn = ta.Callable[..., MaysyncGen[T]]  # ta.TypeAlias
-
-
-##
-
-
-@dc.dataclass(frozen=True)
-class MaysyncOp(ta.Generic[T]):
-    fn: ta.Callable[..., T]
-    a_fn: ta.Callable[..., ta.Awaitable[T]]
-
-    args: ta.Optional[Args] = None
-
-    def __call__(self, *args: ta.Any, **kwargs: ta.Any) -> 'MaysyncOp[T]':
-        if self.args is not None:
-            raise RuntimeError('Args already bound')
-        return dc.replace(self, args=Args(*args, **kwargs))
-
-
-maysync_op = MaysyncOp
-
-
-##
-
-
-def maysync(fn: MaysyncFn[T], *args: ta.Any, **kwargs: ta.Any) -> T:
-    g = fn(*args, **kwargs)
-    i = None
-    while True:
-        try:
-            o = g.send(i)
-        except StopIteration as e:
-            return e.value
-        i = Args.call(o.fn, o.args)
-
-
-async def a_maysync(fn: MaysyncFn[T], *args: ta.Any, **kwargs: ta.Any) -> T:
-    g = fn(*args, **kwargs)
-    i = None
-    while True:
-        try:
-            o = g.send(i)
-        except StopIteration as e:
-            return e.value
-        i = await Args.call(o.a_fn, o.args)
-
-
-##
-
-
-class MaysyncRunnable(abc.ABC, ta.Generic[T]):
-    @abc.abstractmethod
-    def m_run(self) -> MaysyncGen[T]:
-        raise NotImplementedError
-
-    @ta.final
-    def run(self) -> T:
-        return maysync(self.m_run)
-
-    @ta.final
-    async def a_run(self) -> T:
-        return await a_maysync(self.m_run)