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

04 Dec 2023

python

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

1. PyMySQL 설치하기

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

pip install PyMySQL

2. 데이터베이스 연결

데이터베이스에 연결하기 위해 connect() 함수를 사용합니다. 필요한 정보를 인자로 전달하여 데이터베이스에 연결할 수 있습니다.

import pymysql

# 데이터베이스 연결 정보
host = "localhost"
user = "root"
password = "password"
database = "mydatabase"

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

3. 레코드 백업하기

레코드를 백업하기 전에, 백업할 테이블의 이름과 백업할 파일의 경로를 지정해야 합니다.

table_name = "mytable"
backup_file_path = "/path/to/backup.sql"

백업을 수행하기 위해 SELECT 쿼리를 사용합니다. 쿼리 결과를 파일에 기록하여 백업합니다.

query = "SELECT * FROM {}".format(table_name)

# 쿼리 실행
cursor = conn.cursor()
cursor.execute(query)

# 쿼리 결과를 파일에 저장
with open(backup_file_path, "w") as f:
    for row in cursor:
        f.write(str(row) + "\n")

위의 코드를 실행하면, backup_file_path에 지정된 경로에 레코드가 백업됩니다.

4. 백업한 레코드 복원하기

백업한 레코드를 복원하기 위해서는 LOAD DATA INFILE 문을 사용합니다. 백업 파일의 경로와 복원할 테이블의 이름을 지정합니다.

backup_file_path = "/path/to/backup.sql"
table_name = "mytable"

복원을 수행하기 위해 LOAD DATA INFILE 문을 사용합니다.

query = """
LOAD DATA INFILE "{}"
INTO TABLE {}
FIELDS TERMINATED BY ','
LINES TERMINATED BY '\n'
""".format(backup_file_path, table_name)

# 쿼리 실행
cursor = conn.cursor()
cursor.execute(query)

# 변경 사항을 커밋
conn.commit()

위의 코드를 실행하면, 백업 파일에서 읽어 들인 레코드가 지정된 테이블에 복원됩니다.

5. 데이터베이스 연결 종료하기

데이터베이스 작업이 끝나면, 연결을 종료해야 합니다. close() 메서드를 사용하여 데이터베이스 연결을 닫을 수 있습니다.

conn.close()

위의 예제 코드를 참고하여 PyMySQL을 사용하여 데이터베이스의 레코드를 백업하고 복원하는 방법에 대해 알아보았습니다. PyMySQL을 사용하면 간편하게 데이터베이스 작업을 수행할 수 있습니다.

더 자세한 내용은 PyMySQL 문서를 참고하시기 바랍니다.