Importe la biblioteca psycopg2 para interactuar con su base de AWS Lambda datos PostgreSQL

Louis Hourcade, Amazon Web Services

Resumen

Psycopg es un adaptador de bases de datos de PostgresSQL para Python. Los desarrolladores utilizan la biblioteca psycopg2 para escribir aplicaciones de Python que interactúan con las bases de datos de PostgreSQL.

En Amazon Web Services (AWS), los desarrolladores también utilizan AWS Lambda para ejecutar código de aplicaciones o servicios de backend. Lambda es un servicio de computación sin servidor y basado en eventos que ejecuta código sin necesidad de aprovisionar ni administrar servidores.

De forma predeterminada, cuando se crea una función nueva que utiliza un tiempo de ejecución de Python compatible con Lambda, el entorno de tiempo de ejecución de Lambda se crea a partir de una imagen base de Lambda proporcionada por AWS. Bibliotecas como pandas o psycopg2 no se incluyen en la imagen base. Para usar una biblioteca, debe agruparla en un paquete personalizado y adjuntarla a Lambda.

Hay varias formas de agrupar y adjuntar una biblioteca, incluidas las siguientes:

Este patrón muestra las dos primeras opciones.

Con un paquete de implementación .zip, agregar la biblioteca pandas a la función de Lambda es relativamente sencillo. Cree una carpeta en su máquina Linux, agregue el script de Lambda junto con la biblioteca pandas y las dependencias de la biblioteca a la carpeta, comprima la carpeta y proporciónela como origen para su función de Lambda.

Aunque el uso de un paquete de implementación .zip es una práctica habitual, ese enfoque no funciona para la biblioteca psycopg2. Este patrón muestra primero el error que se produce si se utiliza un paquete de implementación .zip para agregar la biblioteca psycopg2 a la función de Lambda. A continuación, el patrón muestra cómo implementar Lambda desde un Dockerfile y editar la imagen de Lambda para que la biblioteca psycopg2 funcione.

Para obtener información sobre los tres recursos que implementa el patrón, consulte la sección Información adicional.

Requisitos previos y limitaciones

Requisitos previos

Limitaciones

Versiones de producto

Arquitectura

Información general de la solución

Para ilustrar los desafíos a los que puede enfrentarse al utilizar la biblioteca psycopg2 en Lambda, el patrón implementa dos funciones de Lambda:

La primera función de Lambda instala la biblioteca pandas y sus dependencias en un archivo .zip; y Lambda puede usar esa biblioteca.

La segunda función de Lambda demuestra que, al crear una imagen de contenedor para la función de Lambda, puede ejecutar las bibliotecas pandas y psycopg2 en Lambda.

Tools (Herramientas)

Servicios de AWS

Otras herramientas

Repositorio de código

El código de este patrón está disponible en el repositorio import-psycopg2 - -. in-lambda-to-interact with-postgres-database GitHub

Prácticas recomendadas

Este patrón le proporciona un ejemplo práctico de cómo crear una función Lambda AWS CDK a partir de un Dockerfile. Si reutiliza este código en su aplicación, asegúrese de que los recursos implementados cumplan con todos los requisitos de seguridad. Utilice herramientas como Checkov, que analiza las configuraciones de la infraestructura en la nube para detectar errores de configuración antes de implementar la infraestructura.

Epics

git clone https://github.com/aws-samples/import-psycopg2-in-lambda-to-interact-with-postgres-database.git
cd AWS-lambda-psycopg2

aws_acccount = "AWS_ACCOUNT_ID"
region = "AWS_REGION"
# Select the CPU architecture you are using to build the image (ARM or X86)
architecture = "ARM"

cdk bootstrap aws://<tooling-account-id>/<aws-region>

cdk deploy AWSLambdaPyscopg2

# Start from lambda Python3.13 image
FROM public.ecr.aws/lambda/python:3.13

# Copy the lambda code, together with its requirements
COPY lambda/requirements.txt ${LAMBDA_TASK_ROOT}
COPY lambda/lambda_code.py ${LAMBDA_TASK_ROOT}

# Install postgresql-devel in your image
RUN yum install -y gcc postgresql-devel

# install the requirements for the Lambda code
RUN pip3 install -r requirements.txt --target "${LAMBDA_TASK_ROOT}"

# Command can be overwritten by providing a different command in the template directly.
CMD ["lambda_code.handler"]

Recursos relacionados

Información adicional

En este patrón, la AWS CDK plantilla proporciona una AWS pila con tres recursos:

El script de ambas funciones de Lambda comprueba si las bibliotecas pandas y psycopg2 se han importado correctamente:

import pandas
print("pandas successfully imported")

import psycopg2
print("psycopg2 successfully imported")

def handler(event, context):
    """Function that checks whether psycopg2  and pandas are successfully imported or not"""
    return {"Status": "psycopg2 and pandas successfully imported"}

El paquete de implementación lambda_deploy.zip se crea con el script de bash Constructs/lambda/build.sh. Este script crea una carpeta, copia el script de Lambda, instala las bibliotecas pandas y psycopg2 y genera el archivo .zip. Para generar el archivo.zip usted mismo, ejecute este script bash y vuelva a implementar la pila. AWS CDK

El Dockerfile comienza con la imagen base AWS proporcionada para Lambda con un motor de ejecución de Python. El Dockerfile instala las bibliotecas pandas y psycopg2 sobre la imagen predeterminada.