diff --git a/abschliessen.py b/abschliessen.py index d404e31..ee1c7d5 100755 --- a/abschliessen.py +++ b/abschliessen.py @@ -119,9 +119,12 @@ def deletion_blockers(folder): def size(path): + """Size of a file or folder, readable: "830 B", "17 KB".""" if path.is_file(): - return path.stat().st_size - return sum(p.stat().st_size for p in path.rglob("*") if p.is_file()) + total = path.stat().st_size + else: + total = sum(p.stat().st_size for p in path.rglob("*") if p.is_file()) + return f"{total} B" if total < 1024 else f"{total / 1024:.0f} KB" def confirm(question): @@ -229,7 +232,7 @@ def main(): print(f"\nArbeitsordner {folder}:") for entry in sorted(folder.iterdir()): - print(f" {entry.name}{'/' if entry.is_dir() else ''} ({size(entry) / 1024:.0f} KB)") + print(f" {entry.name}{'/' if entry.is_dir() else ''} ({size(entry)})") if not args.ja and not confirm(f"Arbeitsordner {folder} löschen? [j/N] "): print("Nicht gelöscht.") return