diff options
| author | Erovia <erovia@users.noreply.github.com> | 2019-10-13 20:23:11 +0200 |
|---|---|---|
| committer | skullydazed <skullydazed@users.noreply.github.com> | 2019-10-23 22:46:30 -0700 |
| commit | a5a31a5fc0f14f4f66cf362ee85747be159e364d (patch) | |
| tree | 6e9788f22fa7c8207bd217bae7a2b6e50450b7a5 /lib/python/milc.py | |
| parent | 4da9d2ef6f3f3a91738a209f8e692d7294ef70d5 (diff) | |
| download | qmk_firmware-a5a31a5fc0f14f4f66cf362ee85747be159e364d.tar.gz qmk_firmware-a5a31a5fc0f14f4f66cf362ee85747be159e364d.zip | |
MILC: Use dashes instead of underscores for subcommands
The subcommand functions' name follows the Python convention of using
snake case, but looks odd on the command line.
Fix it by converting underscores to dashes, eg.: list_keyboards ->
list-keyboards.
Diffstat (limited to 'lib/python/milc.py')
| -rw-r--r-- | lib/python/milc.py | 7 |
1 files changed, 4 insertions, 3 deletions
diff --git a/lib/python/milc.py b/lib/python/milc.py index 1a29bb25c8..7b130bdea6 100644 --- a/lib/python/milc.py +++ b/lib/python/milc.py @@ -429,11 +429,12 @@ class MILC(object): self.arg_only.append(arg_name) del kwargs['arg_only'] + name = handler.__name__.replace("_", "-") if handler is self._entrypoint: self.add_argument(*args, **kwargs) - elif handler.__name__ in self.subcommands: - self.subcommands[handler.__name__].add_argument(*args, **kwargs) + elif name in self.subcommands: + self.subcommands[name].add_argument(*args, **kwargs) else: raise RuntimeError('Decorated function is not entrypoint or subcommand!') @@ -599,7 +600,7 @@ class MILC(object): self.add_subparsers() if not name: - name = handler.__name__ + name = handler.__name__.replace("_", "-") self.acquire_lock() kwargs['help'] = description |