Added basic 'argcomplete' completion

completes argument names, sub_parser names and available cores/tools/generators

argcomplete completion can be enabled by two ways:
1. Global:
    can be enabled with the 'activate-global-python-argcomplete' command
    additionaly the fusesoc binary needs the string PYTHON_ARGCOMPLETE_OK in the first 1024 bytes, this needs to be added to the pip binary to work
    this only works for zsh and bash
2: Per command:
    for zsh and bash:
        eval "$(register-python-argcomplete fusesoc)"
    for other shells (e.g. powershell):
        https://github.com/kislyuk/argcomplete/tree/develop/contrib

Author: nicorum

fusesoc/main.py

@@ -1,4 +1,5 @@
 #!/usr/bin/env python
+# PYTHON_ARGCOMPLETE_OK
 # Copyright FuseSoC contributors
 # Licensed under the 2-Clause BSD License, see LICENSE for details.
 # SPDX-License-Identifier: BSD-2-Clause
@@ -11,6 +12,8 @@
 import warnings
 from pathlib import Path
 
+import argcomplete
+
 try:
     from fusesoc.version import version as __version__
 except ImportError:
@@ -393,6 +396,41 @@ def update(fs, args):
     fs.update_libraries(args.libraries)
 
 
+class CoreCompleter:
+    def __call__(self, parsed_args, **kwargs):
+        config = Config(parsed_args.config)
+        args_to_config(parsed_args, config)
+        fs = Fusesoc(config)
+        cores = fs.get_cores()
+        return cores
+
+
+class ToolCompleter:
+    def __call__(self, parsed_args, **kwargs):
+        from edalize.edatool import get_edatool, walk_tool_packages
+
+        _tp = list(walk_tool_packages())
+        tools = []
+        for tool_name in _tp:
+            try:
+                tool_class = get_edatool(tool_name)
+                if tool_class.get_doc(0)["description"]:
+                    tools += [tool_name]
+            # Ignore any misbehaving backends
+            except Exception:
+                pass
+        return tools
+
+
+class GenCompleter:
+    def __call__(self, parsed_args, **kwargs):
+        config = Config(parsed_args.config)
+        args_to_config(parsed_args, config)
+        fs = Fusesoc(config)
+        cores = fs.get_generators()
+        return cores
+
+
 def get_parser():
     parser = argparse.ArgumentParser()
     subparsers = parser.add_subparsers()
@@ -444,7 +482,9 @@ def get_parser():
     parser_core_show = core_subparsers.add_parser(
         "show", help="Show information about a core"
     )
-    parser_core_show.add_argument("core", help="Name of the core to show")
+    parser_core_show.add_argument(
+        "core", help="Name of the core to show"
+    ).completer = CoreCompleter()
     parser_core_show.set_defaults(func=core_info)
 
     # tool subparser
@@ -466,7 +506,7 @@ def get_parser():
     parser_core_info = subparsers.add_parser(
         "core-info", help="Display details about a core"
     )
-    parser_core_info.add_argument("core")
+    parser_core_info.add_argument("core").completer = CoreCompleter()
     parser_core_info.set_defaults(func=core_info)
 
     # gen subparser
@@ -486,7 +526,9 @@ def get_parser():
     parser_gen_show = gen_subparsers.add_parser(
         "show", help="Show information about a generator"
     )
-    parser_gen_show.add_argument("generator", help="Name of the generator to show")
+    parser_gen_show.add_argument(
+        "generator", help="Name of the generator to show"
+    ).completer = GenCompleter()
     parser_gen_show.set_defaults(func=gen_show)
 
     # gen clean subparser
@@ -581,7 +623,9 @@ def get_parser():
     parser_run.add_argument("--build", action="store_true", help="Execute build stage")
     parser_run.add_argument("--run", action="store_true", help="Execute run stage")
     parser_run.add_argument("--target", help="Override default target")
-    parser_run.add_argument("--tool", help="Override default tool for target")
+    parser_run.add_argument(
+        "--tool", help="Override default tool for target"
+    ).completer = ToolCompleter()
     parser_run.add_argument(
         "--flag",
         help="Set custom use flags. Can be specified multiple times",
@@ -601,7 +645,9 @@ def get_parser():
         action="store_true",
         help="Allow additional properties in core files",
     )
-    parser_run.add_argument("system", help="Select a system to operate on")
+    parser_run.add_argument(
+        "system", help="Select a system to operate on"
+    ).completer = CoreCompleter()
     parser_run.add_argument(
         "backendargs", nargs=argparse.REMAINDER, help="arguments to be sent to backend"
     )
@@ -624,6 +670,7 @@ def get_parser():
 def parse_args(argv):
     parser = get_parser()
 
+    argcomplete.autocomplete(parser, always_complete_options=False)
     args = parser.parse_args(argv)
 
     if hasattr(args, "func"):

setup.py

@@ -57,6 +57,7 @@ def read(fname):
         "simplesat>=0.8.0",
         "fastjsonschema",
         "jsonschema2md",
+        "argcomplete",
     ],
     # Supported Python versions: 3.6+
     python_requires=">=3.6, <4",