118 lines
4.2 KiB
Python
Executable File
118 lines
4.2 KiB
Python
Executable File
#!/bin/python3
|
|
|
|
import os
|
|
from sys import argv
|
|
|
|
|
|
class Tcolors:
|
|
HEADER = '\033[95m'
|
|
OKBLUE = '\033[94m'
|
|
OKCYAN = '\033[96m'
|
|
OKGREEN = '\033[92m'
|
|
DARKERGREEN = '\033[32m'
|
|
WARNING = '\033[93m'
|
|
FAIL = '\033[91m'
|
|
ENDC = '\033[0m'
|
|
BOLD = '\033[1m'
|
|
UNDERLINE = '\033[4m'
|
|
|
|
|
|
def main():
|
|
if len(argv) > 2:
|
|
print(f"{Tcolors.FAIL}[Error] Too many arguments.{Tcolors.ENDC}")
|
|
exit(1)
|
|
path = argv[1] if len(argv) > 1 and argv[1] != '-r' else os.getcwd()
|
|
reverse = True if '-r' in argv else False
|
|
print_files_size(path, reverse)
|
|
|
|
|
|
def print_files_size(path, reverse):
|
|
files = []
|
|
try:
|
|
directory = os.listdir(path)
|
|
except FileNotFoundError:
|
|
print(f"{Tcolors.FAIL}[Error] Unable to find that directory: '{os.path.abspath(path)}'{Tcolors.ENDC}")
|
|
exit(1)
|
|
except NotADirectoryError:
|
|
print(f"{Tcolors.FAIL}[Error] Path is not a directory: '{os.path.abspath(path)}'{Tcolors.ENDC}")
|
|
exit(1)
|
|
except PermissionError:
|
|
print(f"{Tcolors.FAIL}[Error] Permission denied for: '{os.path.abspath(path)}'{Tcolors.ENDC}")
|
|
exit(1)
|
|
else:
|
|
print(f"{Tcolors.HEADER}Scanning....{Tcolors.ENDC}")
|
|
total_size = 0
|
|
for file in directory:
|
|
try:
|
|
size = get_dir_size(f'{path}/{file}') if os.path.isdir(f'{path}/{file}') else os.stat(f'{path}/{file}').st_size
|
|
except PermissionError:
|
|
size = 0
|
|
except FileNotFoundError:
|
|
size = 0
|
|
except OSError:
|
|
size = 0
|
|
total_size += size
|
|
compacted_size, color = compact_size(size)
|
|
if os.path.isdir(f'{path}/{file}'):
|
|
compacted_size = f"{'<DIR>':<5} {compacted_size:>12}"
|
|
files.append((file, size, compacted_size, color))
|
|
files = sorted(files, key=lambda x: x[1], reverse=reverse)
|
|
|
|
print(f"{Tcolors.OKBLUE}|{'-'*(os.get_terminal_size().columns-3)}|\n| {Tcolors.OKCYAN}ID {Tcolors.OKBLUE} | {Tcolors.OKCYAN}'File'{' '*(os.get_terminal_size().columns-31)}Size{Tcolors.OKBLUE} | {Tcolors.OKCYAN} ID{Tcolors.OKBLUE} |\n|{'-'*(os.get_terminal_size().columns-3)}|")
|
|
for index, (file, size, compacted_size, color) in enumerate(files, 1):
|
|
print(f"| {color}{index:<5}{Tcolors.OKBLUE} | {color}{file}{' '*(os.get_terminal_size().columns-len(file)-len(compacted_size)-21)}{compacted_size}{Tcolors.OKBLUE} | {color}{index:>5}{Tcolors.OKBLUE} |")
|
|
total_size, color = compact_size(total_size)
|
|
path_info = f"{len(directory)} files/directories found in {os.path.abspath(path)}"
|
|
path_size_info = f"Total size: {color}{total_size}"
|
|
print(f"|{'-'*(os.get_terminal_size().columns-3)}|\n| {Tcolors.OKCYAN}{path_info}{' '*(os.get_terminal_size().columns-len(path_info)-len(path_size_info))}{path_size_info}{Tcolors.OKBLUE} |\n|{'-'*(os.get_terminal_size().columns-3)}|{Tcolors.ENDC}")
|
|
|
|
|
|
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'
|
|
color = Tcolors.DARKERGREEN
|
|
elif size < 1048576:
|
|
size = str(round(size / 1024, 2)) + ' KB'
|
|
color = Tcolors.DARKERGREEN
|
|
elif size < 1073741824:
|
|
size = str(round(size / 1048576, 2)) + ' MB'
|
|
color = Tcolors.DARKERGREEN
|
|
if int(size.split('.')[0]) >= 50:
|
|
color = Tcolors.OKGREEN
|
|
if int(size.split('.')[0]) >= 500:
|
|
color = Tcolors.WARNING
|
|
elif size < 1099511627776:
|
|
size = str(round(size / 1073741824, 2)) + ' GB'
|
|
if int(size.split('.')[0]) >= 10:
|
|
color = Tcolors.FAIL
|
|
else:
|
|
color = Tcolors.WARNING
|
|
# elif size < 1125899906842624:
|
|
else:
|
|
size = str(round(size / 1099511627776, 2)) + ' TB'
|
|
color = Tcolors.FAIL
|
|
return size, color
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|