def image_type()

in eksupdate/src/eks_get_image_type.py [0:0]

• 22 lines of code
• 8 McCabe index (conditional complexity)

def image_type(node_type,Presentversion,inst,regionName):
    ''' returning image location '''
    ec2_client = boto3.client('ec2',region_name=regionName)
    if node_type=="Amazon Linux 2":
        filters = [{'Name':'owner-id','Values':['602401143452']},{'Name': 'name', 'Values': ["amazon-eks-node-*"]},{'Name':'is-public','Values':['true']}]
    elif "ubuntu" in node_type.lower():
        filters = [{'Name':'owner-id','Values':['099720109477']},{'Name': 'name', 'Values': ["ubuntu-eks/k8s_*"]},{'Name':'is-public','Values':['true']}]
    elif "bottlerocket" in node_type.lower():
        filters = [{'Name':'owner-id','Values':['092701018921']},{'Name': 'name', 'Values': ["bottlerocket-aws-k8s-*"]},{'Name':'is-public','Values':['true']}]
    elif  "Windows" in node_type:
         filters = [{'Name':'owner-id','Values':['801119661308']},{'Name': 'name', 'Values': ["Windows_Server-*-English-*-EKS_Optimized-*"]},{'Name':'is-public','Values':['true']}]
    else:
        return True
    ''' decribing image types'''
    images = ec2_client.describe_images(Filters=filters)
    instances_list=[]
    for i in images.get('Images'):
        instances_list.append([i.get('ImageId'),i.get('Name')])
    for i in instances_list:
        if inst in i[0]:
            return i[1]
    return inst in instances_list