Skip to content

Commit 7a3da26

Browse files
committed
Added Callable types for argparse-based commands which use with_unknown_args.
1 parent 7dbd3a3 commit 7a3da26

File tree

1 file changed

+19
-11
lines changed

1 file changed

+19
-11
lines changed

cmd2/decorators.py

Lines changed: 19 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -118,17 +118,17 @@ def _arg_swap(args: Union[Sequence[Any]], search_arg: Any, *replace_arg: Any) ->
118118
return args_list
119119

120120

121-
#: Function signature for an Command Function that accepts a pre-processed argument list from user input
121+
#: Function signature for a command function that accepts a pre-processed argument list from user input
122122
#: and optionally returns a boolean
123123
ArgListCommandFuncOptionalBoolReturn = Callable[[CommandParent, List[str]], Optional[bool]]
124-
#: Function signature for an Command Function that accepts a pre-processed argument list from user input
124+
#: Function signature for a command function that accepts a pre-processed argument list from user input
125125
#: and returns a boolean
126126
ArgListCommandFuncBoolReturn = Callable[[CommandParent, List[str]], bool]
127-
#: Function signature for an Command Function that accepts a pre-processed argument list from user input
127+
#: Function signature for a command function that accepts a pre-processed argument list from user input
128128
#: and returns Nothing
129129
ArgListCommandFuncNoneReturn = Callable[[CommandParent, List[str]], None]
130130

131-
#: Aggregate of all accepted function signatures for Command Functions that accept a pre-processed argument list
131+
#: Aggregate of all accepted function signatures for command functions that accept a pre-processed argument list
132132
ArgListCommandFunc = Union[
133133
ArgListCommandFuncOptionalBoolReturn[CommandParent],
134134
ArgListCommandFuncBoolReturn[CommandParent],
@@ -249,21 +249,29 @@ def _set_parser_prog(parser: argparse.ArgumentParser, prog: str) -> None:
249249
req_args.append(action.dest)
250250

251251

252-
#: Function signature for a Command Function that uses an argparse.ArgumentParser to process user input
253-
#: and optionally returns a boolean
252+
#: Function signatures for command functions that use an argparse.ArgumentParser to process user input
253+
#: and optionally return a boolean
254254
ArgparseCommandFuncOptionalBoolReturn = Callable[[CommandParent, argparse.Namespace], Optional[bool]]
255-
#: Function signature for a Command Function that uses an argparse.ArgumentParser to process user input
256-
#: and returns a boolean
255+
ArgparseCommandFuncWithUnknownArgsOptionalBoolReturn = Callable[[CommandParent, argparse.Namespace, List[str]], Optional[bool]]
256+
257+
#: Function signatures for command functions that use an argparse.ArgumentParser to process user input
258+
#: and return a boolean
257259
ArgparseCommandFuncBoolReturn = Callable[[CommandParent, argparse.Namespace], bool]
258-
#: Function signature for an Command Function that uses an argparse.ArgumentParser to process user input
259-
#: and returns nothing
260+
ArgparseCommandFuncWithUnknownArgsBoolReturn = Callable[[CommandParent, argparse.Namespace, List[str]], bool]
261+
262+
#: Function signatures for command functions that use an argparse.ArgumentParser to process user input
263+
#: and return nothing
260264
ArgparseCommandFuncNoneReturn = Callable[[CommandParent, argparse.Namespace], None]
265+
ArgparseCommandFuncWithUnknownArgsNoneReturn = Callable[[CommandParent, argparse.Namespace, List[str]], None]
261266

262-
#: Aggregate of all accepted function signatures for an argparse Command Function
267+
#: Aggregate of all accepted function signatures for an argparse command function
263268
ArgparseCommandFunc = Union[
264269
ArgparseCommandFuncOptionalBoolReturn[CommandParent],
270+
ArgparseCommandFuncWithUnknownArgsOptionalBoolReturn[CommandParent],
265271
ArgparseCommandFuncBoolReturn[CommandParent],
272+
ArgparseCommandFuncWithUnknownArgsBoolReturn[CommandParent],
266273
ArgparseCommandFuncNoneReturn[CommandParent],
274+
ArgparseCommandFuncWithUnknownArgsNoneReturn[CommandParent],
267275
]
268276

269277

0 commit comments

Comments
 (0)