Source code for troposphere_mate.associate.ec2

# -*- coding: utf-8 -*-

"""
https://docs.aws.amazon.com/AWSCloudFormation/latest/UserGuide/aws-properties-ec2-instance.html
"""

from troposphere_mate import Ref
from troposphere_mate import ec2, iam
from ..core.associate_linker import Linker, x_depends_on_y, LinkerApi as LinkerApi_


class LinkerApi(LinkerApi_):
    class Ec2InstanceWithSubnet(Linker):
        rtype1 = ec2.Instance
        rtype2 = ec2.Subnet

        def associate(self, ec2_inst, subnet, **kwargs):
            ec2_inst.SubnetId = Ref(subnet)
            x_depends_on_y(ec2_inst, subnet)

    class Ec2InstanceWithSecurityGroup(Linker):
        rtype1 = ec2.Instance
        rtype2 = ec2.SecurityGroup

        def associate(self, ec2_inst, sg, **kwargs):
            try:
                ec2_inst.SecurityGroupIds.append(Ref(sg))
            except AttributeError:
                ec2_inst.SecurityGroupIds = [Ref(sg)]
            x_depends_on_y(ec2_inst, sg)

    class Ec2InstanceWithIamInstanceProfile(Linker):
        rtype1 = ec2.Instance
        rtype2 = iam.InstanceProfile

        def associate(self, ec2_inst, iam_inst_profile, **kwargs):
            ec2_inst.IamInstanceProfile = Ref(iam_inst_profile)
            x_depends_on_y(ec2_inst, iam_inst_profile)

    class SecurityGroupWithVpc(Linker):
        rtype1 = ec2.SecurityGroup
        rtype2 = ec2.VPC

        def associate(self, sg, vpc, **kwargs):
            sg.VpcId = Ref(vpc)
            x_depends_on_y(sg, vpc)