aboutsummaryrefslogtreecommitdiff
path: root/lib/python/milc.py
diff options
context:
space:
mode:
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