diff options
Diffstat (limited to 'lib')
| -rw-r--r-- | lib/python/milc.py | 7 | ||||
| -rw-r--r-- | lib/python/qmk/tests/test_cli_commands.py | 2 |
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 | ||
| 42 | def test_list_keyboards(): | 42 | def 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 |
