ez-a-sync 0.32.10__cp311-cp311-win32.whl → 0.32.11__cp311-cp311-win32.whl

a_sync/_smart.pyx

@@ -9,14 +9,15 @@ import asyncio
 import typing
 import weakref
 from logging import getLogger
+from types import TracebackType
 
 cimport cython
 from cpython.object cimport PyObject
 from cpython.ref cimport Py_DECREF, Py_INCREF
 
-from a_sync._typing import *
+from a_sync._typing import T
 
-if TYPE_CHECKING:
+if typing.TYPE_CHECKING:
     from a_sync import SmartProcessingQueue
 
 cdef extern from "weakrefobject.h":
@@ -50,6 +51,7 @@ del getLogger
 cdef object Any = typing.Any
 cdef object Generic = typing.Generic
 cdef object Tuple = typing.Tuple
+cdef object Union = typing.Union
 del typing
 
 
@@ -173,7 +175,7 @@ cdef inline bint _is_cancelled(fut: Future):
 
 
 @cython.linetrace(False)
-cdef object _get_result(fut: Future):
+cdef object _get_result(fut: Union["SmartFuture", "SmartTask"]):
     """Return the result this future represents.
 
     If the future has been cancelled, raises CancelledError.  If the
@@ -185,7 +187,11 @@ cdef object _get_result(fut: Future):
         fut._Future__log_traceback = False
         exc = fut._exception
         if exc is not None:
-            raise exc.with_traceback(exc.__traceback__)
+            cached_traceback = fut.__traceback__
+            if cached_traceback is None:
+                cached_traceback = exc.__traceback__
+                fut.__traceback__ = cached_traceback
+            raise exc.with_traceback(cached_traceback) from exc.__cause__
         return fut._result
     if state == "CANCELLED":
         raise fut._make_cancelled_error()
@@ -230,6 +236,8 @@ class SmartFuture(Future, Generic[T]):
     _key: Optional[Key] = None
     
     _waiters: "weakref.WeakSet[SmartTask[T]]"
+    
+    __traceback__: Optional[TracebackType] = None
 
     def __init__(
         self,
@@ -408,7 +416,10 @@ class SmartTask(Task, Generic[T]):
     See Also:
         - :class:`asyncio.Task`
     """
+    
     _waiters: Set["Task[T]"]
+    
+    __traceback__: Optional[TracebackType] = None
 
     @cython.linetrace(False)
     def __init__(
@@ -623,7 +634,7 @@ cdef tuple _get_done_callbacks(inner: Task, outer: Future):
         if _is_cancelled(outer):
             if not _is_cancelled(inner):
                 # Mark inner's result as retrieved.
-                _get_exception(inner)
+                inner._Future__log_traceback = False
             return
 
         if _is_cancelled(inner):
@@ -633,7 +644,7 @@ cdef tuple _get_done_callbacks(inner: Task, outer: Future):
             if exc is not None:
                 outer.set_exception(exc)
             else:
-                outer.set_result(_get_result(inner))
+                outer.set_result(inner._result)
 
     def _outer_done_callback(outer):
         if _is_not_done(inner):