aboutsummaryrefslogtreecommitdiff
path: root/lib/python
diff options
context:
space:
mode:
Diffstat (limited to 'lib/python')
-rw-r--r--lib/python/milc.py7
-rw-r--r--lib/python/qmk/tests/test_cli_commands.py2
2 files changed, 5 insertions, 4 deletions
diff --git a/lib/python/milc.py b/lib/python/milc.py
index 1a29bb25c..7b130bdea 100644
--- a/lib/python/milc.py
+++ b/lib/python/milc.py
@@ -429,11 +429,12 @@ class MILC(object):
429 self.arg_only.append(arg_name) 429 self.arg_only.append(arg_name)
430 del kwargs['arg_only'] 430 del kwargs['arg_only']
431 431
432 name = handler.__name__.replace("_", "-")
432 if handler is self._entrypoint: 433 if handler is self._entrypoint:
433 self.add_argument(*args, **kwargs) 434 self.add_argument(*args, **kwargs)
434 435
435 elif handler.__name__ in self.subcommands: 436 elif name in self.subcommands:
436 self.subcommands[handler.__name__].add_argument(*args, **kwargs) 437 self.subcommands[name].add_argument(*args, **kwargs)
437 438
438 else: 439 else:
439 raise RuntimeError('Decorated function is not entrypoint or subcommand!') 440 raise RuntimeError('Decorated function is not entrypoint or subcommand!')
@@ -599,7 +600,7 @@ class MILC(object):
599 self.add_subparsers() 600 self.add_subparsers()
600 601
601 if not name: 602 if not name:
602 name = handler.__name__ 603 name = handler.__name__.replace("_", "-")
603 604
604 self.acquire_lock() 605 self.acquire_lock()
605 kwargs['help'] = description 606 kwargs['help'] = description
diff --git a/lib/python/qmk/tests/test_cli_commands.py b/lib/python/qmk/tests/test_cli_commands.py
index c9d632517..9a9dc4b95 100644
--- a/lib/python/qmk/tests/test_cli_commands.py
+++ b/lib/python/qmk/tests/test_cli_commands.py
@@ -40,7 +40,7 @@ def test_pyformat():
40 40
41 41
42def test_list_keyboards(): 42def test_list_keyboards():
43 result = check_subcommand('list_keyboards') 43 result = check_subcommand('list-keyboards')
44 assert result.returncode == 0 44 assert result.returncode == 0
45 # check to see if a known keyboard is returned 45 # check to see if a known keyboard is returned
46 # this will fail if handwired/onekey/pytest is removed 46 # this will fail if handwired/onekey/pytest is removed