From 884555cfb9c0ea0e1fa9623f4e666a4af7837aeb Mon Sep 17 00:00:00 2001
From: djib <me@djib.fr>
Date: Sat, 7 Dec 2024 21:32:27 +0100
Subject: [PATCH] Count number of files

---
 fileextensionstats.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/fileextensionstats.py b/fileextensionstats.py
index cef94af..62f395a 100644
--- a/fileextensionstats.py
+++ b/fileextensionstats.py
@@ -8,9 +8,15 @@ grouped by extension.
 """
 
 stats = defaultdict(int)
-for (dirpath, _, filenames) in os.walk(r"/put/some/path/here"):
+count = defaultdict(int)
+total = 0
+for (dirpath, _, filenames) in os.walk(r"/home/djib/Images/original"):
     for filename in filenames:
         _, extension = os.path.splitext(filename)
         stats[extension] += os.path.getsize(os.path.join(dirpath, filename))
+        count[extension] += 1
+        total += 1
 
-pprint(stats)
\ No newline at end of file
+pprint(stats)
+pprint(count)
+pprint(total)