[python] Psycopg2를 사용하여 데이터베이스 테이블의 데이터를 백업 및 다른 데이터베이스로 복사하는 방법은?

22 Nov 2023

python

Psycopg2는 Python에서 PostgreSQL 데이터베이스에 연결하고 작업하는 데 사용되는 인기있는 라이브러리입니다. 이 라이브러리를 사용하여 데이터베이스 테이블의 데이터를 백업하고 다른 데이터베이스로 복사하는 방법을 살펴보겠습니다.

1. 데이터베이스 연결

먼저, Psycopg2를 사용하여 데이터베이스에 연결해야 합니다. 다음 예제 코드는 로컬 PostgreSQL 데이터베이스에 연결하는 방법을 보여줍니다.

import psycopg2

# 데이터베이스 연결 정보
conn = psycopg2.connect(
    host="localhost",
    database="mydatabase",
    user="myuser",
    password="mypassword"
)

위의 코드에서 host, database, user, password 부분을 실제 데이터베이스 연결 정보로 대체해야 합니다.

2. 데이터 백업

데이터베이스 테이블의 데이터를 백업하기 위해 psql 명령어를 사용할 수 있습니다. Psycopg2는 psql 명령어를 실행하는 기능을 제공하므로 이를 이용하여 백업을 수행할 수 있습니다.

import subprocess

# 데이터 백업 명령어 실행
subprocess.call(['psql', '-c', 'COPY mytable TO \'/path/to/backup/file.csv\' WITH (FORMAT csv);'], stdin=conn)

# 연결 종료
conn.close()

위의 예제 코드에서 mytable은 백업하려는 테이블의 이름이고, /path/to/backup/file.csv는 백업 파일의 경로와 이름을 나타냅니다. 이 코드는 테이블의 데이터를 CSV 파일로 백업하는 예제입니다.

3. 데이터 복사

다른 데이터베이스로 데이터를 복사하기 위해서는 두 개의 데이터베이스 연결을 생성하고 데이터를 복사하는 과정이 필요합니다. 다음은 데이터베이스 간 데이터 복사를 수행하는 예제 코드입니다.

# 대상 데이터베이스에 연결
target_conn = psycopg2.connect(
    host="localhost",
    database="target_database",
    user="target_user",
    password="target_password"
)

# 데이터 복사 실행
with conn.cursor() as source_cursor, target_conn.cursor() as target_cursor:
    source_cursor.execute("SELECT * FROM mytable")
    records = source_cursor.fetchall()

    for record in records:
        target_cursor.execute("INSERT INTO target_table VALUES (%s, %s)", record)

# 작업 완료 후 커밋
target_conn.commit()

# 연결 종료
target_conn.close()
conn.close()

위의 예제 코드에서 target_database, target_user, target_password는 데이터를 복사하고자 하는 대상 데이터베이스의 연결 정보입니다. mytable은 백업한 테이블의 데이터를 읽어올 테이블 이름이고, target_table은 데이터를 복사할 대상 테이블이름입니다.

백업과 데이터 복사는 Psycopg2를 사용하여 수행할 수 있는 간단한 방법입니다. 이를 응용하여 데이터베이스와 상호작용하는 다양한 작업을 수행할 수 있습니다.

참고 자료

Psycopg2 공식 문서: https://www.psycopg.org/docs/
PostgreSQL 공식 문서: https://www.postgresql.org/docs/