opensearch-pyd/samples/bulk/bulk-array.py

#!/usr/bin/env python

# -*- coding: utf-8 -*-
# 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 os
from typing import Any

from opensearchpy import OpenSearch

# connect to an instance of OpenSearch

host = os.getenv("HOST", default="localhost")
port = int(os.getenv("PORT", 9200))
auth = (os.getenv("USERNAME", "admin"), os.getenv("PASSWORD", "admin"))

client = OpenSearch(
    hosts=[{"host": host, "port": port}],
    http_auth=auth,
    use_ssl=True,
    verify_certs=False,
    ssl_show_warn=False,
)

# check whether an index exists
index_name = "my-index"

if not client.indices.exists(index_name):
    client.indices.create(
        index_name,
        body={
            "mappings": {
                "properties": {
                    "value": {"type": "float"},
                }
            }
        },
    )

# index data
data: Any = []
for i in range(100):
    data.append({"index": {"_index": index_name, "_id": i}})
    data.append({"value": i})

rc = client.bulk(data)
if rc["errors"]:
    print("There were errors:")
    for item in rc["items"]:
        print(f"{item['index']['status']}: {item['index']['error']['type']}")
else:
    print(f"Bulk-inserted {len(rc['items'])} items.")

# delete index
client.indices.delete(index=index_name)