ez-a-sync 0.32.12__cp311-cp311-musllinux_1_2_i686.whl → 0.32.14__cp311-cp311-musllinux_1_2_i686.whl

a_sync/_smart.pyx

@@ -34,12 +34,12 @@ cdef object get_event_loop = asyncio.get_event_loop
 cdef object AbstractEventLoop = asyncio.AbstractEventLoop
 cdef object Future = asyncio.Future
 cdef object Task = asyncio.Task
+cdef object CancelledError = asyncio.CancelledError
 cdef object InvalidStateError = asyncio.InvalidStateError
 cdef dict[object, object] _current_tasks = asyncio.tasks._current_tasks
 cdef object _future_init = asyncio.Future.__init__
 cdef object _get_loop = asyncio.futures._get_loop
 cdef object _task_init = asyncio.Task.__init__
-del asyncio
 
 # cdef logging
 cdef public object logger = getLogger(__name__)
@@ -510,6 +510,22 @@ class SmartTask(Task, Generic[T]):
         if _is_not_done(self):
             (<set>self._waiters).remove(waiter)
 
+    def _make_cancelled_error(self) -> asyncio.CancelledError:
+        # this function is not present in python3.8 so we're backporting it
+        """Create the CancelledError to raise if the Future is cancelled.
+
+        This should only be called once when handling a cancellation since
+        it erases the saved context exception value.
+        """
+        if self._cancel_message is None:
+            exc = CancelledError()
+        else:
+            exc = CancelledError(self._cancel_message)
+        exc.__context__ = self._cancelled_exc
+        # Remove the reference since we don't need this anymore.
+        self._cancelled_exc = None
+        return exc
+
 
 cdef object _SmartTask = SmartTask
 

a_sync/debugging.c

@@ -1517,12 +1517,13 @@ struct __pyx_defaults {
  * 
  * cdef class _ASyncGeneratorFunction:             # <<<<<<<<<<<<<<
  *     cdef readonly object _cache_handle
- *     cdef readonly object __weakself__
+ *     # must be public for functools.wraps
  */
 struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction {
   PyObject_HEAD
   struct __pyx_vtabstruct_6a_sync_4iter__ASyncGeneratorFunction *__pyx_vtab;
   PyObject *_cache_handle;
+  PyObject *__wrapped__;
   PyObject *__weakself__;
   PyObject *_ASyncGeneratorFunction__weakself_ptr;
 };
@@ -1609,7 +1610,7 @@ struct __pyx_obj_6a_sync_9debugging___pyx_scope_struct_3__stuck_debug_task {
  * 
  * cdef class _ASyncGeneratorFunction:             # <<<<<<<<<<<<<<
  *     cdef readonly object _cache_handle
- *     cdef readonly object __weakself__
+ *     # must be public for functools.wraps
  */
 
 struct __pyx_vtabstruct_6a_sync_4iter__ASyncGeneratorFunction {

a_sync/iter.c

@@ -1603,12 +1603,13 @@ struct __pyx_obj_6a_sync_9functools_cached_property_unsafe {
  * 
  * cdef class _ASyncGeneratorFunction:             # <<<<<<<<<<<<<<
  *     cdef readonly object _cache_handle
- *     cdef readonly object __weakself__
+ *     # must be public for functools.wraps
  */
 struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction {
   PyObject_HEAD
   struct __pyx_vtabstruct_6a_sync_4iter__ASyncGeneratorFunction *__pyx_vtab;
   PyObject *_cache_handle;
+  PyObject *__wrapped__;
   PyObject *__weakself__;
   PyObject *_ASyncGeneratorFunction__weakself_ptr;
 };
@@ -3705,6 +3706,9 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_6__get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self, PyObject *__pyx_v_instance, CYTHON_UNUSED PyObject *__pyx_v_owner); /* proto */
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_8__self_____get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_13_cache_handle___get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
+static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped_____get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
+static int __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___2__set__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self, PyObject *__pyx_v_value); /* proto */
+static int __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___4__del__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_12__weakself_____get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_8__reduce_cython__(CYTHON_UNUSED struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self); /* proto */
 static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_10__setstate_cython__(CYTHON_UNUSED struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self, CYTHON_UNUSED PyObject *__pyx_v___pyx_state); /* proto */
@@ -11498,7 +11502,11 @@ static int __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction___init__(struct __py
  * 
  */
   __Pyx_TraceLine(479,0,__PYX_ERR(0, 479, __pyx_L1_error))
-  if (__Pyx_PyObject_SetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_wrapped_2, __pyx_v_async_gen_func) < 0) __PYX_ERR(0, 479, __pyx_L1_error)
+  __Pyx_INCREF(__pyx_v_async_gen_func);
+  __Pyx_GIVEREF(__pyx_v_async_gen_func);
+  __Pyx_GOTREF(__pyx_v_self->__wrapped__);
+  __Pyx_DECREF(__pyx_v_self->__wrapped__);
+  __pyx_v_self->__wrapped__ = __pyx_v_async_gen_func;
 
   /* "a_sync/iter.pyx":482
  *         "The actual async generator function."
@@ -11696,8 +11704,7 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_2__repr__(stru
   PyObject *__pyx_t_3 = NULL;
   PyObject *__pyx_t_4 = NULL;
   PyObject *__pyx_t_5 = NULL;
-  PyObject *__pyx_t_6 = NULL;
-  unsigned int __pyx_t_7;
+  unsigned int __pyx_t_6;
   int __pyx_lineno = 0;
   const char *__pyx_filename = NULL;
   int __pyx_clineno = 0;
@@ -11727,17 +11734,6 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_2__repr__(stru
   __pyx_t_3 = __Pyx_PyObject_GetAttrStr(((PyObject *)Py_TYPE(((PyObject *)__pyx_v_self))), __pyx_n_s_name); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 494, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_3);
 
-  /* "a_sync/iter.pyx":495
- *         return "<{} for {} at {}>".format(
- *             type(self).__name__,
- *             self.__wrapped__,             # <<<<<<<<<<<<<<
- *             hex(id(self))
- *         )
- */
-  __Pyx_TraceLine(495,0,__PYX_ERR(0, 495, __pyx_L1_error))
-  __pyx_t_4 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_wrapped_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 495, __pyx_L1_error)
-  __Pyx_GOTREF(__pyx_t_4);
-
   /* "a_sync/iter.pyx":496
  *             type(self).__name__,
  *             self.__wrapped__,
@@ -11746,32 +11742,31 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_2__repr__(stru
  * 
  */
   __Pyx_TraceLine(496,0,__PYX_ERR(0, 496, __pyx_L1_error))
-  __pyx_t_5 = __Pyx_PyObject_CallOneArg(__pyx_builtin_id, ((PyObject *)__pyx_v_self)); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 496, __pyx_L1_error)
+  __pyx_t_4 = __Pyx_PyObject_CallOneArg(__pyx_builtin_id, ((PyObject *)__pyx_v_self)); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 496, __pyx_L1_error)
+  __Pyx_GOTREF(__pyx_t_4);
+  __pyx_t_5 = __Pyx_PyObject_CallOneArg(__pyx_builtin_hex, __pyx_t_4); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 496, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_5);
-  __pyx_t_6 = __Pyx_PyObject_CallOneArg(__pyx_builtin_hex, __pyx_t_5); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 496, __pyx_L1_error)
-  __Pyx_GOTREF(__pyx_t_6);
-  __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
-  __pyx_t_5 = NULL;
-  __pyx_t_7 = 0;
+  __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
+  __pyx_t_4 = NULL;
+  __pyx_t_6 = 0;
   #if CYTHON_UNPACK_METHODS
   if (likely(PyMethod_Check(__pyx_t_2))) {
-    __pyx_t_5 = PyMethod_GET_SELF(__pyx_t_2);
-    if (likely(__pyx_t_5)) {
+    __pyx_t_4 = PyMethod_GET_SELF(__pyx_t_2);
+    if (likely(__pyx_t_4)) {
       PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_2);
-      __Pyx_INCREF(__pyx_t_5);
+      __Pyx_INCREF(__pyx_t_4);
       __Pyx_INCREF(function);
       __Pyx_DECREF_SET(__pyx_t_2, function);
-      __pyx_t_7 = 1;
+      __pyx_t_6 = 1;
     }
   }
   #endif
   {
-    PyObject *__pyx_callargs[4] = {__pyx_t_5, __pyx_t_3, __pyx_t_4, __pyx_t_6};
-    __pyx_t_1 = __Pyx_PyObject_FastCall(__pyx_t_2, __pyx_callargs+1-__pyx_t_7, 3+__pyx_t_7);
-    __Pyx_XDECREF(__pyx_t_5); __pyx_t_5 = 0;
+    PyObject *__pyx_callargs[4] = {__pyx_t_4, __pyx_t_3, __pyx_v_self->__wrapped__, __pyx_t_5};
+    __pyx_t_1 = __Pyx_PyObject_FastCall(__pyx_t_2, __pyx_callargs+1-__pyx_t_6, 3+__pyx_t_6);
+    __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
     __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
-    __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
-    __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
+    __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
     if (unlikely(!__pyx_t_1)) __PYX_ERR(0, 493, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_1);
     __Pyx_DECREF(__pyx_t_2); __pyx_t_2 = 0;
@@ -11795,7 +11790,6 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_2__repr__(stru
   __Pyx_XDECREF(__pyx_t_3);
   __Pyx_XDECREF(__pyx_t_4);
   __Pyx_XDECREF(__pyx_t_5);
-  __Pyx_XDECREF(__pyx_t_6);
   __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__repr__", __pyx_clineno, __pyx_lineno, __pyx_filename);
   __pyx_r = NULL;
   __pyx_L0:;
@@ -11863,9 +11857,8 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
   PyObject *__pyx_t_3 = NULL;
   PyObject *__pyx_t_4 = NULL;
   PyObject *__pyx_t_5 = NULL;
-  PyObject *__pyx_t_6 = NULL;
-  unsigned int __pyx_t_7;
-  PyObject *__pyx_t_8 = NULL;
+  unsigned int __pyx_t_6;
+  PyObject *__pyx_t_7 = NULL;
   int __pyx_lineno = 0;
   const char *__pyx_filename = NULL;
   int __pyx_clineno = 0;
@@ -11894,33 +11887,30 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
     __Pyx_XDECREF(__pyx_r);
     __Pyx_GetModuleGlobalName(__pyx_t_3, __pyx_n_s_ASyncIterator); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 507, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_3);
-    __pyx_t_4 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_wrapped_2); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 507, __pyx_L1_error)
+    __pyx_t_4 = PyDict_Copy(__pyx_v_kwargs); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 507, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_4);
-    __pyx_t_5 = PyDict_Copy(__pyx_v_kwargs); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 507, __pyx_L1_error)
+    __pyx_t_5 = __Pyx_PyObject_Call(__pyx_v_self->__wrapped__, __pyx_v_args, __pyx_t_4); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 507, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_5);
-    __pyx_t_6 = __Pyx_PyObject_Call(__pyx_t_4, __pyx_v_args, __pyx_t_5); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 507, __pyx_L1_error)
-    __Pyx_GOTREF(__pyx_t_6);
     __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
-    __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
-    __pyx_t_5 = NULL;
-    __pyx_t_7 = 0;
+    __pyx_t_4 = NULL;
+    __pyx_t_6 = 0;
     #if CYTHON_UNPACK_METHODS
     if (unlikely(PyMethod_Check(__pyx_t_3))) {
-      __pyx_t_5 = PyMethod_GET_SELF(__pyx_t_3);
-      if (likely(__pyx_t_5)) {
+      __pyx_t_4 = PyMethod_GET_SELF(__pyx_t_3);
+      if (likely(__pyx_t_4)) {
         PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_3);
-        __Pyx_INCREF(__pyx_t_5);
+        __Pyx_INCREF(__pyx_t_4);
         __Pyx_INCREF(function);
         __Pyx_DECREF_SET(__pyx_t_3, function);
-        __pyx_t_7 = 1;
+        __pyx_t_6 = 1;
       }
     }
     #endif
     {
-      PyObject *__pyx_callargs[2] = {__pyx_t_5, __pyx_t_6};
-      __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_7, 1+__pyx_t_7);
-      __Pyx_XDECREF(__pyx_t_5); __pyx_t_5 = 0;
-      __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
+      PyObject *__pyx_callargs[2] = {__pyx_t_4, __pyx_t_5};
+      __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_6, 1+__pyx_t_6);
+      __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
+      __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
       if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 507, __pyx_L1_error)
       __Pyx_GOTREF(__pyx_t_2);
       __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
@@ -11949,8 +11939,6 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
   __Pyx_XDECREF(__pyx_r);
   __Pyx_GetModuleGlobalName(__pyx_t_3, __pyx_n_s_ASyncIterator); if (unlikely(!__pyx_t_3)) __PYX_ERR(0, 508, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_3);
-  __pyx_t_6 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_wrapped_2); if (unlikely(!__pyx_t_6)) __PYX_ERR(0, 508, __pyx_L1_error)
-  __Pyx_GOTREF(__pyx_t_6);
   __pyx_t_5 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_self_2); if (unlikely(!__pyx_t_5)) __PYX_ERR(0, 508, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_5);
   __pyx_t_4 = PyTuple_New(1); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 508, __pyx_L1_error)
@@ -11963,13 +11951,12 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   __pyx_t_4 = PyDict_Copy(__pyx_v_kwargs); if (unlikely(!__pyx_t_4)) __PYX_ERR(0, 508, __pyx_L1_error)
   __Pyx_GOTREF(__pyx_t_4);
-  __pyx_t_8 = __Pyx_PyObject_Call(__pyx_t_6, __pyx_t_5, __pyx_t_4); if (unlikely(!__pyx_t_8)) __PYX_ERR(0, 508, __pyx_L1_error)
-  __Pyx_GOTREF(__pyx_t_8);
-  __Pyx_DECREF(__pyx_t_6); __pyx_t_6 = 0;
+  __pyx_t_7 = __Pyx_PyObject_Call(__pyx_v_self->__wrapped__, __pyx_t_5, __pyx_t_4); if (unlikely(!__pyx_t_7)) __PYX_ERR(0, 508, __pyx_L1_error)
+  __Pyx_GOTREF(__pyx_t_7);
   __Pyx_DECREF(__pyx_t_5); __pyx_t_5 = 0;
   __Pyx_DECREF(__pyx_t_4); __pyx_t_4 = 0;
   __pyx_t_4 = NULL;
-  __pyx_t_7 = 0;
+  __pyx_t_6 = 0;
   #if CYTHON_UNPACK_METHODS
   if (unlikely(PyMethod_Check(__pyx_t_3))) {
     __pyx_t_4 = PyMethod_GET_SELF(__pyx_t_3);
@@ -11978,15 +11965,15 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
       __Pyx_INCREF(__pyx_t_4);
       __Pyx_INCREF(function);
       __Pyx_DECREF_SET(__pyx_t_3, function);
-      __pyx_t_7 = 1;
+      __pyx_t_6 = 1;
     }
   }
   #endif
   {
-    PyObject *__pyx_callargs[2] = {__pyx_t_4, __pyx_t_8};
-    __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_7, 1+__pyx_t_7);
+    PyObject *__pyx_callargs[2] = {__pyx_t_4, __pyx_t_7};
+    __pyx_t_2 = __Pyx_PyObject_FastCall(__pyx_t_3, __pyx_callargs+1-__pyx_t_6, 1+__pyx_t_6);
     __Pyx_XDECREF(__pyx_t_4); __pyx_t_4 = 0;
-    __Pyx_DECREF(__pyx_t_8); __pyx_t_8 = 0;
+    __Pyx_DECREF(__pyx_t_7); __pyx_t_7 = 0;
     if (unlikely(!__pyx_t_2)) __PYX_ERR(0, 508, __pyx_L1_error)
     __Pyx_GOTREF(__pyx_t_2);
     __Pyx_DECREF(__pyx_t_3); __pyx_t_3 = 0;
@@ -12009,8 +11996,7 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_4__call__(stru
   __Pyx_XDECREF(__pyx_t_3);
   __Pyx_XDECREF(__pyx_t_4);
   __Pyx_XDECREF(__pyx_t_5);
-  __Pyx_XDECREF(__pyx_t_6);
-  __Pyx_XDECREF(__pyx_t_8);
+  __Pyx_XDECREF(__pyx_t_7);
   __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__call__", __pyx_clineno, __pyx_lineno, __pyx_filename);
   __pyx_r = NULL;
   __pyx_L0:;
@@ -12064,8 +12050,7 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_6__get__(struc
   PyObject *__pyx_t_9 = NULL;
   PyObject *__pyx_t_10 = NULL;
   PyObject *__pyx_t_11 = NULL;
-  PyObject *__pyx_t_12 = NULL;
-  unsigned int __pyx_t_13;
+  unsigned int __pyx_t_12;
   int __pyx_lineno = 0;
   const char *__pyx_filename = NULL;
   int __pyx_clineno = 0;
@@ -12199,27 +12184,24 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_6__get__(struc
       __Pyx_TraceLine(523,0,__PYX_ERR(0, 523, __pyx_L6_except_error))
       __Pyx_GetModuleGlobalName(__pyx_t_10, __pyx_n_s_ASyncGeneratorFunction); if (unlikely(!__pyx_t_10)) __PYX_ERR(0, 523, __pyx_L6_except_error)
       __Pyx_GOTREF(__pyx_t_10);
-      __pyx_t_11 = __Pyx_PyObject_GetAttrStr(((PyObject *)__pyx_v_self), __pyx_n_s_wrapped_2); if (unlikely(!__pyx_t_11)) __PYX_ERR(0, 523, __pyx_L6_except_error)
-      __Pyx_GOTREF(__pyx_t_11);
-      __pyx_t_12 = NULL;
-      __pyx_t_13 = 0;
+      __pyx_t_11 = NULL;
+      __pyx_t_12 = 0;
       #if CYTHON_UNPACK_METHODS
       if (unlikely(PyMethod_Check(__pyx_t_10))) {
-        __pyx_t_12 = PyMethod_GET_SELF(__pyx_t_10);
-        if (likely(__pyx_t_12)) {
+        __pyx_t_11 = PyMethod_GET_SELF(__pyx_t_10);
+        if (likely(__pyx_t_11)) {
           PyObject* function = PyMethod_GET_FUNCTION(__pyx_t_10);
-          __Pyx_INCREF(__pyx_t_12);
+          __Pyx_INCREF(__pyx_t_11);
           __Pyx_INCREF(function);
           __Pyx_DECREF_SET(__pyx_t_10, function);
-          __pyx_t_13 = 1;
+          __pyx_t_12 = 1;
         }
       }
       #endif
       {
-        PyObject *__pyx_callargs[3] = {__pyx_t_12, __pyx_t_11, __pyx_v_instance};
-        __pyx_t_9 = __Pyx_PyObject_FastCall(__pyx_t_10, __pyx_callargs+1-__pyx_t_13, 2+__pyx_t_13);
-        __Pyx_XDECREF(__pyx_t_12); __pyx_t_12 = 0;
-        __Pyx_DECREF(__pyx_t_11); __pyx_t_11 = 0;
+        PyObject *__pyx_callargs[3] = {__pyx_t_11, __pyx_v_self->__wrapped__, __pyx_v_instance};
+        __pyx_t_9 = __Pyx_PyObject_FastCall(__pyx_t_10, __pyx_callargs+1-__pyx_t_12, 2+__pyx_t_12);
+        __Pyx_XDECREF(__pyx_t_11); __pyx_t_11 = 0;
         if (unlikely(!__pyx_t_9)) __PYX_ERR(0, 523, __pyx_L6_except_error)
         __Pyx_GOTREF(__pyx_t_9);
         __Pyx_DECREF(__pyx_t_10); __pyx_t_10 = 0;
@@ -12314,7 +12296,6 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_6__get__(struc
   __Pyx_XDECREF(__pyx_t_9);
   __Pyx_XDECREF(__pyx_t_10);
   __Pyx_XDECREF(__pyx_t_11);
-  __Pyx_XDECREF(__pyx_t_12);
   __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
   __pyx_r = NULL;
   __pyx_L0:;
@@ -12766,8 +12747,8 @@ static void __pyx_f_6a_sync_4iter_23_ASyncGeneratorFunction__ASyncGeneratorFunct
  * 
  * cdef class _ASyncGeneratorFunction:
  *     cdef readonly object _cache_handle             # <<<<<<<<<<<<<<
- *     cdef readonly object __weakself__
- *     cdef PyObject *__weakself_ptr
+ *     # must be public for functools.wraps
+ *     cdef public object __wrapped__
  */
 
 /* Python wrapper */
@@ -12810,9 +12791,141 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_13_cache_handl
   return __pyx_r;
 }
 
-/* "a_sync/iter.pxd":5
- * cdef class _ASyncGeneratorFunction:
+/* "a_sync/iter.pxd":6
  *     cdef readonly object _cache_handle
+ *     # must be public for functools.wraps
+ *     cdef public object __wrapped__             # <<<<<<<<<<<<<<
+ *     cdef readonly object __weakself__
+ *     cdef PyObject *__weakself_ptr
+ */
+
+/* Python wrapper */
+static PyObject *__pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___1__get__(PyObject *__pyx_v_self); /*proto*/
+static PyObject *__pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___1__get__(PyObject *__pyx_v_self) {
+  CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
+  PyObject *__pyx_r = 0;
+  __Pyx_RefNannyDeclarations
+  __Pyx_RefNannySetupContext("__get__ (wrapper)", 0);
+  __pyx_kwvalues = __Pyx_KwValues_VARARGS(__pyx_args, __pyx_nargs);
+  __pyx_r = __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped_____get__(((struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *)__pyx_v_self));
+
+  /* function exit code */
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped_____get__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self) {
+  PyObject *__pyx_r = NULL;
+  __Pyx_TraceDeclarations
+  __Pyx_RefNannyDeclarations
+  int __pyx_lineno = 0;
+  const char *__pyx_filename = NULL;
+  int __pyx_clineno = 0;
+  __Pyx_RefNannySetupContext("__get__", 1);
+  __Pyx_TraceCall("__get__", __pyx_f[2], 6, 0, __PYX_ERR(2, 6, __pyx_L1_error));
+  __Pyx_XDECREF(__pyx_r);
+  __Pyx_INCREF(__pyx_v_self->__wrapped__);
+  __pyx_r = __pyx_v_self->__wrapped__;
+  goto __pyx_L0;
+
+  /* function exit code */
+  __pyx_L1_error:;
+  __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__wrapped__.__get__", __pyx_clineno, __pyx_lineno, __pyx_filename);
+  __pyx_r = NULL;
+  __pyx_L0:;
+  __Pyx_XGIVEREF(__pyx_r);
+  __Pyx_TraceReturn(__pyx_r, 0);
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+/* Python wrapper */
+static int __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value); /*proto*/
+static int __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___3__set__(PyObject *__pyx_v_self, PyObject *__pyx_v_value) {
+  CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
+  int __pyx_r;
+  __Pyx_RefNannyDeclarations
+  __Pyx_RefNannySetupContext("__set__ (wrapper)", 0);
+  __pyx_kwvalues = __Pyx_KwValues_VARARGS(__pyx_args, __pyx_nargs);
+  __pyx_r = __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___2__set__(((struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *)__pyx_v_self), ((PyObject *)__pyx_v_value));
+
+  /* function exit code */
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+static int __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___2__set__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self, PyObject *__pyx_v_value) {
+  int __pyx_r;
+  __Pyx_TraceDeclarations
+  __Pyx_RefNannyDeclarations
+  int __pyx_lineno = 0;
+  const char *__pyx_filename = NULL;
+  int __pyx_clineno = 0;
+  __Pyx_RefNannySetupContext("__set__", 1);
+  __Pyx_TraceCall("__set__", __pyx_f[2], 6, 0, __PYX_ERR(2, 6, __pyx_L1_error));
+  __Pyx_INCREF(__pyx_v_value);
+  __Pyx_GIVEREF(__pyx_v_value);
+  __Pyx_GOTREF(__pyx_v_self->__wrapped__);
+  __Pyx_DECREF(__pyx_v_self->__wrapped__);
+  __pyx_v_self->__wrapped__ = __pyx_v_value;
+
+  /* function exit code */
+  __pyx_r = 0;
+  goto __pyx_L0;
+  __pyx_L1_error:;
+  __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__wrapped__.__set__", __pyx_clineno, __pyx_lineno, __pyx_filename);
+  __pyx_r = -1;
+  __pyx_L0:;
+  __Pyx_TraceReturn(Py_None, 0);
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+/* Python wrapper */
+static int __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___5__del__(PyObject *__pyx_v_self); /*proto*/
+static int __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___5__del__(PyObject *__pyx_v_self) {
+  CYTHON_UNUSED PyObject *const *__pyx_kwvalues;
+  int __pyx_r;
+  __Pyx_RefNannyDeclarations
+  __Pyx_RefNannySetupContext("__del__ (wrapper)", 0);
+  __pyx_kwvalues = __Pyx_KwValues_VARARGS(__pyx_args, __pyx_nargs);
+  __pyx_r = __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___4__del__(((struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *)__pyx_v_self));
+
+  /* function exit code */
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+static int __pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___4__del__(struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *__pyx_v_self) {
+  int __pyx_r;
+  __Pyx_TraceDeclarations
+  __Pyx_RefNannyDeclarations
+  int __pyx_lineno = 0;
+  const char *__pyx_filename = NULL;
+  int __pyx_clineno = 0;
+  __Pyx_RefNannySetupContext("__del__", 1);
+  __Pyx_TraceCall("__del__", __pyx_f[2], 6, 0, __PYX_ERR(2, 6, __pyx_L1_error));
+  __Pyx_INCREF(Py_None);
+  __Pyx_GIVEREF(Py_None);
+  __Pyx_GOTREF(__pyx_v_self->__wrapped__);
+  __Pyx_DECREF(__pyx_v_self->__wrapped__);
+  __pyx_v_self->__wrapped__ = Py_None;
+
+  /* function exit code */
+  __pyx_r = 0;
+  goto __pyx_L0;
+  __pyx_L1_error:;
+  __Pyx_AddTraceback("a_sync.iter._ASyncGeneratorFunction.__wrapped__.__del__", __pyx_clineno, __pyx_lineno, __pyx_filename);
+  __pyx_r = -1;
+  __pyx_L0:;
+  __Pyx_TraceReturn(Py_None, 0);
+  __Pyx_RefNannyFinishContext();
+  return __pyx_r;
+}
+
+/* "a_sync/iter.pxd":7
+ *     # must be public for functools.wraps
+ *     cdef public object __wrapped__
  *     cdef readonly object __weakself__             # <<<<<<<<<<<<<<
  *     cdef PyObject *__weakself_ptr
  *     cdef inline void _set_cache_handle(self, object handle)
@@ -12841,7 +12954,7 @@ static PyObject *__pyx_pf_6a_sync_4iter_23_ASyncGeneratorFunction_12__weakself__
   const char *__pyx_filename = NULL;
   int __pyx_clineno = 0;
   __Pyx_RefNannySetupContext("__get__", 1);
-  __Pyx_TraceCall("__get__", __pyx_f[2], 5, 0, __PYX_ERR(2, 5, __pyx_L1_error));
+  __Pyx_TraceCall("__get__", __pyx_f[2], 7, 0, __PYX_ERR(2, 7, __pyx_L1_error));
   __Pyx_XDECREF(__pyx_r);
   __Pyx_INCREF(__pyx_v_self->__weakself__);
   __pyx_r = __pyx_v_self->__weakself__;
@@ -21572,6 +21685,7 @@ static PyObject *__pyx_tp_new_6a_sync_4iter__ASyncGeneratorFunction(PyTypeObject
   p = ((struct __pyx_obj_6a_sync_4iter__ASyncGeneratorFunction *)o);
   p->__pyx_vtab = __pyx_vtabptr_6a_sync_4iter__ASyncGeneratorFunction;
   p->_cache_handle = Py_None; Py_INCREF(Py_None);
+  p->__wrapped__ = Py_None; Py_INCREF(Py_None);
   p->__weakself__ = Py_None; Py_INCREF(Py_None);
   return o;
 }
@@ -21587,6 +21701,7 @@ static void __pyx_tp_dealloc_6a_sync_4iter__ASyncGeneratorFunction(PyObject *o)
   #endif
   PyObject_GC_UnTrack(o);
   Py_CLEAR(p->_cache_handle);
+  Py_CLEAR(p->__wrapped__);
   Py_CLEAR(p->__weakself__);
   #if CYTHON_USE_TYPE_SLOTS || CYTHON_COMPILING_IN_PYPY
   (*Py_TYPE(o)->tp_free)(o);
@@ -21604,6 +21719,9 @@ static int __pyx_tp_traverse_6a_sync_4iter__ASyncGeneratorFunction(PyObject *o,
   if (p->_cache_handle) {
     e = (*v)(p->_cache_handle, a); if (e) return e;
   }
+  if (p->__wrapped__) {
+    e = (*v)(p->__wrapped__, a); if (e) return e;
+  }
   if (p->__weakself__) {
     e = (*v)(p->__weakself__, a); if (e) return e;
   }
@@ -21616,6 +21734,9 @@ static int __pyx_tp_clear_6a_sync_4iter__ASyncGeneratorFunction(PyObject *o) {
   tmp = ((PyObject*)p->_cache_handle);
   p->_cache_handle = Py_None; Py_INCREF(Py_None);
   Py_XDECREF(tmp);
+  tmp = ((PyObject*)p->__wrapped__);
+  p->__wrapped__ = Py_None; Py_INCREF(Py_None);
+  Py_XDECREF(tmp);
   tmp = ((PyObject*)p->__weakself__);
   p->__weakself__ = Py_None; Py_INCREF(Py_None);
   Py_XDECREF(tmp);
@@ -21638,6 +21759,19 @@ static PyObject *__pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction__cache_ha
   return __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_13_cache_handle_1__get__(o);
 }
 
+static PyObject *__pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction___wrapped__(PyObject *o, CYTHON_UNUSED void *x) {
+  return __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___1__get__(o);
+}
+
+static int __pyx_setprop_6a_sync_4iter_23_ASyncGeneratorFunction___wrapped__(PyObject *o, PyObject *v, CYTHON_UNUSED void *x) {
+  if (v) {
+    return __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___3__set__(o, v);
+  }
+  else {
+    return __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_11__wrapped___5__del__(o);
+  }
+}
+
 static PyObject *__pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction___weakself__(PyObject *o, CYTHON_UNUSED void *x) {
   return __pyx_pw_6a_sync_4iter_23_ASyncGeneratorFunction_12__weakself___1__get__(o);
 }
@@ -21656,6 +21790,7 @@ static PyMethodDef __pyx_methods_6a_sync_4iter__ASyncGeneratorFunction[] = {
 static struct PyGetSetDef __pyx_getsets_6a_sync_4iter__ASyncGeneratorFunction[] = {
   {(char *)"__self__", __pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction___self__, 0, (char *)PyDoc_STR("_ASyncGeneratorFunction.__self__: object"), 0},
   {(char *)"_cache_handle", __pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction__cache_handle, 0, (char *)0, 0},
+  {(char *)"__wrapped__", __pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction___wrapped__, __pyx_setprop_6a_sync_4iter_23_ASyncGeneratorFunction___wrapped__, (char *)PyDoc_STR("__wrapped__: object"), 0},
   {(char *)"__weakself__", __pyx_getprop_6a_sync_4iter_23_ASyncGeneratorFunction___weakself__, 0, (char *)0, 0},
   {0, 0, 0, 0, 0}
 };

a_sync/iter.pxd

@@ -2,6 +2,8 @@ from cpython.object cimport PyObject
 
 cdef class _ASyncGeneratorFunction:
     cdef readonly object _cache_handle
+    # must be public for functools.wraps
+    cdef public object __wrapped__
     cdef readonly object __weakself__
     cdef PyObject *__weakself_ptr
     cdef inline void _set_cache_handle(self, object handle)

a_sync/task.py

@@ -43,6 +43,9 @@ logger = getLogger(__name__)
 MappingFn = Callable[Concatenate[K, P], Awaitable[V]]
 
 
+_args = WeakKeyDictionary()
+
+
 class TaskMapping(DefaultDict[K, "Task[V]"], AsyncIterable[Tuple[K, V]]):
     """
     A mapping of keys to asynchronous tasks with additional functionality.
@@ -157,6 +160,9 @@ class TaskMapping(DefaultDict[K, "Task[V]"], AsyncIterable[Tuple[K, V]]):
 
         if iterables:
 
+            set_next = self._next.set
+            clear_next = self._next.clear
+
             @wraps(wrapped_func)
             async def _wrapped_set_next(
                 *args: P.args, __a_sync_recursion: int = 0, **kwargs: P.kwargs
@@ -167,17 +173,27 @@ class TaskMapping(DefaultDict[K, "Task[V]"], AsyncIterable[Tuple[K, V]]):
                     e.args = *e.args, f"wrapped:{self.__wrapped__}"
                     raise
                 except TypeError as e:
-                    if __a_sync_recursion > 2 or not (
-                        str(e).startswith(wrapped_func.__name__)
-                        and "got multiple values for argument" in str(e)
+                    if (
+                        args is None
+                        or __a_sync_recursion > 2
+                        or not (
+                            str(e).startswith(wrapped_func.__name__)
+                            and "got multiple values for argument" in str(e)
+                        )
                     ):
                         raise
+
                     # NOTE: args ordering is clashing with provided kwargs. We can handle this in a hacky way.
                     # TODO: perform this check earlier and pre-prepare the args/kwargs ordering
+                    try:
+                        argspec = _args[self.__wrapped__]
+                    except KeyError:
+                        argspec = _args[self.__wrapped__] = getfullargspec(self.__wrapped__).args
+
                     new_args = list(args)
                     new_kwargs = dict(kwargs)
                     try:
-                        for i, arg in enumerate(getfullargspec(self.__wrapped__).args):
+                        for i, arg in enumerate(argspec):
                             if arg in kwargs:
                                 new_args.insert(i, new_kwargs.pop(arg))
                             else:
@@ -194,8 +210,8 @@ class TaskMapping(DefaultDict[K, "Task[V]"], AsyncIterable[Tuple[K, V]]):
                             else e2.with_traceback(e2.__traceback__)
                         )
                 finally:
-                    self._next.set()
-                    self._next.clear()
+                    set_next()
+                    clear_next()
 
             self._wrapped_func = _wrapped_set_next
             init_loader_queue: Queue[Tuple[K, "Future[V]"]] = Queue()

{ez_a_sync-0.32.12.dist-info → ez_a_sync-0.32.14.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ez_a_sync
-Version: 0.32.12
+Version: 0.32.14
 Summary: A library that makes it easy to define objects that can be used for both sync and async use cases.
 Home-page: https://github.com/BobTheBuidler/a-sync
 Author: BobTheBuidler