파이썬(Python)에서 보안이 강화된 임시 파일을 안전하게 생성하는 방법

파이썬에서는 tempfile 모듈을 사용하여 가장 안전한 방식으로 고유한 임시 파일을 생성할 수 있습니다. 특히 mkstemp() 함수를 활용하면 파일 생성 과정에서 경쟁 상태(race condition)가 발생하지 않으며, 생성된 파일은 해당 파일을 만든 사용자 ID만 읽고 쓸 수 있도록 접근 권한이 제한됩니다.

다만 한 가지 주의할 점이 있습니다. mkstemp()로 생성한 임시 파일은 자동으로 삭제되지 않기 때문에, 사용이 끝난 후에는 개발자가 직접 파일을 삭제해야 합니다.

예제

import tempfile
_, temp_file_path = tempfile.mkstemp()
print("File path: " + temp_file_path)

위 코드를 실행하면 시스템의 임시 디렉터리에 고유한 이름을 가진 새로운 임시 파일이 생성되고, 해당 파일의 경로가 출력됩니다. mkstemp()는 두 개의 값을 반환하는데, 첫 번째 값은 파일 디스크립터이며 여기서는 언더스코어(_)를 사용해 무시했습니다.

사용 후 임시 파일 삭제하기

mkstemp()는 파일을 자동으로 정리해 주지 않으므로, 작업이 끝난 후에는 반드시 직접 삭제해야 합니다. 아래와 같이 os.remove()를 사용하면 됩니다.

import os
import tempfile

fd, temp_file_path = tempfile.mkstemp()
try:
    # 임시 파일을 사용하는 코드 작성
    ...
finally:
    os.close(fd)
    os.remove(temp_file_path)  # 사용 후 파일 삭제

이처럼 try...finally 구문을 활용하면 예외가 발생하더라도 임시 파일이 확실하게 삭제되도록 보장할 수 있습니다. 또한 tempfile.NamedTemporaryFile(delete=True)나 컨텍스트 관리자(with 문)를 사용하면 블록이 종료될 때 파일이 자동으로 삭제되므로, 더욱 편리하고 안전하게 임시 파일을 처리할 수 있습니다.