Crear una SSH conexión mediante el SSHOperator

Para crear una nueva SSH conexión mediante la interfaz de usuario de Apache Airflow

1. Abre la página Entornos en la MWAA consola de Amazon.

2. En la lista de entornos, elija Abrir la interfaz de usuario de Airflow para su entorno.

3. En la página de interfaz de usuario de Apache Airflow, seleccione Administrador en la barra de navegación superior para ampliar la lista desplegable y, a continuación, seleccione Conexiones.

4. En la página Listar conexiones, seleccione + o el botón Añadir un nuevo registro para añadir una nueva conexión.

5. En la página Conectar a AD, proporcione la siguiente información:

   1. En Nombre de conexión introduzca ssh_new.

   2. Para el tipo de conexión, elige una opción SSHde la lista desplegable.

      nota

      Si el tipo de SSHconexión no está disponible en la lista, Amazon no MWAA ha instalado el apache-airflow-providers-ssh paquete necesario. Actualice el archivo requirements.txt para incluir este paquete e inténtelo de nuevo.

   3. En Host, introduce la dirección IP de la EC2 instancia de Amazon a la que quieres conectarte. Por ejemplo, 12.345.67.89.

   4. En Nombre de usuario, introduce ec2-user si te estás conectando a una EC2 instancia de Amazon. Su nombre de usuario puede ser diferente, según el tipo de instancia remota a la que desee que se conecte Apache Airflow.

   5. Para Extra, introduce el siguiente par clave-valor en JSON formato:

      { "key_file": "/usr/local/airflow/dags/your-secret-key.pem" }

      Este par clave-valor indica a Apache Airflow que busque la clave secreta en el directorio local /dags.

1. Abre una terminal y navega hasta el directorio donde está almacenado el DAG código. Por ejemplo:

   cd dags

2. Copie el contenido del código de ejemplo siguiente y guárdelo localmente como ssh.py.

   from airflow.decorators import dag
   from datetime import datetime
   from airflow.providers.ssh.operators.ssh import SSHOperator
   
   @dag(
       dag_id="ssh_operator_example",
       schedule_interval=None,     
       start_date=datetime(2022, 1, 1),
       catchup=False,
       )
   def ssh_dag():
       task_1=SSHOperator(
           task_id="ssh_task",
           ssh_conn_id='ssh_new',
           command='hostname',
       )
   
   my_ssh_dag = ssh_dag()

3. Ejecute el siguiente AWS CLI comando para copiarlo en el bucket de su entorno y, DAG a continuación, actívelo DAG mediante la interfaz de usuario de Apache Airflow.

   $ aws s3 cp your-dag.py s3://your-environment-bucket/dags/

4. Si se ejecuta correctamente, verás un resultado similar al siguiente en los registros de ssh_task tareas de ssh_operator_exampleDAG:

   [2022-01-01, 12:00:00 UTC] {{base.py:79}} INFO - Using connection to: id: ssh_new. Host: 12.345.67.89, Port: None,
   Schema: , Login: ec2-user, Password: None, extra: {'key_file': '/usr/local/airflow/dags/your-secret-key.pem'}
   [2022-01-01, 12:00:00 UTC] {{ssh.py:264}} WARNING - Remote Identification Change is not verified. This won't protect against Man-In-The-Middle attacks
   [2022-01-01, 12:00:00 UTC] {{ssh.py:270}} WARNING - No Host Key Verification. This won't protect against Man-In-The-Middle attacks
   [2022-01-01, 12:00:00 UTC] {{transport.py:1819}} INFO - Connected (version 2.0, client OpenSSH_7.4)
   [2022-01-01, 12:00:00 UTC] {{transport.py:1819}} INFO - Authentication (publickey) successful!
   [2022-01-01, 12:00:00 UTC] {{ssh.py:139}} INFO - Running command: hostname
   [2022-01-01, 12:00:00 UTC]{{ssh.py:171}} INFO - ip-123-45-67-89.us-west-2.compute.internal
   [2022-01-01, 12:00:00 UTC] {{taskinstance.py:1280}} INFO - Marking task as SUCCESS. dag_id=ssh_operator_example, task_id=ssh_task, execution_date=20220712T200914, start_date=20220712T200915, end_date=20220712T200916