Question

给定一个生成器对象，是否可以测试它是否是由给定的生成器创建的？也许更好的说法是，有可能测试＆＃34;键入＆＃34;我们有发电机吗？由于生成器对象的类型为generator，因此使用type或isinstance进行测试将不起作用。

请考虑以下代码：

>>> def gen1():
...     yield 1
... 
>>> def gen2():
...     yield 2
... 
>>> g1 = gen1()
>>> g2 = gen2()
>>> 
>>> def do_something(f):
...     # need to know if f is a gen1 generator or a gen2 generator here
...     # isinstance(f, gen1)  raises a TypeError since gen1 is not a type
...     # type(f) is gen1 returns false
...     print(f)
... 
>>> do_something(g1)
<generator object gen1 at 0x100dcb370>
>>> do_something(g2)
<generator object gen2 at 0x100dcb3c0>

请注意，如果我使用迭代器类来实现生成器，那么type和isinstance都可以工作，因为生成的对象将是迭代器类给出的类型： / p>

>>> class gen():
...     def __next__(self):
...          return 1
... 
>>> f = gen()
>>> type(f) is gen
True
>>> isinstance(f, gen)
True

是否可以通过使用yield生成的生成器实现简单（不使用将数据包包装在类中，或使用装饰器）等方面实现相同的功能？

Answer 1

生成函数的名称（如果有）作为字符串存储在__name__属性中。如果字符串是你需要的全部，那么你已经完成了。

要获得实际的功能对象，它很复杂。生成器本身不保存对函数对象的任何引用，逻辑使用对代码对象的引用进行操作。

function object (`gen1`) -----> <code object >
                                    ^
generator object (`g1`)  -----------╯

使用生成器函数的原始名称，您可以想象在包含模块上使用getattr来尝试检索原始函数。这种方法不一定可靠：这里假设函数对象仍然存在并且仍然绑定到该名称。但是，该名称可能已被删除或反弹。在CPython中，如果引用计数减少到0，函数也可能已被垃圾收集。

在Python中，有没有办法测试生成器对象以找出哪个生成器创建它？

1 个答案: