omlish-cext 0.0.0.dev474__tar.gz → 0.0.0.dev476__tar.gz

{omlish_cext-0.0.0.dev474/omlish_cext.egg-info → omlish_cext-0.0.0.dev476}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omlish-cext
-Version: 0.0.0.dev474
+Version: 0.0.0.dev476
 Summary: omlish
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev474
+Requires-Dist: omlish==0.0.0.dev476
 Dynamic: license-file
 
 # Overview

omlish_cext-0.0.0.dev476/omlish/lang/_asyncs.cc

@@ -0,0 +1,186 @@
+// @omlish-cext
+#define PY_SSIZE_T_CLEAN
+#include <Python.h>
+
+//
+
+#define _MODULE_NAME "_asyncs"
+#define _PACKAGE_NAME "omlish.lang"
+#define _MODULE_FULL_NAME _PACKAGE_NAME "." _MODULE_NAME
+
+//
+
+typedef struct {
+    PyObject *SyncAwaitCoroutineNotTerminatedError;
+    PyObject *str_close;
+    PyObject *str___await__;
+} module_state;
+
+static module_state *
+get_module_state(PyObject *module)
+{
+    return (module_state *) PyModule_GetState(module);
+}
+
+//
+
+static void
+suppress_close(PyObject *iter, module_state *state)
+{
+    PyObject *res = PyObject_CallMethodNoArgs(iter, state->str_close);
+    if (!res) {
+        PyErr_Clear();
+    } else {
+        Py_DECREF(res);
+    }
+}
+
+static PyObject *
+sync_await(PyObject *module, PyObject *aw)
+{
+    module_state *state = get_module_state(module);
+    PyObject *await_meth = NULL;
+    PyObject *iter = NULL;
+    PyObject *result = NULL;
+    PySendResult sres;
+
+    await_meth = PyObject_GetAttr(aw, state->str___await__);
+    if (await_meth == NULL) {
+        if (PyErr_ExceptionMatches(PyExc_AttributeError)) {
+            PyErr_SetString(PyExc_TypeError, "object is not awaitable (no __await__)");
+        }
+        goto error;
+    }
+
+    iter = PyObject_CallNoArgs(await_meth);
+    Py_DECREF(await_meth);
+    await_meth = NULL;
+
+    if (iter == NULL) {
+        goto error;
+    }
+
+    if (!PyIter_Check(iter)) {
+        Py_DECREF(iter);
+        PyErr_SetString(PyExc_TypeError, "__await__() must return an iterator");
+        goto error;
+    }
+
+    sres = PyIter_Send(iter, Py_None, &result);
+    if (sres == PYGEN_ERROR) {
+        Py_DECREF(iter);
+        goto error;
+    }
+    if (sres == PYGEN_NEXT) {
+        Py_XDECREF(result);
+        suppress_close(iter, state);
+        Py_DECREF(iter);
+        PyErr_SetString(state->SyncAwaitCoroutineNotTerminatedError, "Not terminated");
+        goto error;
+    }
+
+    Py_DECREF(iter);
+    return result;
+
+error:
+    return NULL;
+}
+
+//
+
+static PyMethodDef mod_methods[] = {
+    {"sync_await", sync_await, METH_O, "sync_await"},
+    {NULL, NULL, 0, NULL}
+};
+
+static int
+module_traverse(PyObject *module, visitproc visit, void *arg)
+{
+    module_state *state = get_module_state(module);
+    Py_VISIT(state->SyncAwaitCoroutineNotTerminatedError);
+    Py_VISIT(state->str_close);
+    Py_VISIT(state->str___await__);
+    return 0;
+}
+
+static int
+module_clear(PyObject *module)
+{
+    module_state *state = get_module_state(module);
+    Py_CLEAR(state->SyncAwaitCoroutineNotTerminatedError);
+    Py_CLEAR(state->str_close);
+    Py_CLEAR(state->str___await__);
+    return 0;
+}
+
+static void
+module_free(void *module)
+{
+    module_clear((PyObject *) module);
+}
+
+static int
+module_exec(PyObject *module)
+{
+    module_state *state = get_module_state(module);
+
+    state->str_close = PyUnicode_InternFromString("close");
+    if (!state->str_close) {
+        return -1;
+    }
+
+    state->str___await__ = PyUnicode_InternFromString("__await__");
+    if (!state->str___await__) {
+        return -1;
+    }
+
+    PyObject *asyncs_module = PyImport_ImportModule("omlish.lite.asyncs");
+    if (!asyncs_module) {
+        return -1;
+    }
+
+    state->SyncAwaitCoroutineNotTerminatedError = PyObject_GetAttrString(
+        asyncs_module,
+        "SyncAwaitCoroutineNotTerminatedError"
+    );
+    Py_DECREF(asyncs_module);
+
+    if (!state->SyncAwaitCoroutineNotTerminatedError) {
+        return -1;
+    }
+
+    return 0;
+}
+
+//
+
+PyDoc_STRVAR(module_doc, _MODULE_NAME);
+
+static PyModuleDef_Slot module_slots[] = {
+    {Py_mod_exec, (void *) module_exec},
+    {Py_mod_gil, Py_MOD_GIL_NOT_USED},
+    {Py_mod_multiple_interpreters, Py_MOD_PER_INTERPRETER_GIL_SUPPORTED},
+    {0, NULL}
+};
+
+static PyModuleDef module_def = {
+    .m_base = PyModuleDef_HEAD_INIT,
+    .m_name = _MODULE_NAME,
+    .m_doc = module_doc,
+    .m_size = sizeof(module_state),
+    .m_methods = mod_methods,
+    .m_slots = module_slots,
+    .m_traverse = module_traverse,
+    .m_clear = module_clear,
+    .m_free = module_free,
+};
+
+extern "C" {
+
+PyMODINIT_FUNC
+PyInit__asyncs(void)
+{
+    return PyModuleDef_Init(&module_def);
+}
+
+}

{omlish_cext-0.0.0.dev474 → omlish_cext-0.0.0.dev476}/omlish/lang/imports/_capture.cc

@@ -11,7 +11,7 @@
 #undef Py_BUILD_CORE
 
 #if PY_VERSION_HEX < 0x030D0000
-#  error "This extension requires CPython 3.13+"
+#error "This extension requires CPython 3.13+"
 #endif
 
 //
@@ -74,13 +74,13 @@ static PyMethodDef capture_methods[] = {
     {NULL, NULL, 0, NULL}
 };
 
-static struct PyModuleDef_Slot capture_slots[] = {
+static PyModuleDef_Slot capture_slots[] = {
     {Py_mod_gil, Py_MOD_GIL_NOT_USED},
     {Py_mod_multiple_interpreters, Py_MOD_PER_INTERPRETER_GIL_SUPPORTED},
     {0, NULL}
 };
 
-static struct PyModuleDef capture_module = {
+static PyModuleDef capture_module = {
     .m_base = PyModuleDef_HEAD_INIT,
     .m_name = _MODULE_NAME,
     .m_doc = capture_doc,
@@ -94,7 +94,8 @@ static struct PyModuleDef capture_module = {
 
 extern "C" {
 
-PyMODINIT_FUNC PyInit__capture(void)
+PyMODINIT_FUNC
+PyInit__capture(void)
 {
     return PyModuleDef_Init(&capture_module);
 }

{omlish_cext-0.0.0.dev474 → omlish_cext-0.0.0.dev476/omlish_cext.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: omlish-cext
-Version: 0.0.0.dev474
+Version: 0.0.0.dev476
 Summary: omlish
 Author: wrmsr
 License-Expression: BSD-3-Clause
@@ -14,7 +14,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
 License-File: LICENSE
-Requires-Dist: omlish==0.0.0.dev474
+Requires-Dist: omlish==0.0.0.dev476
 Dynamic: license-file
 
 # Overview

{omlish_cext-0.0.0.dev474 → omlish_cext-0.0.0.dev476}/omlish_cext.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 pyproject.toml
 setup.py
+omlish/lang/_asyncs.cc
 omlish/lang/imports/_capture.cc
 omlish_cext.egg-info/PKG-INFO
 omlish_cext.egg-info/SOURCES.txt

omlish_cext-0.0.0.dev476/omlish_cext.egg-info/requires.txt

@@ -0,0 +1 @@
+omlish==0.0.0.dev476

{omlish_cext-0.0.0.dev474 → omlish_cext-0.0.0.dev476}/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.dev474'
+version = '0.0.0.dev476'
 classifiers = [
     'Development Status :: 2 - Pre-Alpha',
     'Intended Audience :: Developers',
@@ -24,7 +24,7 @@ classifiers = [
 ]
 description = 'omlish'
 dependencies = [
-    'omlish == 0.0.0.dev474',
+    'omlish == 0.0.0.dev476',
 ]
 
 [tool.setuptools]

{omlish_cext-0.0.0.dev474 → omlish_cext-0.0.0.dev476}/setup.py

@@ -3,6 +3,11 @@ import setuptools as st
 
 st.setup(
     ext_modules=[
+        st.Extension(
+            name='omlish.lang._asyncs',
+            sources=['omlish/lang/_asyncs.cc'],
+            extra_compile_args=['-std=c++20'],
+        ),
         st.Extension(
             name='omlish.lang.imports._capture',
             sources=['omlish/lang/imports/_capture.cc'],

omlish_cext-0.0.0.dev474/omlish_cext.egg-info/requires.txt

@@ -1 +0,0 @@
-omlish==0.0.0.dev474