对象 str 不能用于 psycopg3 中的 await 表达式答案

【问题标题】：object str cannot be used in await expression in psycopg3对象 str 不能用于 psycopg3 中的 await 表达式
【发布时间】：2023-02-22 00:11:27
【问题描述】：

我有一个将 csv 文件复制到数据库的功能。我正在尝试异步执行此操作：

import psycopg
from config import config
from pathlib import WindowsPath
from psycopg import sql
import asyncio

async def main():
    conn = await psycopg.AsyncConnection.connect(f'postgresql://{config.USER_PG}:{config.PASS_PG}@{config.HOST_PG}:{config.PORT_PG}/{config.DATABASE_PG}')

    p = WindowsPath(r'.\data\product_version.csv')

    async with conn:
        if p.exists():
            with p.open(encoding='utf-8-sig') as f:

                columns = list(next(f).strip().lower().split(','))


                async with conn.cursor() as cur:
                    await cur.execute(sql.SQL("TRUNCATE TABLE {} RESTART IDENTITY CASCADE").format(sql.Identifier('product_version_map')))

                    async with cur.copy(sql.SQL("COPY {} ({}) FROM STDIN WITH CSV").format(sql.Identifier('product_version_map'),sql.SQL(', ').join(map(sql.Identifier, columns)))) as copy:
                            while data :=  await f.read():
                                await copy.write(data)

        else:
            print(f'You need the product_version file')



if __name__=='__main__':
    asyncio.set_event_loop_policy(asyncio.WindowsSelectorEventLoopPolicy())
    asyncio.run(main())

但我收到此错误：

psycopg.errors.QueryCanceled: COPY from stdin failed: error from Python: TypeError - object str can't be used in 'await' expression

f 这里有类字符串，它是文件的行。错误来自这一行：

while data :=  await f.read():

这是我在构建此代码时所指的文档：

https://www.psycopg.org/psycopg3/docs/basic/copy.html#asynchronous-copy-support

【问题讨论】：

标签： python asynchronous python-asyncio psycopg2 psycopg3

【解决方案1】：

您正在打开文件以作为同步文件读取 - 没有 await 可从中读取。 f.read() 被解析并在 Python 到达 await 部分之前返回一个字符串：关键字期望右边的目标表达式是一个“等待”对象——事实并非如此。（这正是错误消息告诉您的内容）。

只需将“await”关键字放在这一行，您的程序就会通过这一部分。

在您展示的代码中，您并没有真正利用异步代码可能的并行化优势，因此，只需将 await 拿走。如果你在一个更大的系统的生产中有这个，这实际上很重要，并且你在所有其他步骤中获得了最大的性能，那么你可以引入异步文件读取并插入这部分以获得一些收益。

【讨论】：