Daniel (dB.) Doubrovkine
2d6ae74211
* Updated dependencies. Signed-off-by: dblock <dblock@amazon.com> * Lock setuptools at 71.1.0. Newer version seems to cause UserWarning: Unknown distribution option: 'test_suite'. Signed-off-by: dblock <dblock@amazon.com> * Updated opensearch-py APIs to reflect opensearch-api-specification@9d3bc34. Signed-off-by: dblock <dblock@amazon.com> * Do not test against flaky 2.0. Signed-off-by: dblock <dblock@amazon.com> * Reduce test matrix and add 2.16. Signed-off-by: dblock <dblock@amazon.com> * Fix missing spaces in generated descriptions. Signed-off-by: dblock <dblock@amazon.com> * Removed 2.2.1. Signed-off-by: dblock <dblock@amazon.com> --------- Signed-off-by: dblock <dblock@amazon.com>
107 lines
2.7 KiB
Python
Executable File
107 lines
2.7 KiB
Python
Executable File
#!/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 asyncio
|
|
import os
|
|
|
|
from opensearchpy import AsyncOpenSearch
|
|
|
|
|
|
async def main() -> None:
|
|
"""
|
|
an example showing how to create an asynchronous connection
|
|
to OpenSearch, create an index, index a document and
|
|
search to return the document
|
|
"""
|
|
# connect to OpenSearch
|
|
host = "localhost"
|
|
port = 9200
|
|
auth = (
|
|
"admin",
|
|
os.getenv("OPENSEARCH_PASSWORD", "admin"),
|
|
) # For testing only. Don't store credentials in code.
|
|
|
|
client = AsyncOpenSearch(
|
|
hosts=[{"host": host, "port": port}],
|
|
http_auth=auth,
|
|
use_ssl=True,
|
|
verify_certs=False,
|
|
ssl_show_warn=False,
|
|
)
|
|
|
|
try:
|
|
info = await client.info()
|
|
print(
|
|
f"Welcome to {info['version']['distribution']} {info['version']['number']}!"
|
|
)
|
|
|
|
# create an index
|
|
|
|
index_name = "test-index"
|
|
|
|
index_body = {"settings": {"index": {"number_of_shards": 4}}}
|
|
|
|
if not await client.indices.exists(index=index_name):
|
|
await client.indices.create(index_name, body=index_body)
|
|
|
|
# add some documents to the index, asynchronously
|
|
await asyncio.gather(
|
|
*[
|
|
client.index(
|
|
index=index_name,
|
|
body={
|
|
"title": f"Moneyball {i}",
|
|
"director": "Bennett Miller",
|
|
"year": "2011",
|
|
},
|
|
id=i,
|
|
)
|
|
for i in range(10)
|
|
]
|
|
)
|
|
|
|
# refresh the index
|
|
await client.indices.refresh(index=index_name)
|
|
|
|
# search for a document
|
|
user_query = "miller"
|
|
|
|
query = {
|
|
"size": 5,
|
|
"query": {
|
|
"multi_match": {"query": user_query, "fields": ["title^2", "director"]}
|
|
},
|
|
}
|
|
|
|
results = await client.search(body=query, index=index_name)
|
|
|
|
for hit in results["hits"]["hits"]:
|
|
print(hit)
|
|
|
|
# delete the documents
|
|
await asyncio.gather(
|
|
*[client.delete(index=index_name, id=i) for i in range(10)]
|
|
)
|
|
|
|
# delete the index
|
|
await client.indices.delete(index=index_name)
|
|
|
|
finally:
|
|
await client.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
loop = asyncio.new_event_loop()
|
|
asyncio.set_event_loop(loop)
|
|
loop.run_until_complete(main())
|
|
loop.close()
|