TiagoRG
/
uaveiro-leci
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added invalid directory exceptions to fileList.py

This commit is contained in:
tiagorg 2022-11-04 20:02:20 +00:00
parent 4e4438a5ee
commit 693e64dd07
1 changed files with 25 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

41
1ano/fp/aula06/fileList.py
View File

@ -8,22 +8,31 @@ def main():
def printFilesSize(path):
print(f'|{"-"*78}|')
print(f'| {"File":<63} {"Size":>12} |\n|{"-"*78}|')
for file in os.listdir(path):
base_size = os.stat(f'{path}/{file}').st_size
if os.path.isdir(f'{path}/{file}'):
size = "-Directory-"
elif base_size < 1024:
size = str(base_size) + ' B'
elif base_size < 1024 ** 2:
size = f'{base_size // 1024}.{str(base_size % 1024)[0]} KB'
elif base_size < 1024 ** 3:
size = f'{base_size // (1024**2)}.{str(base_size % (1024**2))[0]} MB'
else:
size = f'{base_size // (1024**3)}.{str(base_size % (1024**3))[0]} GB'
print(f'| {file:<63} {size:>12} |')
print(f'|{"-"*78}|')
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'|{"-"*78}|')
print(f'| {"File":<63} {"Size":>12} |\n|{"-"*78}|')
for file in directory:
base_size = os.stat(f'{path}/{file}').st_size
if os.path.isdir(f'{path}/{file}'):
size = "-Directory-"
elif base_size < 1024:
size = str(base_size) + ' B'
elif base_size < 1024 ** 2:
size = f'{base_size // 1024}.{str(base_size % 1024)[0]} KB'
elif base_size < 1024 ** 3:
size = f'{base_size // (1024**2)}.{str(base_size % (1024**2))[0]} MB'
else:
size = f'{base_size // (1024**3)}.{str(base_size % (1024**3))[0]} GB'
print(f'| {file:<63} {size:>12} |')
print(f'|{"-"*78}|')
if __name__ == "__main__":