[containerapp] suspend/resume a container apps job

src/containerapp/HISTORY.rst

@@ -4,6 +4,8 @@ Release History
 ===============
 upcoming
 ++++++
+* 'az containerapp job suspend': Support command to suspend a container apps job
+* 'az containerapp job resume': Support command to resume a container apps job
 * 'az containerapp create': Fix Role assignment error when the default Azure Container Registry could not be found
 *  Upgrade api-version to 2024-10-02-preview
 * 'az containerapp create/update': `--yaml` support property pollingInterval and cooldownPeriod

src/containerapp/azext_containerapp/_clients.py

@@ -81,6 +81,38 @@ def get_auth_token(cls, cmd, resource_group_name, name):
         r = send_raw_request(cmd.cli_ctx, "POST", request_url)
         return r.json()
 
+    @classmethod
+    def suspend(cls, cmd, resource_group_name, name):
+        management_hostname = cmd.cli_ctx.cloud.endpoints.resource_manager
+        sub_id = get_subscription_id(cmd.cli_ctx)
+        url_fmt = "{}/subscriptions/{}/resourceGroups/{}/providers/Microsoft.App/jobs/{}/suspend?api-version={}"
+        request_url = url_fmt.format(
+            management_hostname.strip('/'),
+            sub_id,
+            resource_group_name,
+            name,
+            cls.api_version)
+
+        r = send_raw_request(cmd.cli_ctx, "POST", request_url)
+        if r.status_code == 202:
+            return
+
+    @classmethod
+    def resume(cls, cmd, resource_group_name, name):
+        management_hostname = cmd.cli_ctx.cloud.endpoints.resource_manager
+        sub_id = get_subscription_id(cmd.cli_ctx)
+        url_fmt = "{}/subscriptions/{}/resourceGroups/{}/providers/Microsoft.App/jobs/{}/resume?api-version={}"
+        request_url = url_fmt.format(
+            management_hostname.strip('/'),
+            sub_id,
+            resource_group_name,
+            name,
+            cls.api_version)
+
+        r = send_raw_request(cmd.cli_ctx, "POST", request_url)
+        if r.status_code == 202:
+            return
+
 
 class ContainerAppsResiliencyPreviewClient():
     api_version = PREVIEW_API_VERSION

src/containerapp/azext_containerapp/_help.py

@@ -543,6 +543,24 @@
       text: az containerapp job stop -n my-containerapp-job -g MyResourceGroup --execution-name-list MyContainerAppJob-66v9xh0,MyContainerAppJob-66v9xh1
 """
 
+helps['containerapp job suspend'] = """
+    type: command
+    short-summary: Suspend a Container Apps Job.
+    examples:
+    - name: Suspend a job.
+      text: |
+          az containerapp job suspend -n my-containerapp-job -g MyResourceGroup
+"""
+
+helps['containerapp job resume'] = """
+    type: command
+    short-summary: Resume a Container Apps Job.
+    examples:
+    - name: Resume a job.
+      text: |
+          az containerapp job resume -n my-containerapp-job -g MyResourceGroup
+"""
+
 # Certificates Commands
 helps['containerapp env certificate'] = """
     type: group

src/containerapp/azext_containerapp/_params.py

@@ -395,6 +395,14 @@ def load_arguments(self, _):
     with self.argument_context('containerapp job registry') as c:
         c.argument('identity', help="The managed identity with which to authenticate to the Azure Container Registry (instead of username/password). Use 'system' for a system-defined identity, Use 'system-environment' for an environment level system-defined identity or a resource id for a user-defined environment/containerapp level identity. The managed identity should have been assigned acrpull permissions on the ACR before deployment (use 'az role assignment create --role acrpull ...').")
 
+    with self.argument_context('containerapp job suspend') as c:
+        c.argument('name', name_type, id_part=None, help="The name of the Containerapp job.")
+        c.argument('resource_group_name', arg_type=resource_group_name_type, id_part=None)
+
+    with self.argument_context('containerapp job resume') as c:
+        c.argument('name', name_type, id_part=None, help="The name of the Containerapp job.")
+        c.argument('resource_group_name', arg_type=resource_group_name_type, id_part=None)
+
     with self.argument_context('containerapp env dotnet-component') as c:
         c.argument('dotnet_component_name', options_list=['--name', '-n'], help="The DotNet component name.")
         c.argument('environment_name', options_list=['--environment'], help="The environment name.")

src/containerapp/azext_containerapp/_sdk_models.py

@@ -8354,6 +8354,9 @@ class ContainerAppsJob(TrackedResource):  # pylint: disable=too-many-instance-at
     :ivar provisioning_state: Provisioning state of the Container Apps Job. Known values are:
      "InProgress", "Succeeded", "Failed", "Canceled", and "Deleting".
     :vartype provisioning_state: str or ~azure.mgmt.appcontainers.models.JobProvisioningState
+    :ivar running_status: Running status of the Container Apps Job. Known values are:
+     "Progressing", "Suspended" and "Ready".
+    :vartype running_status: str or ~azure.mgmt.appcontainers.models.JobRunningState
     :ivar environment_id: Resource ID of environment.
     :vartype environment_id: str
     :ivar workload_profile_name: Workload profile name to pin for container apps job execution.
@@ -8375,6 +8378,7 @@ class ContainerAppsJob(TrackedResource):  # pylint: disable=too-many-instance-at
         "system_data": {"readonly": True},
         "location": {"required": True},
         "provisioning_state": {"readonly": True},
+        "running_status": {"readonly": True},
         "outbound_ip_addresses": {"readonly": True},
         "event_stream_endpoint": {"readonly": True},
     }
@@ -8389,6 +8393,7 @@ class ContainerAppsJob(TrackedResource):  # pylint: disable=too-many-instance-at
         "extended_location": {"key": "extendedLocation", "type": "ExtendedLocation"},
         "identity": {"key": "identity", "type": "ManagedServiceIdentity"},
         "provisioning_state": {"key": "properties.provisioningState", "type": "str"},
+        "running_status": {"key": "properties.runningStatus", "type": "str"},
         "environment_id": {"key": "properties.environmentId", "type": "str"},
         "workload_profile_name": {"key": "properties.workloadProfileName", "type": "str"},
         "configuration": {"key": "properties.configuration", "type": "JobConfiguration"},
@@ -8433,6 +8438,7 @@ def __init__(
         self.extended_location = extended_location
         self.identity = identity
         self.provisioning_state = None
+        self.running_status = None
         self.environment_id = environment_id
         self.workload_profile_name = workload_profile_name
         self.configuration = configuration

src/containerapp/azext_containerapp/commands.py

@@ -43,6 +43,8 @@ def load_command_table(self, args):
         g.custom_command('create', 'create_containerappsjob', supports_no_wait=True, exception_handler=ex_handler_factory(), transform=transform_sensitive_values)
         g.custom_command('update', 'update_containerappsjob', supports_no_wait=True, exception_handler=ex_handler_factory(), transform=transform_sensitive_values)
         g.custom_command('delete', 'delete_containerappsjob', supports_no_wait=True, confirmation=True, exception_handler=ex_handler_factory())
+        g.custom_command('suspend', 'suspend_containerappsjob', supports_no_wait=True, exception_handler=ex_handler_factory())
+        g.custom_command('resume', 'resume_containerappsjob', supports_no_wait=True, exception_handler=ex_handler_factory())
 
     with self.command_group('containerapp job registry', is_preview=True) as g:
         g.custom_command('set', 'set_registry_job', exception_handler=ex_handler_factory())

src/containerapp/azext_containerapp/custom.py

@@ -1017,6 +1017,33 @@ def delete_containerappsjob(cmd, name, resource_group_name, no_wait=False):
     return containerapp_job_decorator.delete()
 
 
+def suspend_containerappsjob(cmd, name, resource_group_name):
+    raw_parameters = locals()
+    containerapp_job_decorator = ContainerAppJobDecorator(
+        cmd=cmd,
+        client=ContainerAppsJobPreviewClient,
+        raw_parameters=raw_parameters,
+        models=CONTAINER_APPS_SDK_MODELS
+    )
+    containerapp_job_decorator.validate_subscription_registered(CONTAINER_APPS_RP)
+
+    return ContainerAppsJobPreviewClient.suspend(cmd=cmd, resource_group_name=resource_group_name, name=name)
+
+
+def resume_containerappsjob(cmd, name, resource_group_name):
+    raw_parameters = locals()
+    containerapp_job_decorator = ContainerAppJobDecorator(
+        cmd=cmd,
+        client=ContainerAppsJobPreviewClient,
+        raw_parameters=raw_parameters,
+        models=CONTAINER_APPS_SDK_MODELS
+    )
+
+    containerapp_job_decorator.validate_subscription_registered(CONTAINER_APPS_RP)
+
+    return ContainerAppsJobPreviewClient.resume(cmd=cmd, resource_group_name=resource_group_name, name=name)
+
+
 def create_or_update_github_action(cmd,
                                    name,
                                    resource_group_name,