跳转到主要内容

将Python 3.x中的`yield from`语义向后移植到Python 2.7

项目描述

yieldfrom是将Python 3.x中的yield from语义向后移植到Python 2.7的工具。它也支持Python 3.x,因此可以作为兼容性库,用于支持Python两个主要版本的代码。

如果你想在Python 3.x中嵌套生成器,你可以使用yield from关键字。这允许你自动遍历子生成器,并将异常和返回值从顶级调用者透明地传递到最低级别的生成器。

def subgen():
    yield 2
    yield 3

def gen():
    yield 1
    yield from subgen()  # Python 3.x only
    yield 4

def main():
    for i in gen():
        print i,

>>> main()
... 1 2 3 4

Python 2.x中不提供此功能,我们通过使用yieldfrom装饰器和辅助类From来模拟它。

from yieldfrom import yieldfrom, From
def subgen():
    yield 2
    yield 3

@yieldfrom
def gen():
    yield 1
    yield From(subgen())
    yield 4

def main():
    for i in gen():
        print i,

>>> main()
... 1 2 3 4

高级用法允许使用StopIteration从子生成器返回一个值。使用Return可以方便地做到这一点。

from yieldfrom import yieldfrom, From, Return

def subgen():
    yield 2
    yield 3
    Return(100)  # Raises `StopIteration(100)`

@yieldfrom
def gen():
    yield 1
    ret = (yield From(subgen()))
    yield 4
    yield ret

def main():
    for i in gen():
        print i,

>>> main()
... 1 2 3 4 100

子生成器可以在多个级别上嵌套,每个级别都需要通过yieldfrom进行额外的装饰。

def subsubgen():
    yield 2

@yieldfrom
def subgen():
    yield From(subsubgen())
    yield 3

@yieldfrom
def gen():
    yield 1
    yield From(subgen())
    yield 4

def main():
    for i in gen():
        print i,

>>> main()
... 1 2 3 4

可以由相关子生成器处理抛入顶级生成器的异常。

def subsubgen():
    try:
        yield 2
    except ValueError:
        yield 200

@yieldfrom
def subgen():
    yield From(subsubgen())
    yield 3

@yieldfrom
def gen():
    yield 1
    yield From(subgen())
    yield 4

def main():
    try:
        g = gen()
        while True:
            i = g.next()
            if i == 2:
                i = g.throw(ValueError())
        print i,
    except StopIteration:
        pass

>>> main()
... 1 200 3 4

请注意,如果你在简单的可迭代对象(例如listtuple等)上使用yield From(),则迭代器的各个成员将在每次迭代中产生(在这种情况下,你可能需要使用通常的yield)。

@yieldfrom
def gen():
    yield From([1, 2, 3])
    yield [1, 2, 3]

def main():
    for i in gen():
        print i

>>> main()
... 1
... 2
... 3
... [1, 2, 3]

将非可迭代对象传递给From将导致产生一个空生成器,该生成器不执行任何操作。

@yieldfrom
def gen():
    yield From(None)
    yield 1

def main():
    for i in gen():
        print i

>>> main()
... 1

本模块是对以下Python菜谱的改编:http://code.activestate.com/recipes/576727
修改包括异常处理、命名、文档、空生成器的处理等。

项目详情


下载文件

下载适用于您平台的文件。如果您不确定选择哪个,请了解更多关于安装包的信息。

源代码分发

yieldfrom-1.0.5.tar.gz (5.1 kB 查看哈希值)

上传时间: 源代码

构建分发

yieldfrom-1.0.5-py2.py3-none-any.whl (5.9 kB 查看哈希值)

上传时间: Python 2 Python 3

支持者

AWS AWS 云计算和安全赞助商 Datadog Datadog 监控 Fastly Fastly CDN Google Google 下载分析 Microsoft Microsoft PSF 赞助商 Pingdom Pingdom 监控 Sentry Sentry 错误日志 StatusPage StatusPage 状态页面