Como usar o PyMongo para conectar o MongoDB Atlas ao AWS Lambda

1
 cd awslambda-demo

1
mkdir dependencies

1
pip install --target ./dependencies pymongo

1
import os
2
from pymongo import MongoClient
3
4
5
client = MongoClient(host=os.environ.get("ATLAS_URI"))
6
7
8
def lambda_handler(event, context):
9
    # Name of database
10
    db = client.test 
11
12
    # Name of collection
13
    collection = db.test 
14
    
15
    # Document to add inside
16
    document = {"first name": "Anaiya", "last name": "Raisinghani"}
17
18
19
    # Insert document
20
    result = collection.insert_one(document)
21
22
23
    if result.inserted_id:
24
        return "Document inserted successfully"
25
    else:
26
        return "Failed to insert document"

1
cd dependencies
2
zip -r ../deployment.zip *

1
cd ..
2
zip deployment.zip lambda_function.py

1
import os
2
from pymongo import MongoClient
3
4
connect = MongoClient(host=os.environ.get("ATLAS_URI"))
5
6
def lambda_handler(event, context):
7
    # Choose our "sample_restaurants" database and our "restaurants" collection
8
    database = connect.sample_restaurants
9
    collection = database.restaurants
10
11
    # This is our aggregation pipeline
12
    pipeline = [
13
14
        # We are finding American restaurants in Brooklyn
15
        {"$match": {"borough": "Brooklyn", "cuisine": "American"}},
16
17
        # We only want 5 out of our over 20k+ documents
18
        {"$limit": 5},
19
20
        # We don't want all the details, project what you need
21
        {"$project": {"_id": 0, "name": 1, "borough": 1, "cuisine": 1}}
22
        
23
    ]
24
25
    # This will show our pipeline 
26
    result = list(collection.aggregate(pipeline))
27
28
    # Print the result
29
    for restaurant in result:
30
        print(restaurant)