diff options
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 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 |