2021-08-04 12:40:10 +00:00
|
|
|
import boto3
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
DEFAULT_REGION = "eu-central-1"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _get_clients():
|
|
|
|
|
return (
|
|
|
|
|
boto3.client("ec2", region_name=DEFAULT_REGION),
|
|
|
|
|
boto3.client("iam", region_name=DEFAULT_REGION),
|
|
|
|
|
boto3.client("ecs", region_name=DEFAULT_REGION),
|
|
|
|
|
boto3.client("logs", region_name=DEFAULT_REGION),
|
|
|
|
|
boto3.client("batch", region_name=DEFAULT_REGION),
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _setup(ec2_client, iam_client):
|
|
|
|
|
"""
|
|
|
|
|
Do prerequisite setup
|
|
|
|
|
:return: VPC ID, Subnet ID, Security group ID, IAM Role ARN
|
|
|
|
|
:rtype: tuple
|
|
|
|
|
"""
|
|
|
|
|
resp = ec2_client.create_vpc(CidrBlock="172.30.0.0/24")
|
|
|
|
|
vpc_id = resp["Vpc"]["VpcId"]
|
|
|
|
|
resp = ec2_client.create_subnet(
|
|
|
|
|
AvailabilityZone="eu-central-1a", CidrBlock="172.30.0.0/25", VpcId=vpc_id
|
|
|
|
|
)
|
|
|
|
|
subnet_id = resp["Subnet"]["SubnetId"]
|
|
|
|
|
resp = ec2_client.create_security_group(
|
|
|
|
|
Description="test_sg_desc", GroupName="test_sg", VpcId=vpc_id
|
|
|
|
|
)
|
|
|
|
|
sg_id = resp["GroupId"]
|
|
|
|
|
|
|
|
|
|
resp = iam_client.create_role(
|
|
|
|
|
RoleName="TestRole", AssumeRolePolicyDocument="some_policy"
|
|
|
|
|
)
|
|
|
|
|
iam_arn = resp["Role"]["Arn"]
|
|
|
|
|
iam_client.create_instance_profile(InstanceProfileName="TestRole")
|
|
|
|
|
iam_client.add_role_to_instance_profile(
|
|
|
|
|
InstanceProfileName="TestRole", RoleName="TestRole"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
return vpc_id, subnet_id, sg_id, iam_arn
|
