Allow to document count_tab as "count:" argument

This commit is contained in:
Florian Bruhin 2018-12-03 08:44:51 +01:00
parent ede13c76d9
commit 285db344b3

View File

@ -257,9 +257,13 @@ def _get_command_doc_count(cmd, parser):
yield "==== count" yield "==== count"
try: try:
yield parser.arg_descs[param.name] yield parser.arg_descs[param.name]
except KeyError:
try:
yield parser.arg_descs['count']
except KeyError as e: except KeyError as e:
raise KeyError("No description for count arg {!r} of command " raise KeyError("No description for count arg {!r} of "
"{!r}!".format(param.name, cmd.name)) from e "command {!r}!"
.format(param.name, cmd.name)) from e
def _get_command_doc_notes(cmd): def _get_command_doc_notes(cmd):