def vpc_cleanup()

in src/lambda_codebase/initial_commit/bootstrap_repository/adf-build/provisioner/src/vpc.py [0:0]

• 29 lines of code
• 12 McCabe index (conditional complexity)

def vpc_cleanup(account_id, vpcid, role, region):
    if not vpcid:
        return
    try:
        ec2 = role.resource('ec2', region_name=region)
        ec2client = ec2.meta.client
        vpc = ec2.Vpc(vpcid)
        # detach and delete all gateways associated with the vpc
        for gw in vpc.internet_gateways.all():
            vpc.detach_internet_gateway(InternetGatewayId=gw.id)
            gw.delete()
        # Route table associations
        for rt in vpc.route_tables.all():
            for rta in rt.associations:
                if not rta.main:
                    rta.delete()
        # Security Group
        for sg in vpc.security_groups.all():
            if sg.group_name != 'default':
                sg.delete()
        # Network interfaces
        for subnet in vpc.subnets.all():
            for interface in subnet.network_interfaces.all():
                interface.delete()
            subnet.delete()
        # Delete vpc
        ec2client.delete_vpc(VpcId=vpcid)
        LOGGER.info(f"VPC {vpcid} and associated resources has been deleted.")
    except exceptions.ClientError:
        LOGGER.warning(
            f"WARNING: cannot delete VPC {vpcid} in account {account_id}",
            exc_info=True,
        )
        raise