From 939644289148aafb6956a6ff4dcf9bbb77c4b614 Mon Sep 17 00:00:00 2001
From: Florian Pinault <Florian.Pinault@ecmwf.int>
Date: Fri, 2 Aug 2024 13:40:03 +0000
Subject: [PATCH] autocompletion

---
 pyproject.toml                  | 2 +-
 src/anemoi/registry/__main__.py | 6 ++++++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index b27e4f3..6fc9eaa 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -90,7 +90,7 @@ urls.Documentation = "https://anemoi-registry.readthedocs.io/"
 urls.Homepage = "https://github.com/ecmwf/anemoi-registry/"
 urls.Issues = "https://github.com/ecmwf/anemoi-registry/issues"
 urls.Repository = "https://github.com/ecmwf/anemoi-registry/"
-scripts.anemoi-registry = "anemoi.registry.__main__:main"
+scripts.anemoi-registry = "anemoi.registry.__main__:main_PYTHON_ARGCOMPLETE_OK"
 
 [tool.setuptools.package-data]
 "anemoi.registry" = [
diff --git a/src/anemoi/registry/__main__.py b/src/anemoi/registry/__main__.py
index be940c2..6c51a12 100644
--- a/src/anemoi/registry/__main__.py
+++ b/src/anemoi/registry/__main__.py
@@ -24,5 +24,11 @@ def main():
     cli_main(__version__, __doc__, COMMANDS)
 
 
+def main_PYTHON_ARGCOMPLETE_OK():
+    # this is a workaround for argcomplete
+    # to make sure the srting "PYTHON_ARGCOMPLETE_OK" is in the final executable script
+    return main()
+
+
 if __name__ == "__main__":
     main()