Autocomplete for products

pyproject.toml

@@ -47,6 +47,7 @@ dependencies = [
   "python-dateutil",
   "python-slugify",
   "termcolor>=2.1",
+  "argcomplete>=3.4"
 ]
 optional-dependencies.tests = [
   "freezegun",

src/norwegianblue/__init__.py

@@ -99,20 +99,20 @@ def norwegianblue(
 
     return output
 
+def all_products() -> list[str]:
+    """Get all known products from the API or cache"""
+    return norwegianblue("all").splitlines()
 
 @lru_cache(maxsize=None)
 def suggest_product(product: str) -> str:
+    """Provide the best suggestion based on a typed product"""
     import difflib
 
-    # Get all known products from the API or cache
-    all_products = norwegianblue("all").splitlines()
-
     # Find the closest match
-    result = difflib.get_close_matches(product, all_products, n=1)
+    result = difflib.get_close_matches(product, all_products(), n=1)
     logging.info("Suggestion:\t%s (score: %d)", *result)
     return result[0] if result else ""
 
-
 def _ltsify(data: list[dict]) -> list[dict]:
     """If a cycle is LTS, append LTS to the cycle version and remove the LTS column"""
     for cycle in data:

src/norwegianblue/cli.py

@@ -14,6 +14,7 @@
 from __future__ import annotations
 
 import argparse
+import argcomplete
 import atexit
 import logging
 import platform
@@ -26,6 +27,9 @@
 
 atexit.register(_cache.clear)
 
+def ProductCompleter(**kwargs):
+    """The list of all products to feed autocompletion"""
+    return norwegianblue.all_products()
 
 def main() -> None:
     parser = argparse.ArgumentParser(
@@ -36,7 +40,7 @@ def main() -> None:
         nargs="*",
         default=["all"],
         help="product to check, or 'all' to list all available (default: 'all')",
-    )
+    ).completer =  ProductCompleter
     parser.add_argument(
         "-f",
         "--format",
@@ -115,7 +119,7 @@ def main() -> None:
             help=f"output in {help_text}",
         )
     parser.set_defaults(formatter="pretty")
-
+    argcomplete.autocomplete(parser)
     args = parser.parse_args()
 
     if args.format: