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

04 Dec 2023

python

데이터베이스의 테이블은 중요한 데이터를 포함하고 있어서, 이를 백업하고 복원하는 과정은 매우 중요합니다. 이번 글에서는 Python의 PyMySQL을 이용하여 데이터베이스의 테이블 데이터를 백업하고 복원하는 방법에 대해 알아보겠습니다.

1. PyMySQL 설치하기

먼저 PyMySQL을 설치해야 합니다. 아래의 명령을 사용하여 PyMySQL을 설치합니다.

pip install PyMySQL

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

데이터베이스에 연결하기 위해 PyMySQL을 사용합니다. 아래의 코드를 참고하여 데이터베이스에 연결합니다.

import pymysql

connection = pymysql.connect(
    host='localhost',
    user='username',
    password='password',
    db='database_name',
    charset='utf8mb4',
    cursorclass=pymysql.cursors.DictCursor
)

위의 코드에서 host, user, password, db는 각각 데이터베이스의 호스트, 사용자, 비밀번호, 데이터베이스명으로 바꿔주어야 합니다.

3. 데이터 백업하기

PyMySQL을 이용하여 데이터베이스의 테이블 데이터를 백업하는 방법은 다음과 같습니다.

def backup_table(connection, table_name):
    try:
        cursor = connection.cursor()

        # 테이블 데이터를 백업할 SQL 쿼리 작성
        query = f"SELECT * FROM {table_name}"
        
        # SQL 쿼리 실행
        cursor.execute(query)

        # 결과 가져오기
        result = cursor.fetchall()

        # 결과를 파일로 저장
        with open(f"{table_name}_backup.txt", "w") as file:
            for row in result:
                file.write(str(row) + "\n")

        print(f"테이블 {table_name}의 데이터를 백업하였습니다.")

    except Exception as e:
        print(f"백업 중 에러가 발생하였습니다: {e}")

    finally:
        cursor.close()

# 사용 예시
backup_table(connection, 'example_table')

위의 코드에서 table_name은 백업하고자 하는 테이블의 이름으로 변경해야 합니다. 백업된 데이터는 테이블명_backup.txt 라는 파일로 저장됩니다.

4. 데이터 복원하기

백업된 데이터를 이용하여 테이블을 복원하는 방법은 다음과 같습니다.

def restore_table(connection, table_name):
    try:
        cursor = connection.cursor()

        # 백업된 데이터 파일 읽기
        with open(f"{table_name}_backup.txt", "r") as file:
            rows = file.readlines()

        # 데이터베이스에 데이터 복원하기
        for row in rows:
            query = f"INSERT INTO {table_name} VALUES ({row})"

            cursor.execute(query)
            connection.commit()

        print(f"테이블 {table_name}의 데이터를 복원하였습니다.")

    except Exception as e:
        print(f"복원 중 에러가 발생하였습니다: {e}")

    finally:
        cursor.close()

# 사용 예시
restore_table(connection, 'example_table')

위의 코드에서도 table_name은 복원하고자 하는 테이블의 이름으로 변경해야 합니다. 백업된 데이터 파일은 테이블명_backup.txt 파일이어야 합니다.

결론

위의 방법을 통해 PyMySQL을 이용하여 데이터베이스의 테이블 데이터를 백업하고 복원할 수 있습니다. 데이터 백업과 복원은 중요한 데이터를 보호하고 데이터 손실을 방지하기 위해 반드시 실시해야 하는 작업입니다.