Only suggest yaml or json files with submit

canonical · Nov 23, 2024 · 11ce495 · 11ce495
1 parent 3fb66e0
commit 11ce495
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cli/testflinger_cli/__init__.py b/cli/testflinger_cli/__init__.py
@@ -343,7 +343,11 @@ def _add_submit_args(self, subparsers):
         parser.set_defaults(func=self.submit)
         parser.add_argument("--poll", "-p", action="store_true")
         parser.add_argument("--quiet", "-q", action="store_true")
-        parser.add_argument("filename")
+        parser.add_argument("filename").completer = (
+            argcomplete.completers.FilesCompleter(
+                allowednames=("*.yaml", "*.yml", "*.json")
+            )
+        )
         relative = parser.add_mutually_exclusive_group()
         relative.add_argument(
             "--attachments-relative-to",