From f0a8a0ffe22e4d3fe8cc3fd47d0341293587faf0 Mon Sep 17 00:00:00 2001 From: Kamil Cukrowski Date: Mon, 2 Sep 2024 23:15:52 +0200 Subject: [PATCH] listattributes: add shell completion --- src/nomad_tools/entry_listattributes.py | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/src/nomad_tools/entry_listattributes.py b/src/nomad_tools/entry_listattributes.py index baeab3c..128fbb2 100644 --- a/src/nomad_tools/entry_listattributes.py +++ b/src/nomad_tools/entry_listattributes.py @@ -7,7 +7,8 @@ import click import clickdc -from .common import verbose_option, common_options +from .common import common_options, mynomad, verbose_option +from .common_click import completor from .entry_constrainteval import Args, NodesAttributes from .mytabulate import mytabulate @@ -19,7 +20,11 @@ Works similarly to constrainteval. """, ) -@click.argument("nodenameorid", nargs=-1) +@click.argument( + "nodenameorid", + nargs=-1, + shell_complete=completor(lambda: [v["Name"] for v in mynomad.get("nodes")]), +) @clickdc.adddc("args", Args) @verbose_option() @common_options()