Как я могу получить токен доступа JWT от AWS Cognito в качестве администратора в Python с помощью boto3?

Для моего собственного проекта я также подумывал о похожей стратегии для тестирования Cognito-защищенных API.

Я думаю, что создание временного пользователя со случайным паролем для каждого запуска теста - это справедливый подход.

Я думаю, что для создания пользователя из командной строки существуют более простые вызовы API Cognito: зарегистрироваться и администратор-подтверждение-регистрация, предоставленный в cognito-idp CLI. Благодаря этому вы можете пропустить шаги для решения проблем, и пользователь готов к использованию.

Если вы хотите использовать boto3, вот простая функция для создания нового пользователя:

def create_user(username: str, password: str, 
                user_pool_id: str, app_client_id: str) -> None:
    client = boto3.client('cognito-idp')

    # initial sign up
    resp = client.sign_up(
        ClientId=app_client_id,
        Username=username,
        Password=password,
        UserAttributes=[
            {
                'Name': 'email',
                'Value': 'test@test.com'
            },
        ]
    )

    # then confirm signup
    resp = client.admin_confirm_sign_up(
        UserPoolId=user_pool_id,
        Username=username
    )

    print("User successfully created.")

Затем, чтобы получить JWT,

def authenticate_and_get_token(username: str, password: str, 
                               user_pool_id: str, app_client_id: str) -> None:
    client = boto3.client('cognito-idp')

    resp = client.admin_initiate_auth(
        UserPoolId=user_pool_id,
        ClientId=app_client_id,
        AuthFlow='ADMIN_NO_SRP_AUTH',
        AuthParameters={
            "USERNAME": username,
            "PASSWORD": password
        }
    )

    print("Log in success")
    print("Access token:", resp['AuthenticationResult']['AccessToken'])
    print("ID token:", resp['AuthenticationResult']['IdToken'])