Como criar um token de login na web do Apache Airflow

Você pode usar os comandos nesta página para gerar um token de login na web e, em seguida, fazer chamadas de API do Amazon Managed Workflows for Apache Airflow diretamente no seu shell de comando. Por exemplo, você pode obter um token e, em seguida, implantar DAGs de forma programática usando as APIs do Amazon MWAA. A seção a seguir inclui as etapas para criar um token da CLI do Apache Airflow usando a AWS CLI, um script bash, uma solicitação POST API ou um script Python. O token retornado na resposta é válido por 60 segundos.

Sumário

• Pré-requisitos
  • Acesso
  • AWS CLI
• Usar a AWS CLI
• Como usar um script bash
• Usando uma solicitação da API POST
• Como usar um script Python
• Próximas etapas

Pré-requisitos

A seção a seguir descreve as etapas preliminares necessárias para usar os comandos e scripts desta página.

Acesso

• Acesso à conta AWS em AWS Identity and Access Management (IAM) à política de permissões do Amazon MWAA em Política de acesso ao Apache Airflow UI: AmazonMWA WebServerAccess.

• Acesso à conta AWS em AWS Identity and Access Management (IAM) à política de permissões do Amazon MWAA em Política completa de acesso à API e ao console: AmazonMWAA FullApiAccess.

AWS CLI

A AWS Command Line Interface (AWS CLI) é uma ferramenta de código aberto que permite interagir com os serviços da AWS usando comandos no shell da linha de comando. Para concluir as etapas nesta página, é necessário o seguinte:

• AWS CLI: instalar a versão 2.

• AWS CLI: configuração rápida com aws configure.

Usar a AWS CLI

O exemplo a seguir usa o comando create-web–login-token na AWS CLI para criar um token da CLI do Apache Airflow.

aws mwaa create-web-login-token --name YOUR_ENVIRONMENT_NAME

Como usar um script bash

O exemplo a seguir usa um script bash para chamar o comando create-cli-token na AWS CLI criar um token da CLI do Apache Airflow.

1. Copie o conteúdo da amostra de código a seguir e salve localmente como get-web-token.sh.

   #!/bin/bash
   HOST=YOUR_HOST_NAME
   YOUR_URL=https://$HOST/aws_mwaa/aws-console-sso?login=true#
   WEB_TOKEN=$(aws mwaa create-web-login-token --name YOUR_ENVIRONMENT_NAME --query WebToken --output text)
   echo $YOUR_URL$WEB_TOKEN

2. Substitua os espaços reservados em vermelho por YOUR_HOST_NAME e YOUR_ENVIRONMENT_NAME. Por exemplo, um nome de host para uma rede pública pode ter a seguinte aparência (sem o https://):

   123456a0-0101-2020-9e11-1b159eec9000.c2.us-east-1.airflow.amazonaws.com

3. (opcional) os usuários do macOS e do Linux podem precisar executar o comando a seguir para garantir que o script seja executável.

   chmod +x get-web-token.sh

4. Execute o script a seguir para obter um token de login na web.

   ./get-web-token.sh

5. Você deve ver o seguinte em seu prompt de comando:

   https://123456a0-0101-2020-9e11-1b159eec9000.c2.us-east-1.airflow.amazonaws.com/aws_mwaa/aws-console-sso?login=true#{your-web-login-token}

Usando uma solicitação da API POST

O exemplo a seguir usa uma solicitação da API POST para criar um token de login na web do Apache Airflow.

1. Copie a seguinte URL e cole no campo URL do seu cliente da API REST.

   https://YOUR_HOST_NAME/aws_mwaa/aws-console-sso?login=true#WebToken

2. Substitua os espaços reservados em vermelho por YOUR_HOST_NAME. Por exemplo, um nome de host para uma rede pública pode ter a seguinte aparência (sem o https://):

   123456a0-0101-2020-9e11-1b159eec9000.c2.us-east-1.airflow.amazonaws.com

3. Copie o seguinte JSON e cole no campo de corpo do seu cliente da API REST.

   {
     "name": "YOUR_ENVIRONMENT_NAME"
   }

4. Substitua os espaços reservados em vermelho por YOUR_ENVIRONMENT_NAME.

5. Adicione pares de chave-valor no campo de autorização. Por exemplo, se você estiver usando o Postman, escolha Assinatura da AWS e, em seguida, insira seu:

   • AWS_ACCESS_KEY_ID em AccessKey

   • AWS_SECRET_ACCESS_KEY em SecretKey

6. Você deverá ver a seguinte resposta:

   {
     "webToken": "<Short-lived token generated for enabling access to the Apache Airflow Webserver UI>",
     "webServerHostname": "<Hostname for the WebServer of the environment>"
   }

Como usar um script Python

O exemplo a seguir usa o método boto3 create_web_login_token em um script Python para criar um token da CLI do Apache Airflow e acionar um DAG. Você pode executar esse script fora do Amazon MWAA. A única coisa que você precisa fazer é instalar a biblioteca boto3. Talvez você queira criar um ambiente virtual para instalar a biblioteca. Ele pressupõe que você tenha configurado as credenciais de autenticação da AWS para sua conta.

1. Copie o conteúdo da amostra de código a seguir e salve localmente como create-web-login-token.py.

   import boto3
   mwaa = boto3.client('mwaa')
   response = mwaa.create_web_login_token(
       Name="YOUR_ENVIRONMENT_NAME"
   )
   webServerHostName = response["WebServerHostname"]
   webToken = response["WebToken"]
   airflowUIUrl = 'https://{0}/aws_mwaa/aws-console-sso?login=true#{1}'.format(webServerHostName, webToken)
   print("Here is your Airflow UI URL: ")
   print(airflowUIUrl)

2. Substitua os espaços reservados em vermelho por YOUR_ENVIRONMENT_NAME.

3. Execute o script a seguir para obter um token de login na web.

   python3 create-web-login-token.py

Próximas etapas

• Explore a operação da API Amazon MWAA usada para criar um token da CLI em CreateCliToken.