70 lines
2.1 KiB
Plaintext
70 lines
2.1 KiB
Plaintext
|
#!/bin/python3.11
|
||
|
|
|
||
|
|
import os
|
||
|
|
from sys import argv
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
path = argv[1] if len(argv) > 1 else os.getcwd()
|
||
|
|
print_files_size(path)
|
||
|
|
|
||
|
|
|
||
|
|
def print_files_size(path):
|
||
|
|
try:
|
||
|
|
directory = os.listdir(path)
|
||
|
|
except FileNotFoundError:
|
||
|
|
print(f"[Error] Unable to find that directory: '{os.path.abspath(path)}'")
|
||
|
|
exit(1)
|
||
|
|
except NotADirectoryError:
|
||
|
|
print(f"[Error] Path is not a directory: '{os.path.abspath(path)}'")
|
||
|
|
exit(1)
|
||
|
|
else:
|
||
|
|
print(f"|{'-'*237}|\n| ID | {'File':<200} {'Size':>24} | ID |\n|{'-'*237}|")
|
||
|
|
total_size = 0
|
||
|
|
for index, file in enumerate(directory, 1):
|
||
|
|
size = get_dir_size(f'{path}/{file}') if os.path.isdir(f'{path}/{file}') else os.stat(f'{path}/{file}').st_size
|
||
|
|
total_size += size
|
||
|
|
size = compact_size(size)
|
||
|
|
if os.path.isdir(f'{path}/{file}'):
|
||
|
|
size = f"{'<DIR>':<5} {size:>12}"
|
||
|
|
print(f"| {index:<2} | {file:<184} {size:>40} | {index:>2} |")
|
||
|
|
total_size = compact_size(total_size)
|
||
|
|
print(f"|{'-'*237}|\n| {f'{len(directory)} files/directories found in {os.path.abspath(path)}':<162} {f'Total size: {total_size}':>72} |\n|{'-'*237}|")
|
||
|
|
|
||
|
|
|
||
|
|
def get_dir_size(path):
|
||
|
|
total = 0
|
||
|
|
try:
|
||
|
|
for file in os.listdir(path):
|
||
|
|
if os.path.isfile(f'{path}/{file}'):
|
||
|
|
total += os.stat(f'{path}/{file}').st_size
|
||
|
|
else:
|
||
|
|
total += get_dir_size(f'{path}/{file}')
|
||
|
|
except PermissionError:
|
||
|
|
return 0
|
||
|
|
except FileNotFoundError:
|
||
|
|
return 0
|
||
|
|
except NotADirectoryError:
|
||
|
|
return 0
|
||
|
|
except OSError:
|
||
|
|
return 0
|
||
|
|
return total
|
||
|
|
|
||
|
|
|
||
|
|
def compact_size(size):
|
||
|
|
if size < 1024:
|
||
|
|
size = str(size) + ' B'
|
||
|
|
elif size < 1048576:
|
||
|
|
size = str(round(size / 1024, 2)) + ' KB'
|
||
|
|
elif size < 1073741824:
|
||
|
|
size = str(round(size / 1048576, 2)) + ' MB'
|
||
|
|
elif size < 1099511627776:
|
||
|
|
size = str(round(size / 1073741824, 2)) + ' GB'
|
||
|
|
elif size < 1125899906842624:
|
||
|
|
size = str(round(size / 1099511627776, 2)) + ' TB'
|
||
|
|
return size
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|