Kind of shi that half works

scripts/list_files/list_files.py

@@ -1,15 +1,17 @@
 from pathlib import Path
 
-def main():
+def main(args):
     # Set the directory path
     script_dir = Path(__file__).resolve().parent
     dir_path = Path("")
 
     # Use the current script's directory or specify a path
     base_path = Path(__file__).resolve().parent 
-
-    for entry in dir_path.iterdir():
-        if entry.is_dir():
-            print(f"[FOLDER] {entry.name}")
-        elif entry.is_file():
-            print(f"[FILE]   {entry.name}")
+    if args == []:
+        for entry in dir_path.iterdir():
+            if entry.is_dir():
+                print(f"[FOLDER] {entry.name}")
+            elif entry.is_file():
+                print(f"[FILE]   {entry.name}")
+    elif args == ["--help"] or args == ["-h"]:
+        print("Bro is lazy on doing this help section")