Fix completion for Fire(fn) (#336)

google · Dec 9, 2022 · b2415b9 · b2415b9
1 parent 7b5d4f7
commit b2415b9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/fire/completion.py b/fire/completion.py
@@ -156,7 +156,11 @@ def _GetOptsAssignmentTemplate(command):
       return opts_assignment_subcommand_template
 
   lines = []
-  for command in set(subcommands_map.keys()).union(set(options_map.keys())):
+  commands_set = set()
+  commands_set.add(name)
+  commands_set = commands_set.union(set(subcommands_map.keys()))
+  commands_set = commands_set.union(set(options_map.keys()))
+  for command in commands_set:
     opts_assignment = _GetOptsAssignmentTemplate(command).format(
         options=' '.join(
             sorted(options_map[command].union(subcommands_map[command]))