141 lines
5.5 KiB
Python
141 lines
5.5 KiB
Python
"""Handles incoming emrcontainers requests, invokes methods, returns responses."""
|
|
import json
|
|
from moto.core.common_types import TYPE_RESPONSE
|
|
from moto.core.responses import BaseResponse
|
|
from .models import emrcontainers_backends, EMRContainersBackend
|
|
|
|
DEFAULT_MAX_RESULTS = 100
|
|
DEFAULT_NEXT_TOKEN = ""
|
|
DEFAULT_CONTAINER_PROVIDER_TYPE = "EKS"
|
|
|
|
|
|
class EMRContainersResponse(BaseResponse):
|
|
"""Handler for EMRContainers requests and responses."""
|
|
|
|
def __init__(self) -> None:
|
|
super().__init__(service_name="emr-containers")
|
|
|
|
@property
|
|
def emrcontainers_backend(self) -> EMRContainersBackend:
|
|
"""Return backend instance specific for this region."""
|
|
return emrcontainers_backends[self.current_account][self.region]
|
|
|
|
def create_virtual_cluster(self) -> TYPE_RESPONSE:
|
|
name = self._get_param("name")
|
|
container_provider = self._get_param("containerProvider")
|
|
client_token = self._get_param("clientToken")
|
|
tags = self._get_param("tags")
|
|
|
|
virtual_cluster = self.emrcontainers_backend.create_virtual_cluster(
|
|
name=name,
|
|
container_provider=container_provider,
|
|
client_token=client_token,
|
|
tags=tags,
|
|
)
|
|
return 200, {}, json.dumps(dict(virtual_cluster))
|
|
|
|
def delete_virtual_cluster(self) -> TYPE_RESPONSE:
|
|
cluster_id = self._get_param("virtualClusterId")
|
|
|
|
virtual_cluster = self.emrcontainers_backend.delete_virtual_cluster(
|
|
cluster_id=cluster_id
|
|
)
|
|
return 200, {}, json.dumps(dict(virtual_cluster))
|
|
|
|
def describe_virtual_cluster(self) -> TYPE_RESPONSE:
|
|
cluster_id = self._get_param("virtualClusterId")
|
|
|
|
virtual_cluster = self.emrcontainers_backend.describe_virtual_cluster(
|
|
cluster_id=cluster_id
|
|
)
|
|
response = {"virtualCluster": virtual_cluster}
|
|
return 200, {}, json.dumps(response)
|
|
|
|
def list_virtual_clusters(self) -> TYPE_RESPONSE:
|
|
container_provider_id = self._get_param("containerProviderId")
|
|
container_provider_type = self._get_param(
|
|
"containerProviderType", DEFAULT_CONTAINER_PROVIDER_TYPE
|
|
)
|
|
created_after = self._get_param("createdAfter")
|
|
created_before = self._get_param("createdBefore")
|
|
states = self.querystring.get("states", [])
|
|
max_results = self._get_int_param("maxResults", DEFAULT_MAX_RESULTS)
|
|
next_token = self._get_param("nextToken", DEFAULT_NEXT_TOKEN)
|
|
|
|
virtual_clusters, next_token = self.emrcontainers_backend.list_virtual_clusters(
|
|
container_provider_id=container_provider_id,
|
|
container_provider_type=container_provider_type,
|
|
created_after=created_after,
|
|
created_before=created_before,
|
|
states=states,
|
|
max_results=max_results,
|
|
next_token=next_token,
|
|
)
|
|
|
|
response = {"virtualClusters": virtual_clusters, "nextToken": next_token}
|
|
return 200, {}, json.dumps(response)
|
|
|
|
def start_job_run(self) -> TYPE_RESPONSE:
|
|
name = self._get_param("name")
|
|
virtual_cluster_id = self._get_param("virtualClusterId")
|
|
client_token = self._get_param("clientToken")
|
|
execution_role_arn = self._get_param("executionRoleArn")
|
|
release_label = self._get_param("releaseLabel")
|
|
job_driver = self._get_param("jobDriver")
|
|
configuration_overrides = self._get_param("configurationOverrides")
|
|
tags = self._get_param("tags")
|
|
|
|
job = self.emrcontainers_backend.start_job_run(
|
|
name=name,
|
|
virtual_cluster_id=virtual_cluster_id,
|
|
client_token=client_token,
|
|
execution_role_arn=execution_role_arn,
|
|
release_label=release_label,
|
|
job_driver=job_driver,
|
|
configuration_overrides=configuration_overrides,
|
|
tags=tags,
|
|
)
|
|
return 200, {}, json.dumps(dict(job))
|
|
|
|
def cancel_job_run(self) -> TYPE_RESPONSE:
|
|
job_id = self._get_param("jobRunId")
|
|
virtual_cluster_id = self._get_param("virtualClusterId")
|
|
|
|
job = self.emrcontainers_backend.cancel_job_run(
|
|
job_id=job_id, virtual_cluster_id=virtual_cluster_id
|
|
)
|
|
return 200, {}, json.dumps(dict(job))
|
|
|
|
def list_job_runs(self) -> TYPE_RESPONSE:
|
|
virtual_cluster_id = self._get_param("virtualClusterId")
|
|
created_before = self._get_param("createdBefore")
|
|
created_after = self._get_param("createdAfter")
|
|
name = self._get_param("name")
|
|
states = self.querystring.get("states", [])
|
|
max_results = self._get_int_param("maxResults", DEFAULT_MAX_RESULTS)
|
|
next_token = self._get_param("nextToken", DEFAULT_NEXT_TOKEN)
|
|
|
|
job_runs, next_token = self.emrcontainers_backend.list_job_runs(
|
|
virtual_cluster_id=virtual_cluster_id,
|
|
created_before=created_before,
|
|
created_after=created_after,
|
|
name=name,
|
|
states=states,
|
|
max_results=max_results,
|
|
next_token=next_token,
|
|
)
|
|
|
|
response = {"jobRuns": job_runs, "nextToken": next_token}
|
|
return 200, {}, json.dumps(response)
|
|
|
|
def describe_job_run(self) -> TYPE_RESPONSE:
|
|
job_id = self._get_param("jobRunId")
|
|
virtual_cluster_id = self._get_param("virtualClusterId")
|
|
|
|
job_run = self.emrcontainers_backend.describe_job_run(
|
|
job_id=job_id, virtual_cluster_id=virtual_cluster_id
|
|
)
|
|
|
|
response = {"jobRun": job_run}
|
|
return 200, {}, json.dumps(response)
|