[python] PyMySQL을 이용하여 데이터베이스의 레코드를 백업하고 복원하는 방법을 알아본다.

04 Dec 2023

python

이번 글에서는 PyMySQL을 사용하여 데이터베이스의 레코드를 백업하고 복원하는 방법에 대해 알아보겠습니다.

1. PyMySQL 설치하기

먼저, PyMySQL을 설치해야 합니다. 아래 명령어를 사용하여 PyMySQL을 설치할 수 있습니다.

pip install PyMySQL

2. 데이터베이스에 연결하기

데이터베이스에 연결하기 위해 필요한 정보를 설정해야 합니다. 아래 코드를 사용하여 데이터베이스에 연결할 수 있습니다.

import pymysql

# 데이터베이스 연결 정보 설정
host = "localhost"
user = "username"
password = "password"
database = "database_name"

# 데이터베이스에 연결
db = pymysql.connect(host=host, user=user, password=password, database=database)

3. 데이터베이스 레코드 백업하기

백업을 위해 SQL 쿼리를 사용하여 데이터베이스 레코드를 가져올 수 있습니다. 예를 들어, 아래 코드는 users 테이블의 모든 레코드를 백업하는 방법을 보여줍니다.

# 백업할 테이블 설정
table = "users"

# 백업 SQL 쿼리 생성
query = f"SELECT * FROM {table}"

# 쿼리 실행
cursor = db.cursor()
cursor.execute(query)
results = cursor.fetchall()

# 백업된 레코드 출력
for row in results:
    print(row)

4. 데이터베이스 레코드 복원하기

복원을 위해서는 백업된 데이터를 다시 데이터베이스에 삽입해야 합니다. 아래 코드는 백업된 데이터를 backup.csv 파일에서 읽어와 users 테이블에 복원하는 방법을 보여줍니다.

import csv

# 복원할 테이블 설정
table = "users"

# 백업 데이터 파일명
backup_file = "backup.csv"

# 백업 데이터 파일 열기
with open(backup_file, "r") as f:
    reader = csv.reader(f)
    records = list(reader)

# 복원 SQL 쿼리 생성
query = f"INSERT INTO {table} VALUES (%s, %s, %s, %s)"

# 쿼리 실행
cursor = db.cursor()
cursor.executemany(query, records)
db.commit()

5. 데이터베이스 연결 닫기

모든 작업이 완료되면 데이터베이스 연결을 닫아야 합니다. 아래 코드를 사용하여 데이터베이스 연결을 닫을 수 있습니다.

db.close()

위의 방법을 사용하면 PyMySQL을 이용하여 데이터베이스의 레코드를 백업하고 복원할 수 있습니다. PyMySQL의 더 자세한 사용법에 대해서는 공식 문서를 참조하시기 바랍니다.