supporting on/off -1

Author: delmorallopez

implement-shell-tools/ls/script-ls.py

@@ -5,45 +5,75 @@
 import argparse
 
 
-def list_directory(path, show_all):
+def list_directory(path, show_all, one_per_line):
     try:
+        # If path is a file, just print it
         if os.path.isfile(path):
-            # ls file.txt → just print the file name
             print(path)
             return
 
+        # If path is a directory, list its contents
         if os.path.isdir(path):
             entries = os.listdir(path)
         else:
-            print(f"ls: cannot access '{path}': No such file or directory", file=sys.stderr)
+            print(
+                f"ls: cannot access '{path}': No such file or directory",
+                file=sys.stderr,
+            )
             return
 
     except PermissionError:
-        print(f"ls: cannot open directory '{path}': Permission denied", file=sys.stderr)
+        print(
+            f"ls: cannot open directory '{path}': Permission denied",
+            file=sys.stderr,
+        )
         return
 
-    # If -a is not provided, hide dotfiles
+    # Hide dotfiles unless -a is used
     if not show_all:
         entries = [e for e in entries if not e.startswith(".")]
 
     entries.sort()
 
-    for entry in entries:
-        print(entry)
+    # Output formatting
+    if one_per_line:
+        for entry in entries:
+            print(entry)
+    else:
+        print(" ".join(entries))
+
 
 def main():
     parser = argparse.ArgumentParser(description="Simple ls implementation")
+
     parser.add_argument(
         "-a",
         action="store_true",
         help="include directory entries whose names begin with a dot",
     )
 
-    parser.add_argument("path", nargs="?", default=".", help="directory to list")
+    parser.add_argument(
+        "-1",
+        dest="one_per_line",
+        action="store_true",
+        help="list one file per line",
+    )
+
+    parser.add_argument(
+        "path",
+        nargs="?",
+        default=".",
+        help="directory or file to list",
+    )
 
     args = parser.parse_args()
 
-    list_directory(args.path, show_all=args.a)
+    list_directory(
+        args.path,
+        show_all=args.a,
+        one_per_line=args.one_per_line,
+    )
+
 
 if __name__ == "__main__":
     main()