Finding the size of the directory in Python · cho.sh

Finding the size of the directory in Python · cho.sh

text

print(getFileSize("C:\\Users\\anacl\\OneDrive\\Documents"))
print(getFileSize("C:\\Users\\anacl\\OneDrive\\Pictures"))
print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Pictures\\screenshot.png"
    )
)

# Expected Output
# [583, 404, 138, 3]
# [749, 426, 83]
# [926, 139]

text

print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Documents", readable=True
    )
)
print(
    getFileSize("C:\\Users\\anacl\\OneDrive\\Pictures", readable=True)
)
print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Pictures\\screenshot.png",
        readable=True,
    )
)
# Expected Output
# 3GB 138MB 404KB 583B
# 83MB 426KB 749B
# 139KB 926B

text

print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Documents",
        readable=True,
        shortRead=True,
    )
)
print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Pictures",
        readable=True,
        shortRead=True,
    )
)
print(
    getFileSize(
        "C:\\Users\\anacl\\OneDrive\\Pictures\\screenshot.png",
        readable=True,
        shortRead=True,
    )
)
# Expected Output
# 3.134GB
# 83.416MB
# 139.904KB

python

import os
import math


def getFileSize(path, kilo=1024, readable=False, shortRead=False):
    size = 0
    sizeArr = []
    units = ["B", "KB", "MB", "GB", "TB", "PB", "EB"]
    if os.path.isdir(path):
        for dirpath, dirnames, filenames in os.walk(path):
            for i in filenames:
                size += os.path.getsize(os.path.join(dirpath, i))
    elif os.path.isfile(path):
        size += os.path.getsize(path)
    unit = math.floor(math.log(size, kilo))
    for k in range(0, unit + 1):
        sizeArr.append(
            math.floor((size % kilo ** (k + 1)) / kilo ** k)
        )

    if readable:
        sizeString = ""
        if not shortRead:
            for x in range(unit, -1, -1):
                sizeString += str(sizeArr[x]) + units[x] + " "
            return sizeString[:-1]
        else:
            return (
                str(sizeArr[-1])
                + "."
                + str(math.floor(sizeArr[-2] / 1.024))
                + units[len(sizeArr) - 1]
            )
    else:
        return sizeArr