elastic search client using aws credentials

aws_elastic_search_client.py

import boto3
from requests_aws4auth import AWS4Auth
from elasticsearch import Elasticsearch, RequestsHttpConnection

class AwsElasticSearchClient(Elasticsearch):
    def __init__(self, **args):
        session = boto3.Session()
        creds = session.get_credentials().get_frozen_credentials()
        awsauth = AWS4Auth(
            creds.access_key, creds.secret_key, args["region"], "es", session_token=creds.token
        )

        args = {
                "hosts": args["hosts"],
                "http_auth": awsauth,
                "use_ssl": True,
                "verify_certs": True,
                "connection_class": RequestsHttpConnection
                }
        super().__init__(**args)