Count number of files

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)
+pprint(stats)
+pprint(count)
+pprint(total)