AWS-Parameters-and-Secrets-Lambda-Extension-python

lambda.py

import logging
import os
import requests


# Logging
logger = logging.getLogger()
logger.setLevel(logging.INFO)

aws_session_token = os.environ.get('AWS_SESSION_TOKEN')


def lambda_handler(event, context):
    """Entrypoint for the lambda function"""

    # Goodbye Old Code
    # ssm = boto3.client('ssm', region_name="eu-west-1")
    # param = ssm.get_parameter(
    #     Name='/lambda/common/service-account-name', WithDecryption=True)

    headers = {
        'X-Aws-Parameters-Secrets-Token': aws_session_token
    }
    param_path = "%2Flambda%2Fcommon%2Fservice-account-name"  # /lambda/common/service-account-name
    param = requests.get(f'http://localhost:2773/systemsmanager/parameters/get?name={param_path}&withDecryption=true', headers=headers)
    logger.info("Received Param: %s", param.json()["Parameter"]["Value"])