actually fix tasks now

Author: cirospaciari

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "socketify"
-version = "0.0.26"
+version = "0.0.27"
 dynamic = ["dependencies"]
 authors = [
   { name="Ciro Spaciari", email="ciro.spaciari@gmail.com" },

setup.py

@@ -58,7 +58,7 @@
 
 setuptools.setup(
     name="socketify",
-    version="0.0.26",
+    version="0.0.27",
     platforms=["any"],
     author="Ciro Spaciari",
     author_email="ciro.spaciari@gmail.com",

src/socketify/tasks.py

@@ -500,11 +500,11 @@ def __step(self, exc=None):
         coro = self._coro
         self._fut_waiter = None
 
-        # _parent_task = current_task(self._loop)
-        # if _parent_task is not None:
-        #     _leave_task(self._loop, _parent_task)
-        #     self._parent_task = _parent_task
-        # _enter_task(self._loop, self)
+        _parent_task = current_task(self._loop)
+        if _parent_task is not None:
+            _leave_task(self._loop, _parent_task)
+            self._parent_task = _parent_task
+        _enter_task(self._loop, self)
         # Call either coro.throw(exc) or coro.send(None).
         try:
             if exc is None:
@@ -574,10 +574,10 @@ def __step(self, exc=None):
                 new_exc = RuntimeError(f"Task got bad yield: {result!r}")
                 self._loop.call_soon(self.__step, new_exc, context=self._context)
         finally:
-            # _leave_task(self._loop, self)
-            # if self._parent_task is not None:
-            #     _enter_task(self._loop, self._parent_task)
-            #     self._parent_task = None
+            _leave_task(self._loop, self)
+            if self._parent_task is not None:
+                _enter_task(self._loop, self._parent_task)
+                self._parent_task = None
             self = None  # Needed to break cycles when an exception occurs.
 
     def __wakeup(self, future):