可以使用什么样的对象`yield from`？

Question

最初 (PEP 380)，yield from 语法被引入用于委托给“子生成器”。后来它与现在 deprecated 基于生成器的协程一起使用。

我不知道yield from 一般可以应用于什么样的对象。我的第一个猜想是它只需要对象上的__iter__ 方法来返回一个迭代器。事实上，以下内容适用于 Python 3.8：

class C:
    def __init__(self, n):
        self.n = n

    def __iter__(self):
        return iter(range(self.n))

def g(n):
    yield from C(n)

print(tuple(g(3)))

但是，它也适用于一些等待对象，例如 asyncio.sleep(1)，它们没有 __iter__ 方法。

一般规则是什么？什么决定了一个对象是否可以作为yield from 表单的参数？

Answer 1

您可以查看 CPython evaluates that statement 的方式。由此可见，它需要是协程或可迭代的：

case TARGET(GET_YIELD_FROM_ITER): {
    /* before: [obj]; after [getiter(obj)] */
    PyObject *iterable = TOP();
    PyObject *iter;
    if (PyCoro_CheckExact(iterable)) {
        /* `iterable` is a coroutine */
        if (!(co->co_flags & (CO_COROUTINE | CO_ITERABLE_COROUTINE))) {
            /* and it is used in a 'yield from' expression of a
               regular generator. */
            Py_DECREF(iterable);
            SET_TOP(NULL);
            _PyErr_SetString(tstate, PyExc_TypeError,
                             "cannot 'yield from' a coroutine object "
                             "in a non-coroutine generator");
            goto error;
        }
    }
    else if (!PyGen_CheckExact(iterable)) {
        /* `iterable` is not a generator. */
        iter = PyObject_GetIter(iterable);
        Py_DECREF(iterable);
        SET_TOP(iter);
        if (iter == NULL)
            goto error;
    }
    PREDICT(LOAD_CONST);
    DISPATCH();
}