파이썬에서 디렉토리의 총 크기를 계산하는 방법

파이썬에서 디렉토리의 총 크기를 계산하는 방법에 대한 튜토리얼입니다.

파이썬에서 디렉토리의 총 크기를 계산하는 방법에 대한 단계별 튜토리얼입니다.

1. 필요한 모듈을 가져옵니다:

import os

2. 파일 또는 디렉토리의 크기를 계산하는 함수를 정의합니다:

def get_size(path):
    total_size = 0

    # 경로가 파일인 경우 크기를 반환합니다
    if os.path.isfile(path):
        return os.path.getsize(path)

    # 경로가 디렉토리인 경우 내용물을 반복하면서 총 크기를 계산합니다
    for dirpath, dirnames, filenames in os.walk(path):
        # 디렉토리의 모든 파일의 총 크기를 계산합니다
        for f in filenames:
            fp = os.path.join(dirpath, f)
            total_size += os.path.getsize(fp)

    return total_size

3. 사용자에게 디렉토리 경로를 입력하도록 요청합니다:

directory = input("디렉토리 경로를 입력하세요: ")

4. get_size() 함수를 디렉토리 경로와 함께 호출하고 결과를 출력합니다:

size = get_size(directory)
print("총 크기:", size, "바이트")

그게 다입니다! 프로그램을 실행하면 디렉토리 경로를 입력하라는 메시지가 나타납니다. 경로를 입력하면 디렉토리의 총 크기가 계산되어 바이트 단위로 표시됩니다.

다음은 완전한 예제입니다:

import os

def get_size(path):
    total_size = 0

    if os.path.isfile(path):
        return os.path.getsize(path)

    for dirpath, dirnames, filenames in os.walk(path):
        for f in filenames:
            fp = os.path.join(dirpath, f)
            total_size += os.path.getsize(fp)

    return total_size

directory = input("디렉토리 경로를 입력하세요: ")
size = get_size(directory)
print("총 크기:", size, "바이트")

특정 사용 사례에 맞게 코드를 필요에 따라 수정하셔도 됩니다.