如何在Python的__iter__方法中调用异步代码？异步迭代问题解决

为什么原来的__iter__不能用 #

同步迭代协议要求__iter__返回迭代器，但异步操作必须用await，没法放在同步方法里，硬改会违反协议导致报错。

正确实现异步迭代（__aiter__） #

你自己写的__aiter__有个小bug（变量name未定义，应该用db），修正后是完全符合规范的：

async def __aiter__(self):
    for db in await self._databases():
        yield db

使用时要配合async for语法：

dbs = PGServer(...)
async for db in dbs:
    print(db)

Python 3.6+支持异步生成器，这种写法会自动实现__anext__方法，满足异步迭代器协议，运行完全没问题。

两种方案对比：异步迭代 vs 直接调用公开方法 #

方案1：异步迭代（__aiter__） #

• 优势：保留了原有的迭代语法风格，调用方无需关心内部是同步还是异步，接口一致性更强；如果后续要处理大数据量，需要分批拉取数据（比如用conn.cursor()逐行获取），可以直接修改__aiter__实现，调用方代码不用改。
• 劣势：要求调用方使用async for，如果项目里同步逻辑居多，需要额外适配异步上下文。

方案2：直接调用公开的databases()异步方法 #

dbs_list = await dbs.databases()
for db in dbs_list:
    print(db)

• 优势：代码直白易懂，新手更容易上手；如果只需要一次性获取所有数据库列表，这种写法更简洁。
• 劣势：如果后续要改成流式获取数据，调用方代码也得跟着调整；失去了迭代器的惰性加载特性（当前场景是一次性拉取所有数据，影响不大）。

建议 #

如果想保留迭代的语法体验，修正异步生成器的变量错误后，用__aiter__配合async for是完全合规的；如果追求代码直观性，直接调用公开的databases()方法也没问题，两种方案都合理，取决于你的业务场景和团队编码习惯。

内容的提问来源于stack exchange，提问作者odigity