Function _swap_current_task

crates/stdlib/src/_asyncio.rs:2530–2556

(loop_: PyObjectRef, task: PyObjectRef, vm: &VirtualMachine)

    #[pyfunction]
    fn _swap_current_task(loop_: PyObjectRef, task: PyObjectRef, vm: &VirtualMachine) -> PyResult {
        // Per-thread swap, matching CPython's swap_current_task
        let prev = vm
            .asyncio_running_task
            .borrow()
            .clone()
            .unwrap_or_else(|| vm.ctx.none());

        if vm.is_none(&task) {
            *vm.asyncio_running_task.borrow_mut() = None;
        } else {
            *vm.asyncio_running_task.borrow_mut() = Some(task.clone());
        }

        // Also update the module-level dict for cross-thread queries
        if let Ok(current_tasks) = get_current_tasks_dict(vm)
            && let Ok(dict) = current_tasks.downcast::<rustpython_vm::builtins::PyDict>()
        {
            if vm.is_none(&task) {
                let _ = dict.del_item(&*loop_, vm);
            } else {
                let _ = dict.set_item(&*loop_, task, vm);
            }
        }

        Ok(prev)
    }

    /// Reset task state after fork in child process.
    #[pyfunction]