Importe a biblioteca psycopg2 para interagir com seu banco AWS Lambda de dados PostgreSQL

Louis Hourcade, Amazon Web Services

Resumo

O Psycopg é um adaptador de banco de dados do PostgreSQL para Python. Os desenvolvedores empregram a biblioteca psycopg2 para gravar aplicações em Python que interagem com bancos de dados do PostgreSQL.

Na Amazon Web Services (AWS), os desenvolvedores também usam o AWS Lambda para executar código em aplicações ou em serviços de backend. O Lambda é um serviço de computação orientado a eventos e com tecnologia sem servidor que executa código sem a necessidade de provisionar ou gerenciar servidores.

Por padrão, quando você cria uma nova função que utiliza um runtime do Python com suporte por parte do Lambda, o ambiente de runtime do Lambda é criado com base em uma imagem de base destinada ao Lambda, que é fornecida pela AWS. Bibliotecas, como pandas ou psycopg2, não estão incluídas na imagem de base. Para usar uma biblioteca, é necessário empacotá-la em um pacote personalizado e anexá-la ao Lambda.

Existem várias formas de empacotar e anexar uma biblioteca, incluindo:

Esse padrão demonstra as duas primeiras opções.

Com um pacote de implantação em .zip, adicionar a biblioteca pandas à sua função do Lambda é relativamente simples. Crie uma pasta em sua máquina do Linux, adicione o script do Lambda juntamente com a biblioteca pandas e as respectivas dependências à pasta, realize a compactação da pasta em um arquivo .zip e forneça-o como origem para a função do Lambda.

Embora usar um pacote de implantação em .zip seja uma prática comum, essa abordagem não funciona para a biblioteca psycopg2. Primeiro, este padrão apresenta o erro que ocorre se você tentar adicionar a biblioteca psycopg2 à sua função do Lambda usando um pacote de implantação em .zip. Em seguida, o padrão demonstra como implantar o Lambda usando um Dockerfile e editar a imagem do Lambda para que a biblioteca psycopg2 funcione corretamente.

Para obter informações sobre os três recursos que o padrão implanta, consulte a seção Informações adicionais.

Pré-requisitos e limitações

Pré-requisitos

Limitações

Versões do produto

Arquitetura

Visão geral da solução

Para ilustrar os desafios que você pode enfrentar ao usar a biblioteca psycopg2 no Lambda, o padrão implanta duas funções do Lambda:

A primeira função do Lambda instala a biblioteca pandas e as respectivas dependências em um arquivo .zip, e o Lambda pode usar essa biblioteca.

A segunda função do Lambda demonstra que, ao criar uma imagem de contêiner para a função do Lambda, você pode executar as bibliotecas pandas e psycopg2 no Lambda.

Ferramentas

Serviços da AWS

Outras ferramentas

Repositório de código

O código desse padrão está disponível no repositório import-psycopg2 - - em. in-lambda-to-interact with-postgres-database GitHub

Práticas recomendadas

Esse padrão fornece um exemplo prático de uso AWS CDK para criar uma função Lambda a partir de um Dockerfile. Se você reutilizar esse código em sua aplicação, certifique-se de que os recursos implantados atendam a todos os requisitos de segurança. Empregue ferramentas como o Checkov, que verifica as configurações da infraestrutura de nuvem para identificar falhas de configuração antes da implantação.

Épicos

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

Mais informações

Nesse padrão, o AWS CDK modelo fornece uma AWS pilha com três recursos:

O script de ambas as funções do Lambda verifica se as bibliotecas pandas e psycopg2 foram importadas com êxito:

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"}

O pacote de implantação lambda_deploy.zip é desenvolvido com o script bash Constructs/lambda/build.sh. Esse script cria uma pasta, copia o script do Lambda, instala as bibliotecas pandas e psycopg2, e gera o arquivo .zip. Para gerar você mesmo o arquivo.zip, execute esse script bash e reimplante a pilha. AWS CDK

O Dockerfile começa com a imagem base AWS fornecida para o Lambda com um tempo de execução do Python. O Dockerfile instala as bibliotecas pandas e psycopg2 na imagem padrão.