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",