flesh out lofic for when path is a dir

Author: HassanOHOsman

implement-shell-tools/wc/wc.py

@@ -130,5 +130,25 @@ def counter(item):
             print(f"{stats['characters']} {path}")
         else:
             print(f"{stats['lines']} {stats['words']} {stats['characters']} {path}")
+    elif os.path.isdir(path):
+        directory = "folder"
+
+        for file in os.listdir(directory):
+            path = os.path.join(directory, file)
+
+            if os.path.isfile(path):
+                with open(path, "r") as f:
+                     content = f.read()
+                stats = counter(content)
+                if args.line:
+                    print(f"{stats['lines']} {path}")
+                elif args.word:
+                    print(f"{stats['words']} {path}")
+                elif args.char:
+                    print(f"{stats['characters']} {path}")
+                else:
+                    print(f"{stats['lines']} {stats['words']} {stats['characters']} {path}")
+
+