def getPolicy()

in resource-selector-lambda/handler.py [0:0]

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

    def getPolicy(self):
        # build up boto3 client for iam
        policyArn=[]
        iam = boto3.client('iam', region_name=self.region)
        paginator = iam.get_paginator('list_policies')
        page_iterator = paginator.paginate(Scope='Local')

        for page in page_iterator:
            # Iterate through policies
            for policy in page['Policies']:
                # if policy name provided in configuration check if match with current policy
                if self.policyName and policy['PolicyName'].__contains__(self.policyName):
                    policyArn.append(policy['Arn'])
                # if no policy name provided return all policies
                elif not self.policyName:
                    policyArn.append(policy['Arn'])
        # turn list into a comma separated string and place it in our response
        self.setOutput(policyArn)