Revert "Fix leave_context_asyncio to handle cancelled asyncio task"

This reverts commit 74c86b6.
quora · Apr 18, 2024 · f85c02b · f85c02b
1 parent bf350a7
commit f85c02b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/asynq/contexts.py b/asynq/contexts.py
@@ -93,7 +93,7 @@ def leave_context_asyncio(context):
         debug.write("@async: -context: %s" % debug.str(context))
 
     task = asyncio.current_task()
-    getattr(task, ASYNCIO_CONTEXT_FIELD, {}).pop(id(context), None)  # type: ignore
+    del getattr(task, ASYNCIO_CONTEXT_FIELD)[id(context)]  # type: ignore
 
 
 def pause_contexts_asyncio(task):
@@ -104,7 +104,7 @@ def pause_contexts_asyncio(task):
 
 
 def resume_contexts_asyncio(task):
-    if not getattr(task, ASYNCIO_CONTEXT_ACTIVE_FIELD, True):
+    if not getattr(task, ASYNCIO_CONTEXT_ACTIVE_FIELD, False):
         setattr(task, ASYNCIO_CONTEXT_ACTIVE_FIELD, True)
         for ctx in getattr(task, ASYNCIO_CONTEXT_FIELD, {}).values():
             ctx.resume()