869b3c45c1* Adds initial models and implements create_flow endpoint. * Updates README with a Releases section * Removes unused import * Adds __init__.py files to MediaLive and MediaConnect test directories. * Adds list_flows endpoint and tests. * Adds describe_flow endpoint and tests. * Adds delete_flow endpoint and tests. * Adds URLs * Adds start_flow and stop_flow endpoints and tests. * Adds tag_resource and list_tags_for_resource endpoints and tests. * Uses unquote for ARNs
83 lines
3 KiB
Python
83 lines
3 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import json
|
|
|
|
from moto.core.responses import BaseResponse
|
|
from .models import mediaconnect_backends
|
|
|
|
try:
|
|
from urllib import unquote
|
|
except ImportError:
|
|
from urllib.parse import unquote
|
|
|
|
|
|
class MediaConnectResponse(BaseResponse):
|
|
SERVICE_NAME = "mediaconnect"
|
|
|
|
@property
|
|
def mediaconnect_backend(self):
|
|
return mediaconnect_backends[self.region]
|
|
|
|
def create_flow(self):
|
|
availability_zone = self._get_param("availabilityZone")
|
|
entitlements = self._get_param("entitlements")
|
|
name = self._get_param("name")
|
|
outputs = self._get_param("outputs")
|
|
source = self._get_param("source")
|
|
source_failover_config = self._get_param("sourceFailoverConfig")
|
|
sources = self._get_param("sources")
|
|
vpc_interfaces = self._get_param("vpcInterfaces")
|
|
flow = self.mediaconnect_backend.create_flow(
|
|
availability_zone=availability_zone,
|
|
entitlements=entitlements,
|
|
name=name,
|
|
outputs=outputs,
|
|
source=source,
|
|
source_failover_config=source_failover_config,
|
|
sources=sources,
|
|
vpc_interfaces=vpc_interfaces,
|
|
)
|
|
return json.dumps(dict(flow=flow.to_dict()))
|
|
|
|
def list_flows(self):
|
|
max_results = self._get_int_param("maxResults")
|
|
next_token = self._get_param("nextToken")
|
|
flows, next_token = self.mediaconnect_backend.list_flows(
|
|
max_results=max_results, next_token=next_token,
|
|
)
|
|
return json.dumps(dict(flows=flows, nextToken=next_token))
|
|
|
|
def describe_flow(self):
|
|
flow_arn = unquote(self._get_param("flowArn"))
|
|
flow, messages = self.mediaconnect_backend.describe_flow(flow_arn=flow_arn,)
|
|
return json.dumps(dict(flow=flow, messages=messages))
|
|
|
|
def delete_flow(self):
|
|
flow_arn = unquote(self._get_param("flowArn"))
|
|
flow_arn, status = self.mediaconnect_backend.delete_flow(flow_arn=flow_arn,)
|
|
return json.dumps(dict(flowArn=flow_arn, status=status))
|
|
|
|
def start_flow(self):
|
|
flow_arn = unquote(self._get_param("flowArn"))
|
|
flow_arn, status = self.mediaconnect_backend.start_flow(flow_arn=flow_arn,)
|
|
return json.dumps(dict(flowArn=flow_arn, status=status))
|
|
|
|
def stop_flow(self):
|
|
flow_arn = unquote(self._get_param("flowArn"))
|
|
flow_arn, status = self.mediaconnect_backend.stop_flow(flow_arn=flow_arn,)
|
|
return json.dumps(dict(flowArn=flow_arn, status=status))
|
|
|
|
def tag_resource(self):
|
|
resource_arn = unquote(self._get_param("resourceArn"))
|
|
tags = self._get_param("tags")
|
|
self.mediaconnect_backend.tag_resource(
|
|
resource_arn=resource_arn, tags=tags,
|
|
)
|
|
return json.dumps(dict())
|
|
|
|
def list_tags_for_resource(self):
|
|
resource_arn = unquote(self._get_param("resourceArn"))
|
|
tags = self.mediaconnect_backend.list_tags_for_resource(
|
|
resource_arn=resource_arn,
|
|
)
|
|
return json.dumps(dict(tags=tags))
|