Mark CohenandGitHub
7b0b58d12b
* updated files with docstrings to pass pylint Signed-off-by: Mark Cohen <[email protected]> * updated samples to prepare for enabling missing-docstring linter; will continue to work on this before committing setup.cfg Signed-off-by: Mark Cohen <[email protected]> * removed missing-function-docstring from setup.cfg so the linter doesn't fail while work on docstrings continues Signed-off-by: Mark Cohen <[email protected]> * corrected unnecessary return docstring values Signed-off-by: Mark Cohen <[email protected]> * fixing failure in 'black' on reformatting Signed-off-by: Mark Cohen <[email protected]> --------- Signed-off-by: Mark Cohen <[email protected]>
81 lines
2.4 KiB
Python
81 lines
2.4 KiB
Python
#!/usr/bin/env python
|
|
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
#
|
|
# The OpenSearch Contributors require contributions made to
|
|
# this file be licensed under the Apache-2.0 license or a
|
|
# compatible open source license.
|
|
#
|
|
# Modifications Copyright OpenSearch Contributors. See
|
|
# GitHub history for details.
|
|
|
|
import logging
|
|
from os import environ
|
|
from time import sleep
|
|
from urllib.parse import urlparse
|
|
|
|
from boto3 import Session
|
|
|
|
from opensearchpy import OpenSearch, RequestsAWSV4SignerAuth, RequestsHttpConnection
|
|
|
|
|
|
def main() -> None:
|
|
"""
|
|
connects to a cluster specified in environment variables, creates an index, inserts documents,
|
|
searches the index, deletes the document, deletes the index.
|
|
the environment variables are "ENDPOINT" for the cluster endpoint, AWS_REGION for the region in which the cluster
|
|
is hosted, and SERVICE to indicate if this is an ES 7.10.2 compatible cluster
|
|
"""
|
|
# verbose logging
|
|
logging.basicConfig(format="%(levelname)s:%(message)s", level=logging.INFO)
|
|
|
|
# cluster endpoint, for example: my-test-domain.us-east-1.es.amazonaws.com
|
|
url = urlparse(environ["ENDPOINT"])
|
|
region = environ.get("AWS_REGION", "us-east-1")
|
|
service = environ.get("SERVICE", "es")
|
|
|
|
credentials = Session().get_credentials()
|
|
|
|
auth = RequestsAWSV4SignerAuth(credentials, region, service)
|
|
|
|
client = OpenSearch(
|
|
hosts=[{"host": url.netloc, "port": url.port or 443}],
|
|
http_auth=auth,
|
|
use_ssl=True,
|
|
verify_certs=True,
|
|
connection_class=RequestsHttpConnection,
|
|
timeout=30,
|
|
)
|
|
|
|
# TODO: remove when OpenSearch Serverless adds support for /
|
|
if service == "es":
|
|
info = client.info()
|
|
print(f"{info['version']['distribution']}: {info['version']['number']}")
|
|
|
|
# create an index
|
|
index = "movies"
|
|
client.indices.create(index=index)
|
|
|
|
try:
|
|
# index data
|
|
document = {"director": "Bennett Miller", "title": "Moneyball", "year": 2011}
|
|
client.index(index=index, body=document, id="1")
|
|
|
|
# wait for the document to index
|
|
sleep(1)
|
|
|
|
# search for the document
|
|
results = client.search(body={"query": {"match": {"director": "miller"}}})
|
|
for hit in results["hits"]["hits"]:
|
|
print(hit["_source"])
|
|
|
|
# delete the document
|
|
client.delete(index=index, id="1")
|
|
finally:
|
|
# delete the index
|
|
client.indices.delete(index=index)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|