def waf_update_ip_set()

in lambda/prune_old_entries.py [0:0]

• 24 lines of code
• 3 McCabe index (conditional complexity)

def waf_update_ip_set(ip_set_name, ip_set_id, ip_set_scope, source_ips):
    logger.info('creating waf object')
    waf = boto3.client('wafv2')
    
    for attempt in range(API_CALL_NUM_RETRIES):
        logger.info('type of IPset: %s' % ip_set_id )
        try:
            response = waf.update_ip_set(
                Name = ip_set_name,
                Id = ip_set_id,
                Scope = ip_set_scope,
                LockToken = get_ip_set(ip_set_name, ip_set_id, ip_set_scope)['LockToken'],
                    Addresses=source_ips
                    )
            logger.info(response)
            logger.info("log -- waf_update_ip_set %s IPs %s - type %s successfully..." % (ip_set_id, source_ips, ip_set_scope))
        except Exception as e:
            logger.error(e)
            delay = math.pow(2, attempt)
            logger.info("log -- waf_update_ip_set retrying in %d seconds..." % (delay))
            time.sleep(delay)
        else:
            break
    else:
        logger.error("log -- waf_update_ip_set failed ALL attempts to call API")