From 4d10624db16353afe16f95f38bb1b1effea55914 Mon Sep 17 00:00:00 2001 From: Luis Martinez Date: Mon, 11 Jul 2022 00:04:28 -0500 Subject: [PATCH] fix --self when using completion --- fire/completion.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/fire/completion.py b/fire/completion.py index ed7a1b61..caf84a19 100644 --- a/fire/completion.py +++ b/fire/completion.py @@ -404,7 +404,13 @@ def Completions(component, verbose=False): Returns: A list of completions for a command that would so far return the component. """ - if inspect.isroutine(component) or inspect.isclass(component): + if inspect.isroutine(component): + spec = inspectutils.GetFullArgSpec(component) + if len(component.__qualname__.split('.')) > 1 and 'self' in spec.args: + spec.args.remove('self') + return _CompletionsFromArgs(spec.args + spec.kwonlyargs) + + if inspect.isclass(component): spec = inspectutils.GetFullArgSpec(component) return _CompletionsFromArgs(spec.args + spec.kwonlyargs)