

Las traducciones son generadas a través de traducción automática. En caso de conflicto entre la traducción y la version original de inglés, prevalecerá la version en inglés.

# Creación de un complemento personalizado para Apache Airflow PythonVirtualEnvOperator
<a name="samples-virtualenv"></a>

En el siguiente ejemplo, se muestra cómo parchear `PythonVirtualenvOperator` de Apache Airflow con un complemento personalizado en Amazon Managed Workflows para Apache Airflow.

**Topics**
+ [Versión](#samples-virtualenv-version)
+ [Requisitos previos](#samples-virtualenv-prereqs)
+ [Permisos](#samples-virtualenv-permissions)
+ [Requisitos](#samples-virtualenv-dependencies)
+ [Código de muestra de complemento personalizado](#samples-virtualenv-plugins-code)
+ [Plugins.zip](#samples-virtualenv-pluginszip)
+ [Código de ejemplo](#samples-virtualenv-code)
+ [Opciones de configuración de Airflow](#samples-virtualenv-airflow-config)
+ [Siguientes pasos](#samples-virtualenv-next-up)

## Versión
<a name="samples-virtualenv-version"></a>

Puede usar el código de ejemplo que aparece en esta página con **Apache Airflow v2** en [Python 3.10](https://peps.python.org/pep-0619/) y **Apache Airflow v3** en [Python 3.11](https://peps.python.org/pep-0664/).

## Requisitos previos
<a name="samples-virtualenv-prereqs"></a>

Para usar el código de muestra de esta página, necesitará lo siguiente:
+ Un [entorno de Amazon MWAA](get-started.md).

## Permisos
<a name="samples-virtualenv-permissions"></a>

No se necesitan permisos adicionales para usar el código de ejemplo de esta página.

## Requisitos
<a name="samples-virtualenv-dependencies"></a>

Para usar el código de ejemplo de esta página, agregue las siguientes dependencias a su `requirements.txt`. Consulte [Instalación de dependencias de Python](working-dags-dependencies.md) para obtener más información.

```
virtualenv
```

## Código de muestra de complemento personalizado
<a name="samples-virtualenv-plugins-code"></a>

Apache Airflow ejecutará el contenido de los archivos de Python en la carpeta de complementos durante el arranque. Este complemento parcheará la versión integrada `PythonVirtualenvOperator` durante el proceso de arranque para que sea compatible con Amazon MWAA. En los siguientes pasos, se describe el código de muestra del complemento personalizado.

1. En el símbolo del sistema, vaya hasta el directorio `plugins` del paso anterior. Por ejemplo:

   ```
   cd plugins
   ```

1. Copie el contenido del código de ejemplo siguiente y guárdelo localmente como `virtual_python_plugin.py`.

   ```
   """
   Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
    
   Permission is hereby granted, free of charge, to any person obtaining a copy of
   this software and associated documentation files (the "Software"), to deal in
   the Software without restriction, including without limitation the rights to
   use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
   the Software, and to permit persons to whom the Software is furnished to do so.
    
   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
   FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
   COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
   IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
   CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
   """
   from airflow.plugins_manager import AirflowPlugin
   import airflow.utils.python_virtualenv 
   from typing import List
   
   def _generate_virtualenv_cmd(tmp_dir: str, python_bin: str, system_site_packages: bool) -> List[str]:
       cmd = ['python3','/usr/local/airflow/.local/lib/python3.7/site-packages/virtualenv', tmp_dir]
       if system_site_packages:
           cmd.append('--system-site-packages')
       if python_bin is not None:
           cmd.append(f'--python={python_bin}')
       return cmd
   
   airflow.utils.python_virtualenv._generate_virtualenv_cmd=_generate_virtualenv_cmd
   
   class VirtualPythonPlugin(AirflowPlugin):                
       name = 'virtual_python_plugin'
   ```

## Plugins.zip
<a name="samples-virtualenv-pluginszip"></a>

En los siguientes pasos, se muestra cómo crear el `plugins.zip`.

1. En el símbolo del sistema, vaya hasta el directorio que contiene `virtual_python_plugin.py` del paso anterior. Por ejemplo:

   ```
   cd plugins
   ```

1. Comprima el contenido de la carpeta `plugins`.

   ```
   zip plugins.zip virtual_python_plugin.py
   ```

## Código de ejemplo
<a name="samples-virtualenv-code"></a>

Los siguientes pasos describen cómo crear el código de DAG para el complemento personalizado.

1. En el símbolo del sistema, vaya hasta el directorio en el que esté almacenado el código DAG. Por ejemplo:

   ```
   cd dags
   ```

1. Copie el contenido del código de ejemplo siguiente y guárdelo localmente como `virtualenv_test.py`.

   ```
   """
   Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
    
   Permission is hereby granted, free of charge, to any person obtaining a copy of
   this software and associated documentation files (the "Software"), to deal in
   the Software without restriction, including without limitation the rights to
   use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
   the Software, and to permit persons to whom the Software is furnished to do so.
    
   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
   FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
   COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
   IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
   CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
   """
   from airflow import DAG
   from airflow.operators.python import PythonVirtualenvOperator
   from airflow.utils.dates import days_ago
   import os
   
   os.environ["PATH"] = os.getenv("PATH") + ":/usr/local/airflow/.local/bin"
   
   def virtualenv_fn():
       import boto3
       print("boto3 version ",boto3.__version__)
   
   with DAG(dag_id="virtualenv_test", schedule_interval=None, catchup=False, start_date=days_ago(1)) as dag:
       virtualenv_task = PythonVirtualenvOperator(
           task_id="virtualenv_task",
           python_callable=virtualenv_fn,
           requirements=["boto3>=1.17.43"],
           system_site_packages=False,
           dag=dag,
       )
   ```

## Opciones de configuración de Airflow
<a name="samples-virtualenv-airflow-config"></a>

Si utiliza Apache Airflow v2, agregue `core.lazy_load_plugins : False` como opción de configuración de Apache Airflow. Para obtener más información, consulte [Uso de las opciones de configuración para cargar complementos en la versión 2](configuring-env-variables.md#configuring-2.0-airflow-override).

## Siguientes pasos
<a name="samples-virtualenv-next-up"></a>
+ Aprenda a cargar el archivo `requirements.txt` de este ejemplo a su bucket de Amazon S3 en [Instalación de dependencias de Python](working-dags-dependencies.md).
+ Aprenda a cargar el código el DAG de este ejemplo en la carpeta `dags` de su bucket de Amazon S3 en [Cómo añadir o actualizar DAG](configuring-dag-folder.md).
+ Obtenga más información sobre cómo cargar el archivo `plugins.zip` de este ejemplo a su bucket de Amazon S3 en [Instalación de complementos personalizados](configuring-dag-import-plugins.md).