Custom user confirmation

src/k8s-extension/azext_k8s_extension/commands.py

@@ -20,6 +20,6 @@ def load_command_table(self, _):
                             is_preview=True) \
             as g:
         g.custom_command('create', 'create_k8s_extension')
-        g.custom_command('delete', 'delete_k8s_extension', confirmation=True)
+        g.custom_command('delete', 'delete_k8s_extension')
         g.custom_command('list', 'list_k8s_extension', table_transformer=k8s_extension_list_table_format)
         g.custom_show_command('show', 'show_k8s_extension', table_transformer=k8s_extension_show_table_format)

src/k8s-extension/azext_k8s_extension/custom.py

@@ -191,7 +191,7 @@ def update_k8s_extension(client, resource_group_name, cluster_type, cluster_name
     # return client.update(resource_group_name, cluster_rp, cluster_type, cluster_name, name, upd_extension)
 
 
-def delete_k8s_extension(client, resource_group_name, cluster_name, name, cluster_type):
+def delete_k8s_extension(client, resource_group_name, cluster_name, name, cluster_type, yes=False):
     """Delete an existing Kubernetes Extension.
 
     """
@@ -201,12 +201,12 @@ def delete_k8s_extension(client, resource_group_name, cluster_name, name, cluste
     try:
         extension = client.get(resource_group_name, cluster_rp, cluster_type, cluster_name, name)
     except HttpResponseError:
-        logger.warning("No extension with name '%s' found on cluster '%s', so nothing to delete", cluster_name, name)
+        logger.warning("No extension with name '%s' found on cluster '%s', so nothing to delete", name, cluster_name)
         return None
     extension_class = ExtensionFactory(extension.extension_type.lower())
 
     # If there is any custom delete logic, this will call the logic
-    extension_class.Delete(client, resource_group_name, cluster_name, name, cluster_type)
+    extension_class.Delete(client, resource_group_name, cluster_name, name, cluster_type, yes)
 
     return client.delete(resource_group_name, cluster_rp, cluster_type, cluster_name, name)
 

src/k8s-extension/azext_k8s_extension/partner_extensions/AzureDefender.py

@@ -12,13 +12,13 @@
 from ..vendored_sdks.models import ScopeCluster
 from ..vendored_sdks.models import Scope
 
-from .PartnerExtensionModel import PartnerExtensionModel
+from .DefaultExtension import DefaultExtension
 from .ContainerInsights import _get_container_insights_settings
 
 logger = get_logger(__name__)
 
 
-class AzureDefender(PartnerExtensionModel):
+class AzureDefender(DefaultExtension):
     def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_type, extension_type,
                scope, auto_upgrade_minor_version, release_train, version, target_namespace,
                release_namespace, configuration_settings, configuration_protected_settings,
@@ -59,17 +59,3 @@ def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_t
             configuration_protected_settings=configuration_protected_settings
         )
         return extension_instance, name, create_identity
-
-    def Update(self, extension, auto_upgrade_minor_version, release_train, version):
-        """ExtensionType 'microsoft.azuredefender.kubernetes' specific validations & defaults for Update
-           Must create and return a valid 'ExtensionInstanceUpdate' object.
-
-        """
-        return ExtensionInstanceUpdate(
-            auto_upgrade_minor_version=auto_upgrade_minor_version,
-            release_train=release_train,
-            version=version
-        )
-
-    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type):
-        pass

src/k8s-extension/azext_k8s_extension/partner_extensions/AzureMLKubernetes.py

@@ -10,6 +10,7 @@
 import copy
 from hashlib import md5
 from typing import Any, Dict, List, Tuple
+from azext_k8s_extension.partner_extensions.DefaultExtension import DefaultExtension
 
 import azure.mgmt.relay
 import azure.mgmt.relay.models
@@ -27,7 +28,7 @@
 from msrestazure.azure_exceptions import CloudError
 
 from .._client_factory import cf_resources
-from .PartnerExtensionModel import PartnerExtensionModel
+from .DefaultExtension import DefaultExtension, default_user_confirmation
 from ..vendored_sdks.models import (
     ExtensionInstance,
     ExtensionInstanceUpdate,
@@ -41,7 +42,7 @@
 
 
 # pylint: disable=too-many-instance-attributes
-class AzureMLKubernetes(PartnerExtensionModel):
+class AzureMLKubernetes(DefaultExtension):
     def __init__(self):
         # constants for configuration settings.
         self.DEFAULT_RELEASE_NAMESPACE = 'azureml'
@@ -155,18 +156,12 @@ def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_t
         )
         return extension_instance, name, create_identity
 
-    def Update(self, extension, auto_upgrade_minor_version, release_train, version):
-        return ExtensionInstanceUpdate(
-            auto_upgrade_minor_version=auto_upgrade_minor_version,
-            release_train=release_train,
-            version=version
-        )
-
-    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type):
+    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type, yes):
         # Give a warning message
         logger.warning("If nvidia.com/gpu or fuse resource is not recognized by kubernetes after this deletion, "
                        "you probably have installed nvidia-device-plugin or fuse-device-plugin before installing AMLArc extension. "
                        "Please try to reinstall device plugins to fix this issue.")
+        default_user_confirmation(yes)
 
     def __validate_config(self, configuration_settings, configuration_protected_settings):
         # perform basic validation of the input config

src/k8s-extension/azext_k8s_extension/partner_extensions/ContainerInsights.py

@@ -21,15 +21,15 @@
 from ..vendored_sdks.models import ScopeCluster
 from ..vendored_sdks.models import Scope
 
-from .PartnerExtensionModel import PartnerExtensionModel
+from .DefaultExtension import DefaultExtension
 
 from .._client_factory import (
     cf_resources, cf_resource_groups, cf_log_analytics)
 
 logger = get_logger(__name__)
 
 
-class ContainerInsights(PartnerExtensionModel):
+class ContainerInsights(DefaultExtension):
     def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_type, extension_type,
                scope, auto_upgrade_minor_version, release_train, version, target_namespace,
                release_namespace, configuration_settings, configuration_protected_settings,
@@ -71,20 +71,6 @@ def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_t
         )
         return extension_instance, name, create_identity
 
-    def Update(self, extension, auto_upgrade_minor_version, release_train, version):
-        """ExtensionType 'microsoft.azuremonitor.containers' specific validations & defaults for Update
-           Must create and return a valid 'ExtensionInstanceUpdate' object.
-
-        """
-        return ExtensionInstanceUpdate(
-            auto_upgrade_minor_version=auto_upgrade_minor_version,
-            release_train=release_train,
-            version=version
-        )
-
-    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type):
-        pass
-
 
 # Custom Validation Logic for Container Insights
 

src/k8s-extension/azext_k8s_extension/partner_extensions/DefaultExtension.py

@@ -5,6 +5,8 @@
 
 # pylint: disable=unused-argument
 
+from azure.cli.core.util import user_confirmation
+
 from ..vendored_sdks.models import ExtensionInstance
 from ..vendored_sdks.models import ExtensionInstanceUpdate
 from ..vendored_sdks.models import ScopeCluster
@@ -56,5 +58,10 @@ def Update(self, extension, auto_upgrade_minor_version, release_train, version):
             version=version
         )
 
-    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type):
-        pass
+    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type, yes):
+        default_user_confirmation(yes)
+
+
+def default_user_confirmation(yes):
+    print(yes)
+    return user_confirmation("Are you sure you want to perform this operation?", yes=yes)

src/k8s-extension/azext_k8s_extension/partner_extensions/OpenServiceMesh.py

@@ -7,6 +7,7 @@
 # pylint: disable=redefined-outer-name
 # pylint: disable=no-member
 
+from azext_k8s_extension.partner_extensions.DefaultExtension import DefaultExtension
 from knack.log import get_logger
 
 from azure.cli.core.azclierror import InvalidArgumentValueError, RequiredArgumentMissingError
@@ -16,9 +17,7 @@
 import yaml
 import requests
 
-from ..partner_extensions import PartnerExtensionModel
-
-from .PartnerExtensionModel import PartnerExtensionModel
+from .DefaultExtension import DefaultExtension
 
 from ..vendored_sdks.models import (
     ExtensionInstance,
@@ -32,7 +31,7 @@
 logger = get_logger(__name__)
 
 
-class OpenServiceMesh(PartnerExtensionModel):
+class OpenServiceMesh(DefaultExtension):
 
     def Create(self, cmd, client, resource_group_name, cluster_name, name, cluster_type, extension_type,
                scope, auto_upgrade_minor_version, release_train, version, target_namespace,
@@ -100,9 +99,6 @@ def Update(self, extension, auto_upgrade_minor_version, release_train, version):
             version=version
         )
 
-    def Delete(self, client, resource_group_name, cluster_name, name, cluster_type):
-        pass
-
 
 def _validate_tested_distro(cmd, cluster_resource_group_name, cluster_name, extension_version):
 

src/k8s-extension/azext_k8s_extension/partner_extensions/PartnerExtensionModel.py

@@ -23,5 +23,5 @@ def Update(self, extension: ExtensionInstance, auto_upgrade_minor_version: bool,
         pass
 
     @abstractmethod
-    def Delete(self, client, resource_group_name: str, cluster_name: str, name: str, cluster_type: str):
+    def Delete(self, client, resource_group_name: str, cluster_name: str, name: str, cluster_type: str, yes: bool):
         pass