파이썬에서 백업 파일 및 임시 파일 생성하는 방법

가끔가다가 파일을 읽고 저장하는 과정에서 오류가 발생할 시 파일이 손상되어 더이상 사용할 수 없게 되는 경우들이 발생할 수 있다.
이를 위해 기존 파일에 대한 백업파일을 생성하고, 임시 파일에서 write 작업을 한뒤 기존 파일로 교체해주는 작업을 해줄 수 있다.

1. 백업 파일 생성하는 방법

파이썬의 shutil 라이브러리 내 .copy 함수를 이용하여 백업 파일을 생성하면 된다.

import shutil
 
path = '/path/to/data.pkl'
 
# 임시 및 백업 경로 설정
temp_path = path + '.temp'
backup_path = path + '.backup'
 
# 백업 파일 생성
shutil.copy(path, backup_path)

2. 임시 파일 생성 후 기존 파일로 교체하는 방법

임시파일로 변경된 저장 결과를 저장하고 이를 기존 파일로 교체하기 위해선 os.replace() 를 사용하면 된다.

import os
import pickle
 
data = {} # ...
 
# 임시 파일로 저장
with open(temp_path, "wb") as file:
    pickle.dump(data, file)
 
# 임시 파일을 원본 파일로 교체
os.replace(temp_path, path)

3. 오류 발생 시 원본 파일을 백업에서 복구

오류가 발생해 파일이 정상적으로 종료되지 않았다면 아래와 같이 백업된 파일 경로에서 불러오면 된다.

# backup된 파일을 기존 파일로 복구
shutil.copy(backup_path, path)

4. 임시 파일 or 백업 파일 정리하기 (삭제하기)

백업된 파일과 임시 파일을 제거하고 싶다면 os.remove() 를 사용하면 된다.

if os.path.exists(temp_path):
    os.remove(temp_path)
if os.path.exists(backup_path):
    os.remove(backup_path)