pymongo_mongodb_aws_auth_connection_with_session_token.py

1import os
2from pymongo import MongoClient
3
4# The pymongo-auth-aws package must be installed: 
5# pip install "pymongo[aws]"
6
7# You can provide credentials via the connection string or environment variables.
8# If using environment variables, ensure AWS_ACCESS_KEY_ID and 
9# AWS_SECRET_ACCESS_KEY (and optionally AWS_SESSION_TOKEN) are set.
10
11access_key = os.environ.get("AWS_ACCESS_KEY_ID")
12secret_key = os.environ.get("AWS_SECRET_ACCESS_KEY")
13session_token = os.environ.get("AWS_SESSION_TOKEN")
14
15# Construct the connection string
16# For MONGODB-AWS, the authMechanism is set to 'MONGODB-AWS'
17uri = "mongodb://%s:%s@<hostname>/?authMechanism=MONGODB-AWS" % (
18    access_key, secret_key
19)
20
21# If using a session token, it must be passed in the authMechanismProperties
22if session_token:
23    uri += "&authMechanismProperties=AWS_SESSION_TOKEN:%s" % (session_token)
24
25client = MongoClient(uri)
26
27# Verify connection
28try:
29    client.admin.command('ping')
30    print("Connected successfully to MongoDB using AWS authentication")
31except Exception as e:
32    print(f"Connection failed: {e}")