Fix hyphens in arguments issue

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

View File

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

View File

@ -29,6 +29,12 @@ def test_single_positional_argument():
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():
args, kwargs = parse_args(['app', 'command'])
assert args == ['app', 'command']