Python Elasticsearch DSL Init AWS ES Connection

ESUtils.py

import boto3, os
from requests_aws4auth import AWS4Auth
from elasticsearch import RequestsHttpConnection
from elasticsearch_dsl import connections


def init_connection():
    service = "es"
    region = "us-west-2"
    credentials = boto3.Session().get_credentials()
    awsauth = AWS4Auth(
        credentials.access_key,
        credentials.secret_key,
        region,
        service,
        session_token=credentials.token,
    )
    connections.configure(
        default={
            "hosts": [{"host": "my-non-vpc-elasticsearch.us-west-2.es.amazonaws.com", "port": 443,}],
            "http_auth": awsauth,
            "use_ssl": True,
            "verify_certs": True,
            "connection_class": RequestsHttpConnection,
        }
    )
    connections.get_connection()  # Checks if conn already exists if not creates a new one

def init_local_es_connection():
    es = Elasticsearch("localhost:9200")
    connections.add_connection(alias="default", conn=es)