def _cleanup_old_versions(name):
logger.info('cleaning up old versions of {0}. Keeping {1}'.format(
name, REVISIONS))
versions = _versions(name)
for version in versions[0:(len(versions) - REVISIONS)]:
logger.debug('deleting {} version {}'.format(name, version))
aws_lambda('delete_function', FunctionName=name, Qualifier=version)
def role():
new_role = False
try:
logger.info('finding role')
iam('get_role', RoleName='gimel')
except ClientError:
logger.info('role not found. creating')
iam('create_role',
RoleName='gimel',
AssumeRolePolicyDocument=ASSUMED_ROLE_POLICY)
new_role = True
role_arn = iam('get_role', RoleName='gimel', query='Role.Arn')
logger.debug('role_arn={}'.format(role_arn))
logger.info('updating role policy')
iam('put_role_policy',
RoleName='gimel',
PolicyName='gimel',
PolicyDocument=POLICY)
if new_role:
from time import sleep
logger.info('waiting for role policy propagation')
sleep(5)
return role_arn
def _load_config(config_filename):
try:
with open(config_filename) as config_file:
logger.info('Using config {}'.format(config_filename))
return config_file.name, json.load(config_file)
except IOError:
logger.debug('trying to load {} (not found)'.format(config_filename))
return config_filename, {}
def get_create_api():
api_id = apigateway('get_rest_apis',
query='items[?name==`gimel`] | [0].id')
if not api_id:
api_id = apigateway('create_rest_api',
name='gimel',
description='Gimel API',
query='id')
logger.debug("api_id={}".format(api_id))
return api_id
def create_update_lambda(role_arn, wiring):
name, handler, memory, timeout = (wiring[k]
for k in ('FunctionName', 'Handler',
'MemorySize', 'Timeout'))
try:
logger.info('finding lambda function')
function_arn = aws_lambda('get_function',
FunctionName=name,
query='Configuration.FunctionArn')
except ClientError:
function_arn = None
if not function_arn:
logger.info('creating new lambda function {}'.format(name))
with open('gimel.zip', 'rb') as zf:
function_arn, version = aws_lambda('create_function',
FunctionName=name,
Runtime='python3.8',
Role=role_arn,
Handler=handler,
MemorySize=memory,
Timeout=timeout,
Publish=True,
Code={'ZipFile': zf.read()},
query='[FunctionArn, Version]')
else:
logger.info('updating lambda function {}'.format(name))
aws_lambda('update_function_configuration',
FunctionName=name,
Runtime='python3.8',
Role=role_arn,
Handler=handler,
MemorySize=memory,
Timeout=timeout)
with open('gimel.zip', 'rb') as zf:
function_arn, version = aws_lambda('update_function_code',
FunctionName=name,
Publish=True,
ZipFile=zf.read(),
query='[FunctionArn, Version]')
function_arn = _function_alias(name, version)
_cleanup_old_versions(name)
logger.debug('function_arn={} ; version={}'.format(function_arn, version))
return function_arn
def function_uri(function_arn, region):
uri = ('arn:aws:apigateway:{0}:lambda:path/2015-03-31/functions'
'/{1}/invocations').format(region, function_arn)
logger.debug("uri={0}".format(uri))
return uri