#!/usr/bin/env python3
# This script retrieves AWS credentials from the environment or a specified profile.

import boto3
import os

def get_aws_credentials():
    session = boto3.Session()
    credentials = session.get_credentials().get_frozen_credentials()
    sts_client = session.client('sts')
    assumed_role = sts_client.get_caller_identity()
    return assumed_role

if __name__ == "__main__":
    profile = os.getenv('AWS_PROFILE')
    if profile:
        boto3.setup_default_session(profile_name=profile)
    creds = get_aws_credentials()
    print(creds)