Передайте переменную во входной параметр пошаговой функции start_execution () - PullRequest
Новая
       34

Передайте переменную во входной параметр пошаговой функции start_execution ()

0 голосов
/ 27 марта 2019

Я использую boto3 с Python 3.6 для запуска выполнения пошаговой функции. Функция Step предназначена для обмена моими базовыми значениями AMI во всех моих учетных записях. У меня есть 4 переменные, которые мне нужно передать параметру input, чтобы начать выполнение. Это идентификатор AMI, список учетных записей, которыми я владею, исходная учетная запись и ключ KMS. Идентификатор AMI и список учетных записей построены в моем коде и являются переменными, которые необходимо передавать динамически. Согласно документации, input является string that contains the JSON input data for the execution и дает следующий пример: "input": "{\"ami_id\" : \"ami_id\"}". У меня вопрос, как передать переменные в этот параметр в качестве значения? Мой код ниже с трассировкой:

CODE:

import boto3
import json

#   Get an STS token to assume roles into AWS accounts
def get_sts_token(**kwargs):
    role_arn = kwargs['RoleArn']
    region_name = kwargs['RegionName']
    sts = boto3.client(
        'sts',
    region_name=region_name,
    )
    token = sts.assume_role(
        RoleArn=role_arn,
        RoleSessionName='GetInstances',
        DurationSeconds=900,
    )
    return token["Credentials"]

def get_accounts():
    role_arn = "arn:aws:iam::xxxxxxxxxx:role/list-accounts-role"
    region_name = "us-east-1"

    token = get_sts_token(RoleArn=role_arn, RegionName=region_name)

    access_key = token['AccessKeyId']
    secret_key = token['SecretAccessKey']       
    session_token = token['SessionToken']

    client = boto3.client('organizations', 
    aws_access_key_id=access_key,
    aws_secret_access_key=secret_key,
    aws_session_token=session_token)

    moreAccounts=True
    nexttoken=''
    global accountList
    accountList =[]

    while moreAccounts:
        if (len(nexttoken)>0):
            accounts=client.list_accounts(NextToken=nexttoken)
        else:
            accounts=client.list_accounts()
        if 'NextToken' in accounts:
            nexttoken=accounts['NextToken']
        else:
            moreAccounts=False
        for account in accounts['Accounts']:
            if account['Status'] != 'SUSPENDED' and account['Status'] != 'CLOSED' :
                account_id = account['Id']
                accountList.append(account_id)


    print(accountList)

def trigger_sfn():
    ssm = boto3.client('ssm')
    role_arn = "arn:aws:iam::xxxxxxxx:role/execute-sfn"
    region_name = "us-east-1"

    ami_id = ssm.get_parameter(Name='/BaseAMI/newest')['Parameter']['Value']
    print(ami_id)

    token = get_sts_token(RoleArn=role_arn, RegionName=region_name)
    print(token)

    access_key = token['AccessKeyId']
    secret_key = token['SecretAccessKey']       
    session_token = token['SessionToken']

    sfn = boto3.client('stepfunctions', 
    aws_access_key_id=access_key,
    aws_secret_access_key=secret_key,
    aws_session_token=session_token)

    response = sfn.start_execution(
    stateMachineArn='arn:aws:states:us-east-1:xxxxxxxx:stateMachine:ami-share',
    input="{\"ami_id\": ami_id, \"source_account_id\": \"112233445566\", \"accountList\": accountList, \"kms_key_arn\": \"alias/aws/ebs\"}"
)

    print(response)

TRACE:

An error occurred (InvalidExecutionInput) when calling the 
StartExecution operation: Invalid State Machine Execution Input: 
'com.fasterxml.jackson.core.JsonParseException: Unrecognized token 
'ami_id': was expecting ('true', 'false' or 'null')
at [Source: (String)"{"ami_id": ami_id, "source_account_id": 
"112233445566", "accountList": accountList, "kms_key_arn": 
"alias/aws/ebs"}"; line: 1, column: 18]': InvalidExecutionInput
Traceback (most recent call last):
File "/var/task/lambda_function.py", line 6, in lambda_handler
trigger_sfn()
File "/var/task/lambda_function.py", line 96, in trigger_sfn
input="{\"ami_id\": ami_id, \"source_account_id\": \"112233445566\", 
\"accountList\": accountList, \"kms_key_arn\": \"alias/aws/ebs\"}"
File "/var/runtime/botocore/client.py", line 314, in _api_call
return self._make_api_call(operation_name, kwargs)
File "/var/runtime/botocore/client.py", line 612, in _make_api_call
raise error_class(parsed_response, operation_name)
botocore.errorfactory.InvalidExecutionInput: An error occurred 
(InvalidExecutionInput) when calling the StartExecution operation: 
Invalid State Machine Execution Input: 
'com.fasterxml.jackson.core.JsonParseException: Unrecognized token 
'ami_id': was expecting ('true', 'false' or 'null')
at [Source: (String)"{"ami_id": ami_id, "source_account_id": 
"112233445566", "accountList": accountList, "kms_key_arn": 
"alias/aws/ebs"}"; line: 1, column: 18]'

1 Ответ

1 голос
/ 27 марта 2019

Вы можете добавить значение переменной в строку следующим образом: 

input="{\"ami_id\": \"" + ami_id + "\", \"source_account_id\": \"112233445566\", \"accountList\": accountList, \"kms_key_arn\": \"alias/aws/ebs\"}"

Ваш код в настоящее время отправляет буквенную строку «ami_id» вместо значения переменной ami_id

...