NinoCode/Tool_Box/GetTreeSize.py

"""

GetTreeSize

Just testing the new scandir.ENOENT


"""

import scandir

def get_tree_size(path):
    """Return total size of files in given path and subdirs."""
    total = 0
    for entry in scandir.scandir(path):
        if entry.is_dir(follow_symlinks=False):
            total += get_tree_size(entry.path)
        else:
            total += entry.stat(follow_symlinks=False).st_size
    return total


def human(size):

    B = "B"
    KB = "KB" 
    MB = "MB"
    GB = "GB"
    TB = "TB"
    UNITS = [B, KB, MB, GB, TB]
    HUMANFMT = "%f %s"
    HUMANRADIX = 1024.

    for u in UNITS[:-1]:
        if size < HUMANRADIX : return HUMANFMT % (size, u)
        size /= HUMANRADIX

    return HUMANFMT % (size,  UNITS[-1])
Revision:	635
Committed:	Wed Mar 28 20:14:31 2018 UTC (8 years ago) by nino.borges
Content type:	text/x-python
File size:	766 byte(s)
Log Message:	Initial version of a tree size getter that is new and better than the walk way
#	User	Rev	Content
1	nino.borges	635	"""
2
3			GetTreeSize
4
5			Just testing the new scandir.ENOENT
6
7
8			"""
9
10			import scandir
11
12			def get_tree_size(path):
13			"""Return total size of files in given path and subdirs."""
14			total = 0
15			for entry in scandir.scandir(path):
16			if entry.is_dir(follow_symlinks=False):
17			total += get_tree_size(entry.path)
18			else:
19			total += entry.stat(follow_symlinks=False).st_size
20			return total
21
22
23			def human(size):
24
25			B = "B"
26			KB = "KB"
27			MB = "MB"
28			GB = "GB"
29			TB = "TB"
30			UNITS = [B, KB, MB, GB, TB]
31			HUMANFMT = "%f %s"
32			HUMANRADIX = 1024.
33
34			for u in UNITS[:-1]:
35			if size < HUMANRADIX : return HUMANFMT % (size, u)
36			size /= HUMANRADIX
37
38			return HUMANFMT % (size, UNITS[-1])