Python shutil 모듈을 이용한 파일과 폴더 관리

Python

Author

gabriel yang

Published

October 5, 2024

Python의 shutil 모듈은 고수준의 파일 및 디렉토리 작업을 처리할 수 있는 강력한 도구입니다. 이 모듈은 특히 파일을 복사하거나, 디렉토리를 재귀적으로 삭제하고, 파일을 이동하는 등의 작업에 매우 유용합니다. 이번 글에서는 shutil 모듈을 이용하여 파일과 폴더를 어떻게 관리할 수 있는지 알아보겠습니다.

1. `shutil` 모듈 임포트

shutil 모듈을 사용하려면 먼저 shutil을 임포트해야 합니다.

import shutil

2. 파일 작업

2.1 파일 복사

shutil.copy() 함수는 파일을 복사하는 데 사용됩니다. 복사된 파일은 원본 파일과 동일한 내용을 가지며, 파일의 메타데이터(예: 권한)도 복사됩니다.

shutil.copy('source.txt', 'destination.txt')
print("파일이 복사되었습니다.")

shutil.copy()는 원본 파일의 데이터를 복사하지만, 파일 메타데이터는 복사하지 않습니다.
만약 파일 권한까지 함께 복사하려면 shutil.copy2()를 사용해야 합니다.

shutil.copy2('source.txt', 'destination.txt')
print("파일과 메타데이터가 복사되었습니다.")

2.2 파일 이동

shutil.move()는 파일을 다른 디렉토리로 이동하거나, 파일의 이름을 변경하는 데 사용됩니다.

shutil.move('source.txt', 'new_folder/source.txt')
print("파일이 이동되었습니다.")

이 함수는 파일 이동뿐만 아니라 디렉토리 이동에도 사용할 수 있으며, 같은 이름으로 파일을 옮길 경우 이름을 변경하는 기능도 수행합니다.

2.3 파일 삭제

shutil 모듈은 파일 삭제 기능을 제공하지 않지만, Python의 기본 os.remove() 함수를 함께 사용하여 파일을 삭제할 수 있습니다.

import os

os.remove('source.txt')
print("파일이 삭제되었습니다.")

3. 디렉토리 작업

3.1 디렉토리 복사

shutil.copytree() 함수는 디렉토리와 그 안에 있는 모든 파일 및 하위 디렉토리를 복사합니다.

shutil.copytree('source_folder', 'destination_folder')
print("디렉토리가 복사되었습니다.")

주의: copytree()는 원본 디렉토리 내의 모든 파일 및 하위 디렉토리까지 복사하므로, 대용량의 디렉토리를 복사할 경우 처리 시간이 오래 걸릴 수 있습니다.

3.2 디렉토리 이동

shutil.move()를 사용하여 디렉토리를 다른 위치로 이동할 수 있습니다.

shutil.move('source_folder', 'new_location/source_folder')
print("디렉토리가 이동되었습니다.")

shutil.move()는 디렉토리 구조를 유지한 채로 전체 디렉토리를 이동시킵니다.

3.3 디렉토리 삭제

shutil.rmtree() 함수는 디렉토리와 그 안에 있는 모든 파일을 재귀적으로 삭제합니다. 디렉토리가 비어 있지 않아도 삭제할 수 있는 기능을 제공하므로 매우 강력하지만, 실수로 중요한 데이터를 삭제하지 않도록 주의해야 합니다.

shutil.rmtree('source_folder')
print("디렉토리가 삭제되었습니다.")

3.4 디렉토리 내용 삭제

특정 디렉토리의 내용만 삭제하고, 디렉토리 자체는 유지하고 싶다면 os.listdir()과 shutil.rmtree()를 함께 사용할 수 있습니다.

import os

folder = 'source_folder'

for filename in os.listdir(folder):
    file_path = os.path.join(folder, filename)
    try:
        if os.path.isfile(file_path) or os.path.islink(file_path):
            os.unlink(file_path)  # 파일 삭제
        elif os.path.isdir(file_path):
            shutil.rmtree(file_path)  # 디렉토리 삭제
    except Exception as e:
        print(f'파일 {file_path} 삭제 중 오류 발생: {e}')

이 코드는 source_folder 안에 있는 모든 파일과 폴더를 삭제하지만, source_folder 디렉토리 자체는 삭제하지 않습니다.

4. 압축 및 해제 작업

4.1 디렉토리 압축

shutil 모듈을 사용하면 디렉토리를 쉽게 압축할 수 있습니다. shutil.make_archive() 함수는 디렉토리를 ZIP 또는 TAR 같은 형식으로 압축합니다.

shutil.make_archive('archive_name', 'zip', 'source_folder')
print("디렉토리가 압축되었습니다.")

첫 번째 인자는 압축 파일의 이름이고, 두 번째 인자는 압축 형식입니다(zip, tar 등). 세 번째 인자는 압축할 디렉토리의 경로입니다.

4.2 압축 해제

shutil.unpack_archive()를 사용하여 압축 파일을 해제할 수 있습니다.

shutil.unpack_archive('archive_name.zip', 'output_folder')
print("압축이 해제되었습니다.")

이 함수는 지원하는 여러 압축 형식(zip, tar, gztar, bztar 등)을 해제할 수 있으며, 두 번째 인자로 해제된 파일을 저장할 위치를 지정합니다.

5. 디스크 사용량 확인

shutil.disk_usage() 함수는 특정 경로에 대한 디스크 사용량 정보를 제공합니다.

total, used, free = shutil.disk_usage('/')
print(f"총 용량: {total // (2**30)} GiB")
print(f"사용 중: {used // (2**30)} GiB")
print(f"남은 용량: {free // (2**30)} GiB")

이 함수는 바이트 단위로 총 용량, 사용 중인 용량, 남은 용량을 반환하며, 디스크 사용량을 확인하는 데 유용합니다.

1. shutil 모듈 임포트