给定一个生成器对象,是否可以测试它是否由给定的生成器创建?也许更好地说,是否有可能测试我们拥有的发电机的“类型”?由于生成器对象的类型为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
会的工作,因为生成的对象将是由迭代器类所给出的类型:
>>> class gen():
... def __next__(self):
... return 1
...
>>> f = gen()
>>> type(f) is gen
True
>>> isinstance(f, gen)
True
是否可以通过使用创建的生成器以简单的方式(例如不将生成器包装在类中或使用装饰器之类的方式)实现相同目的yield
?
生成函数的名称(如果有)__name__
以字符串形式存储在属性中。如果只需要字符串,就可以完成。
要获得实际的功能对象,这很复杂。生成器本身不保存对功能对象的任何引用,并且逻辑使用对代码对象的引用进行操作。
function object (`gen1`) -----> <code object >
^
generator object (`g1`) -----------╯
怀着生成器函数的原始名称,可以想象,您可以getattr
在包含模块上使用它来尝试检索原始函数。该方法不一定可靠:这里假设函数对象仍然存在,并且仍然绑定到该名称。但是,该名称可能已被删除或重新绑定。在CPython中,如果引用计数减少到0,则该函数也可能已经被垃圾回收。
本文收集自互联网,转载请注明来源。
如有侵权,请联系[email protected] 删除。
我来说两句