ez-a-sync 0.32.10__cp39-cp39-win32.whl → 0.32.12__cp39-cp39-win32.whl

a_sync/_smart.pyx

@@ -9,14 +9,16 @@ 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 cpython.unicode cimport PyUnicode_CompareWithASCIIString
 
-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 +52,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
 
 
@@ -153,7 +156,7 @@ cdef inline bint _is_done(fut: Future):
     Done means either that a result / exception are available, or that the
     future was cancelled.
     """
-    return <str>fut._state != "PENDING"
+    return PyUnicode_CompareWithASCIIString(fut._state, b"PENDING") != 0
 
 
 @cython.linetrace(False)
@@ -163,17 +166,17 @@ cdef inline bint _is_not_done(fut: Future):
     Done means either that a result / exception are available, or that the
     future was cancelled.
     """
-    return <str>fut._state == "PENDING"
+    return PyUnicode_CompareWithASCIIString(fut._state, b"PENDING") == 0
 
 
 @cython.linetrace(False)
 cdef inline bint _is_cancelled(fut: Future):
     """Return True if the future was cancelled."""
-    return <str>fut._state == "CANCELLED"
+    return PyUnicode_CompareWithASCIIString(fut._state, b"CANCELLED") == 0
 
 
 @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
@@ -181,13 +184,17 @@ cdef object _get_result(fut: Future):
     the future is done and has an exception set, this exception is raised.
     """
     cdef str state = fut._state
-    if state == "FINISHED":
-        fut._Future__log_traceback = False
+    if PyUnicode_CompareWithASCIIString(state, b"FINISHED") == 0:
         exc = fut._exception
         if exc is not None:
-            raise exc.with_traceback(exc.__traceback__)
+            cached_traceback = fut.__traceback__
+            if cached_traceback is None:
+                fut._Future__log_traceback = False
+                cached_traceback = exc.__traceback__
+                fut.__traceback__ = cached_traceback
+            raise exc.with_traceback(cached_traceback) from exc.__cause__
         return fut._result
-    if state == "CANCELLED":
+    if PyUnicode_CompareWithASCIIString(state, b"CANCELLED") == 0:
         raise fut._make_cancelled_error()
     raise InvalidStateError('Result is not ready.')
 
@@ -201,10 +208,10 @@ cdef object _get_exception(fut: Future):
     InvalidStateError.
     """
     cdef str state = fut._state
-    if state == "FINISHED":
+    if PyUnicode_CompareWithASCIIString(state, b"FINISHED") == 0:
         fut._Future__log_traceback = False
         return fut._exception
-    if state == "CANCELLED":
+    if PyUnicode_CompareWithASCIIString(state, b"CANCELLED") == 0:
         raise fut._make_cancelled_error()
     raise InvalidStateError('Exception is not set.')
 
@@ -230,6 +237,8 @@ class SmartFuture(Future, Generic[T]):
     _key: Optional[Key] = None
     
     _waiters: "weakref.WeakSet[SmartTask[T]]"
+    
+    __traceback__: Optional[TracebackType] = None
 
     def __init__(
         self,
@@ -292,7 +301,7 @@ class SmartFuture(Future, Generic[T]):
 
     def __await__(self) -> Generator[Any, None, T]:
         """
-        Await the smart future or task, handling waiters and logging.
+        Await the future, handling waiters and logging.
 
         Yields:
             The result of the future or task.
@@ -408,7 +417,10 @@ class SmartTask(Task, Generic[T]):
     See Also:
         - :class:`asyncio.Task`
     """
+    
     _waiters: Set["Task[T]"]
+    
+    __traceback__: Optional[TracebackType] = None
 
     @cython.linetrace(False)
     def __init__(
@@ -439,7 +451,7 @@ class SmartTask(Task, Generic[T]):
 
     def __await__(self) -> Generator[Any, None, T]:
         """
-        Await the smart future or task, handling waiters and logging.
+        Await the task, handling waiters and logging.
 
         Yields:
             The result of the future or task.
@@ -623,7 +635,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 +645,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):