alainivars/utils2devops

import boto3

from utils2devops.aws import SecurityGroup, Gress

"""
Aws configuration iiles should be present:
 ~/.aws/credentials
 ~/.aws/config
"""


def _make_gress(ports):
    gress = Gress()
    gress.protocol = ports['IpProtocol']
    if 'PortRange' in ports:
        gress.from_port = ports['FromPort']
        gress.to_port = ports['ToPort']
    else:
        gress.from_port = 0
        gress.to_port = 0
    gress.cidr_block = ports['IpRanges'][0]['CidrIp'] if len(ports['IpRanges']) else ''
    return gress


def list_security_groups(name, state=None):
    """This function does something in aws security_groups. TODO DOC

    :param name: The name to use.
    :type name: str.
    :param state: Current state to be in.
    :type state: bool.
    :returns: int -- the return code.
    :raises: AttributeError, KeyError

    """
    session = boto3.Session(profile_name='terraform')
    client = session.client(service_name='ec2', region_name='us-east-2')
    elements = client.describe_security_groups()

    for element in elements['SecurityGroups']:
        x = SecurityGroup(element['GroupId'])
        x.name = element['GroupName']
        x.description = element['Description']
        x.vpc_id = element['VpcId']
        for ports in element['IpPermissions']:
            gress = _make_gress(ports)
            x.ingress.append(gress)
        for ports in element['IpPermissionsEgress']:
            gress = _make_gress(ports)
            x.egress.append(gress)
        print(x)