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)