patrick/pargv
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix hyphens in arguments issue

Browse Source
This commit is contained in:
Patrick Elmer 2023-08-14 11:24:17 +09:00
parent 563b895e42
commit 94d774ed71
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pargv.py
View File

@ -12,7 +12,7 @@ def parse_args(argv=None):
for arg in argv: for arg in argv:
if arg.startswith('-'): if arg.startswith('-'):
break break
args.append(arg) args.append(arg.replace('-', '_'))
argv = argv[len(args):] argv = argv[len(args):]
kwargs = {} kwargs = {}

6
tests/test_functionality.py
View File

@ -29,6 +29,12 @@ def test_single_positional_argument():
assert kwargs == {} assert kwargs == {}
def test_single_positional_argument_with_underscore():
args, kwargs = parse_args(['the-app'])
assert args == ['the_app']
assert kwargs == {}
def test_positional_arguments(): def test_positional_arguments():
args, kwargs = parse_args(['app', 'command']) args, kwargs = parse_args(['app', 'command'])
assert args == ['app', 'command'] assert args == ['app', 'command']