파이썬에서 파일 및 디렉토리 작업을 수행할 때 대부분의 경우 shutil 모듈을 사용합니다. 이 모듈은 파일 및 디렉토리 관리에 편리한 함수들을 제공해줍니다.

이번 블로그에서는 shutil 모듈을 사용하여 디렉토리의 총 크기를 계산하는 방법을 알아보겠습니다.

1. 필요한 패키지 가져오기

shutil 모듈을 사용하기 위해 우선 필요한 패키지를 가져와야 합니다.

import os
import shutil

2. 디렉토리의 총 크기 계산하기

shutil 모듈을 사용하여 디렉토리의 총 크기를 계산하려면 shutil.disk_usage() 함수를 사용해야 합니다. 이 함수는 디렉토리의 경로를 매개변수로 받아 디스크 사용량에 대한 정보를 반환합니다.

def get_directory_size(directory):
    total_size = 0
    for path, dirs, files in os.walk(directory):
        for f in files:
            fp = os.path.join(path, f)
            total_size += os.path.getsize(fp)
    return total_size

위의 코드에서 os.walk() 함수는 디렉토리를 재귀적으로 순회하면서 하위 디렉토리와 파일을 찾습니다. os.path.getsize() 함수를 사용하여 각 파일의 크기를 구하고, 이를 총 크기에 더해줍니다.

3. 예제 실행하기

이제 위에서 작성한 함수를 사용하여 디렉토리의 총 크기를 계산해보겠습니다.

directory_path = '/path/to/directory'
total_size = get_directory_size(directory_path)
print(f'총 크기: {total_size} bytes')

실행 결과로 디렉토리의 총 크기가 바이트 단위로 출력됩니다.

마무리

shutil 모듈을 사용하여 디렉토리의 총 크기를 계산하는 방법에 대해 알아보았습니다. shutil 모듈은 다양한 파일 및 디렉토리 작업을 위해 유용한 기능들을 제공하므로, 다른 작업에도 적극적으로 활용해보시기 바랍니다.