Added 'aks nodepool snapshot update' command

src/aks-preview/HISTORY.rst

@@ -12,6 +12,10 @@ To release a new version, please select a new version number (usually plus 1 to
 Pending
 ++++++
 
+0.5.146
++++++++
+* Add support for new snapshot command `az aks nodepool snapshot update`
+
 0.5.145
 +++++++
 * Add support for option --aks-custom-headers to some aks commands

src/aks-preview/azext_aks_preview/_help.py

@@ -2230,6 +2230,17 @@
     short-summary: List nodepool snapshots.
 """
 
+helps['aks nodepool snapshot update'] = """
+    type: command
+    short-summary: update tags on a snapshot of a nodepool.
+
+    examples:
+        - name: Update tags on a nodepool snapshot.
+          text: az aks nodepool snapshot update -g MyResourceGroup -n snapshot1 --tags "foo=bar" "key1=val1"
+        - name: Clear tags on a nodepool snapshot.
+          text: az aks nodepool snapshot update -g MyResourceGroup -n snapshot1 --tags ""
+"""
+
 helps['aks nodepool snapshot create'] = """
     type: command
     short-summary: Create a nodepool snapshot.

src/aks-preview/azext_aks_preview/_params.py

@@ -847,6 +847,12 @@ def load_arguments(self, _):
                        help='The nodepool id.', validator=validate_nodepool_id)
             c.argument('aks_custom_headers')
 
+    for scope in ['aks nodepool snapshot update']:
+        with self.argument_context(scope) as c:
+            c.argument('snapshot_name', options_list=[
+                       '--name', '-n'], required=True, help='The nodepool snapshot name.', validator=validate_snapshot_name)
+            c.argument('tags', tags_type, required=True, help='The tags to set to the snapshot.')
+
     for scope in ['aks nodepool snapshot show', 'aks nodepool snapshot delete']:
         with self.argument_context(scope) as c:
             c.argument('snapshot_name', options_list=[

src/aks-preview/azext_aks_preview/commands.py

@@ -220,6 +220,7 @@ def load_command_table(self, _):
                               table_transformer=aks_show_nodepool_snapshot_table_format)
         g.custom_command('create', 'aks_nodepool_snapshot_create',
                          supports_no_wait=True)
+        g.custom_command('update', 'aks_nodepool_snapshot_update')
         g.custom_command('delete', 'aks_nodepool_snapshot_delete',
                          supports_no_wait=True)
 

src/aks-preview/azext_aks_preview/custom.py

@@ -2302,6 +2302,20 @@ def aks_nodepool_snapshot_create(cmd,    # pylint: disable=too-many-locals,too-m
     return client.create_or_update(resource_group_name, snapshot_name, snapshot, headers=headers)
 
 
+def aks_nodepool_snapshot_update(cmd, client, resource_group_name, snapshot_name, tags):   # pylint: disable=unused-argument
+    TagsObject = cmd.get_models(
+        "TagsObject",
+        resource_type=CUSTOM_MGMT_AKS_PREVIEW,
+        operation_group="snapshots",
+    )
+    tagsObject = TagsObject(
+        tags=tags
+    )
+
+    snapshot = client.update_tags(resource_group_name, snapshot_name, tagsObject)
+    return snapshot
+
+
 def aks_nodepool_snapshot_show(cmd, client, resource_group_name, snapshot_name):   # pylint: disable=unused-argument
     snapshot = client.get(resource_group_name, snapshot_name)
     return snapshot

src/aks-preview/azext_aks_preview/tests/latest/test_aks_commands.py

@@ -1948,6 +1948,9 @@ def test_aks_nodepool_snapshot(self, resource_group, resource_group_location):
         nodepool_name = self.create_random_name('c', 6)
         nodepool_name2 = self.create_random_name('c', 6)
         snapshot_name = self.create_random_name('s', 16)
+        tagVar = "test"
+        tagVal = "value"
+        tags = tagVar+"="+tagVal
 
         self.kwargs.update({
             'resource_group': resource_group,
@@ -1959,7 +1962,8 @@ def test_aks_nodepool_snapshot(self, resource_group, resource_group_location):
             'snapshot_name': snapshot_name,
             'k8s_version': create_version,
             'upgrade_k8s_version': upgrade_version,
-            'ssh_key_value': self.generate_ssh_keys()
+            'ssh_key_value': self.generate_ssh_keys(),
+            'tags': tags
         })
 
         # create an aks cluster not using snapshot
@@ -1993,6 +1997,12 @@ def test_aks_nodepool_snapshot(self, resource_group, resource_group_location):
         })
         print("The snapshot resource id %s " % snapshot_resource_id)
 
+        # update tags on nodepool snapshot
+        update_snapshot_cmd = 'aks nodepool snapshot update --resource-group {resource_group} --name {snapshot_name} --tags {tags} -o json'
+        response = self.cmd(update_snapshot_cmd, checks=[
+            self.check('tags', {tagVar: tagVal})
+        ]).get_output_in_json()
+
         # delete the original AKS cluster
         self.cmd(
             'aks delete -g {resource_group} -n {name} --yes --no-wait', checks=[self.is_empty()])

src/aks-preview/setup.py

@@ -9,7 +9,7 @@
 
 from setuptools import setup, find_packages
 
-VERSION = "0.5.145"
+VERSION = "0.5.146"
 
 CLASSIFIERS = [
     "Development Status :: 4 - Beta",