如何更正 stringIO() 在 aws lambda 中没有属性错误？

Question

我有一个查询，我试图将它发送到STDIN然后发送到一个 csv 文件，并且复制命令正在失败。 当我调用io.StringIO()时它似乎倒下了

我收到的错误是： Error connecting to postgres instance '_io.StringIO' object has no attribute 'getValue'

我的代码如下所示：

import psycopg2
from io import StringIO
import boto3
bucket = 'my_s3_bucket'
filename = 'test_data'
s3_resource = boto3.resource('s3')
conn = psycopg2.connect(host='localhost'
                        , port='5432'
                        , dbname='billing_test'
                        , user='postgres'
                        , password='password!')
cur = conn.cursor()
sql_query = "SELECT\
 * from public.customers\
  limit 1"
#cur.execute(sql_query)
#records = cur.fetchall()
#print(records)

query = '''COPY ({}) TO STDIN csv header'''.format(sql_query)

file = StringIO()
cur.copy_expert(query, file)
s3_resource.Object(bucket, f'{filename}.csv').put(Body=file.getValue())
cur.close()
conn.close()

Answer 1

您应该阅读文档， help()是您在交互式终端中最好的朋友：

import io
help(io.StringIO)

...
 |  getvalue(self, /)
 |      Retrieve the entire contents of the object.
 |  
...

您应该使用getvalue()而不是getValue()

s3_resource.Object(bucket, f'{filename}.csv').put(Body=file.getvalue())