aboutsummaryrefslogtreecommitdiff
path: root/lib/python/milc.py
diff options
context:
space:
mode:
authorErovia <erovia@users.noreply.github.com>2019-10-13 20:23:11 +0200
committerskullydazed <skullydazed@users.noreply.github.com>2019-10-23 22:46:30 -0700
commita5a31a5fc0f14f4f66cf362ee85747be159e364d (patch)
tree6e9788f22fa7c8207bd217bae7a2b6e50450b7a5 /lib/python/milc.py
parent4da9d2ef6f3f3a91738a209f8e692d7294ef70d5 (diff)
downloadqmk_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.py7
1 files changed, 4 insertions, 3 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