runner.py

import boto3
import json
import logging
import os
import sys
import time
sys.path.insert(0, './vendor')
from sts import establish_role
from boto3.dynamodb.conditions import Key, Attr

environment = os.environ['ENV']
dynamodb = boto3.resource('dynamodb')
client = boto3.client('ec2')
lambda_client = boto3.client('lambda')
function_name = os.environ['run']
logger = logging.getLogger()
logger.setLevel(logging.INFO)

accounts_table = dynamodb.Table(os.environ['DYNAMODB_TABLE_ACCOUNTS'])
accounts = accounts_table.scan()['Items']
regions = ([region['RegionName']
                for region in client.describe_regions()['Regions']])

def invoke_process(fuction_name, account_id, region):
    """Launch target fuction_name on account_id and region
    """
    invoke_payload = (
        json.JSONEncoder().encode(
            {
                "account": account_id,
                "region": region
            }
        )
    )
    lambda_client.invoke(
        FunctionName=fuction_name,
        InvocationType='Event',
        Payload=invoke_payload,
    )

def runner(event, context):
    """Launch child import processes on a per account, per region basis
    """

    logger.info('Running available_ips...')
    for region in regions:
        for acct in accounts:
            logger.info(
                """
                cidr-house-rules-{0}-available_ips on account {1} in region {2}
                """.format(environment, acct['id'], region)
            )
            invoke_process(function_name, acct['id'], region)