From adf734f3c14a19845f2c33372644fb7f4615907f Mon Sep 17 00:00:00 2001
From: Harrison <yunyangdeng@gmail.com>
Date: Fri, 21 Apr 2023 15:52:20 -0500
Subject: [PATCH] Added feature to add basename column to output

---
 .vscode/launch.json |  4 ++--
 csvbyname/cli.py    | 12 +++++++++++-
 2 files changed, 13 insertions(+), 3 deletions(-)

diff --git a/.vscode/launch.json b/.vscode/launch.json
index 610b30d..91dbc3b 100644
--- a/.vscode/launch.json
+++ b/.vscode/launch.json
@@ -8,7 +8,6 @@
             "name": "Use Test Resources",
             "type": "python",
             "request": "launch",
-            "program": "${workspaceFolder}/csvbyname/csvbyname.py",
             "console": "integratedTerminal",
             "args": [
                 "${workspaceFolder}/tests/resources",
@@ -20,7 +19,8 @@
                 "-V",
                 "DEBUG"
             ],
-            "justMyCode": true
+            "justMyCode": true,
+            "module": "csvbyname.cli"
         }
     ]
 }
\ No newline at end of file
diff --git a/csvbyname/cli.py b/csvbyname/cli.py
index 4a815f5..9d7c9cb 100644
--- a/csvbyname/cli.py
+++ b/csvbyname/cli.py
@@ -13,7 +13,7 @@ def run(args):
         args.recursive,
         args.add_re_property,
     )
-    write_collected_to_csv(args.output, collected, pkeys)
+    write_collected_to_csv(args.output, collected, pkeys, args.output_basename)
 
 
 def main():
@@ -64,6 +64,16 @@ def main():
         nargs="+",
         type=str,
     )
+    argparser.add_argument(
+        "-n",
+        "--output-basename",
+        help='Adds a column called "basename" to the resulting CSV where it is just '
+        "The base name of the path instead of the entire path. This is not guaranteed "
+        "to be unique.",
+        default=False,
+        required=False,
+        action="store_true",
+    )
     argparser.add_argument(
         "-V",
         "--verbosity",