将psql表(或查询)作为文件(csv、json)直接导出到AWS S3的方法

A way to export psql table (or query) directly to AWS S3 as file (csv, json)(将psql表(或查询)作为文件(csv、json)直接导出到AWS S3的方法)
本文介绍了将psql表(或查询)作为文件(csv、json)直接导出到AWS S3的方法的处理方法,对大家解决问题具有一定的参考价值,需要的朋友们下面随着小编来一起学习吧!

问题描述

This answer建议使用AWS数据管道,但我想知道是否有一种聪明的方法可以用我自己的机器和Python来做到这一点。

我一直在使用optog2boto3pandas库。表格有5到50列和几百万行。我目前的方法不能很好地处理大数据。

推荐答案

我猜我也可以在这里展示我自己的一个版本,它基于

中的copy_expert
import io
import psycopg2
import boto3

resource = boto3.resource('s3')
conn = psycopg2.connect(dbname=db, user=user, password=pw, host=host)
cur = conn.cursor()

def copyFun(bucket, select_query, filename):
    query = f"""COPY {select_query} TO STDIN 
            WITH (FORMAT csv, DELIMITER ',', QUOTE '"', HEADER TRUE)"""
    file = io.StringIO()
    cur.copy_expert(query, file)
    resource.Object(bucket, f'{filename}.csv').put(Body=file.getvalue())

这篇关于将psql表(或查询)作为文件(csv、json)直接导出到AWS S3的方法的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除!

相关文档推荐

Make gRPC messages JSON serializable(使GRPC消息JSON可序列化)
Send plain JSON to a gRPC server using python(使用Python将纯JSON发送到GRPC服务器)
how to iterate through geojson elements(如何循环访问Geojson元素)
how to validate properties key in a geojson(如何验证Geojson中的属性密钥)
Python JSONDecoder custom translation of null type(空类型的Python JSONDecoder自定义翻译)
Splitting a MultiPolygon(拆分多重多边形)