Added label command to ansible distribution

fixes: pulp#185

CHANGES/185.feature

@@ -0,0 +1 @@
+Added label command to ansible distribution group

pulpcore/cli/ansible/distribution.py

@@ -20,6 +20,7 @@
     create_command,
     destroy_command,
     href_option,
+    label_command,
     label_select_option,
     list_command,
     name_option,
@@ -79,6 +80,7 @@ def distribution(ctx: click.Context, pulp_ctx: PulpContext, distribution_type: s
 distribution.add_command(show_command(decorators=lookup_options))
 distribution.add_command(destroy_command(decorators=lookup_options))
 distribution.add_command(create_command(decorators=create_options))
+distribution.add_command(label_command(needs=[("core", "3.10.0"), ("ansible", "0.8.0.dev")]))
 
 
 # TODO Add content_guard option

pulpcore/cli/common/context.py

@@ -392,14 +392,14 @@ def delete(self, href: str, non_blocking: bool = False) -> Any:
         )
 
     def set_label(self, href: str, key: str, value: str, non_blocking: bool = False) -> Any:
-        entity = self.show(href)
-        entity["pulp_labels"][key] = value
-        return self.update(href, body=entity, non_blocking=non_blocking)
+        labels = self.show(href)["pulp_labels"]
+        labels[key] = value
+        return self.update(href, body={"pulp_labels": labels}, non_blocking=non_blocking)
 
     def unset_label(self, href: str, key: str, non_blocking: bool = False) -> Any:
-        entity = self.show(href)
-        entity["pulp_labels"].pop(key)
-        return self.update(href, body=entity, non_blocking=non_blocking)
+        labels = self.show(href)["pulp_labels"]
+        labels.pop(key)
+        return self.update(href, body={"pulp_labels": labels}, non_blocking=non_blocking)
 
     def show_label(self, href: str, key: str) -> Any:
         entity = self.show(href)

pulpcore/cli/common/generic.py

@@ -390,10 +390,15 @@ def label_command(**kwargs: Any) -> click.Command:
     if "name" not in kwargs:
         kwargs["name"] = "label"
     decorators = kwargs.pop("decorators", [name_option, href_option])
+    needs = kwargs.pop("needs", [("core", "3.10.0")])
 
     @click.group(**kwargs)
-    def label_group() -> None:
-        pass
+    @pass_pulp_context
+    def label_group(pulp_ctx: PulpContext) -> None:
+        for plugin in needs:
+            min_ver = plugin[1] if len(plugin) >= 2 else None
+            max_ver = plugin[2] if len(plugin) == 3 else None
+            pulp_ctx.needs_plugin(plugin[0], min_ver, max_ver)
 
     @click.command(name="set", help=_("Add or update a label"))
     @click.option("--key", required=True, help=_("Key of the label"))

tests/scripts/pulp_ansible/test_distribution.sh

@@ -27,4 +27,13 @@ expect_succ pulp ansible distribution list
 expect_succ pulp ansible distribution show --name "cli_test_ansible_distro"
 expect_succ pulp ansible distribution update --name "cli_test_ansible_distro" --repository ""
 
+if [ "$(pulp debug has-plugin --name "core" --min-version "3.10.0")" ]
+then
+  if [ "$(pulp debug has-plugin --name "ansible" --min-version "0.8.0.dev")" ]
+  then
+    expect_succ pulp ansible distribution label set --name "cli_test_ansible_distro" --key "test" --value "success"
+  else
+    expect_fail pulp ansible distribution label set --name "cli_test_ansible_distro" --key "test" --value "fail"
+  fi
+fi
 expect_succ pulp ansible distribution destroy --name "cli_test_ansible_distro"