Postgres 다시 만들기 Python으로 직접 복사 하시겠습니까?

Question

나는 데이터 블럭을 가지고 있는데, 현재는 n-tuples의리스트이지만 형식은 꽤 유연하다. Postgres 테이블에 추가하고 싶다.이 경우 각 n-tuple은 DB.

이 모든 것을 CSV 파일에 작성한 다음 postgres의 COPY를 사용하여이 모든 것을 데이터베이스에 대량로드합니다. 이 작품은, 그러나 차선책입니다, 나는 파이썬에서 직접이 모든 것을 할 수 있기를 원합니다. Postgres에서 COPY 유형의 대량로드를 복제하기위한 python 내부의 메소드가 있습니까?

Answer 1

psycopg2 드라이버를 사용하는 경우 커서는 copy_to 및 copy_from 기능을 파일과 유사한 객체 (버퍼 StringIO 포함)에서 읽을 수 있습니다.

examples/copy_from.py 및 examples/copy_to.py 파일에 psycopg2 source distribution과 함께 제공되는 파일의 예가 있습니다.

이 발췌는 copy_from.py 예에서입니다 :

conn = psycopg2.connect(DSN) 
curs = conn.cursor() 
curs.execute("CREATE TABLE test_copy (fld1 text, fld2 text, fld3 int4)") 

# anything can be used as a file if it has .read() and .readline() methods 
data = StringIO.StringIO() 
data.write('\n'.join(['Tom\tJenkins\t37', 
        'Madonna\t\N\t45', 
        'Federico\tDi Gregorio\t\N'])) 
data.seek(0) 

curs.copy_from(data, 'test_copy')