build/lib/tools/index-opensearch.py

65 lines
1.6 KiB
Python
Raw Normal View History

#!/usr/bin/env python3
#
# SPDX-License-Identifier: GPL-2.0
#
# Copyright (c) 2013-2023 Igor Pecovnik, igor@armbian.com
#
# This file is a part of the Armbian Build Framework
# https://github.com/armbian/build/
#
import json
import sys
from opensearchpy import OpenSearch # pip3 install opensearch-py
def eprint(*args, **kwargs):
print(*args, file=sys.stderr, **kwargs)
# Read JSON from stdin
# - should be an array of objects
# - loop over array and index each obj into OS in to the passed index
# read_from_stdin = sys.stdin.read()
json_object = json.load(sys.stdin)
eprint("Loaded {} objects from stdin...".format(len(json_object)))
host = '127.0.0.1'
port = 9200
# Create the OpenSearch client.
client = OpenSearch(hosts=[{'host': host, 'port': port}], http_compress=False, use_ssl=False)
# Create an index with non-default settings.
index_name = 'board-vars-build'
index_body = {'settings': {'index': {'number_of_shards': 1, 'number_of_replicas': 0}}}
# Delete the index; remove old data.
try:
delete_response = client.indices.delete(index=index_name)
eprint('\nDeleting index...')
# print(delete_response)
except:
eprint("Failed to delete index {}".format(index_name))
eprint('\nCreating index...')
response_create = client.indices.create(index_name, body=index_body)
# print(response_create)
counter = 0
for obj in json_object:
# Skip desktop builds
if 'BUILD_DESKTOP' in obj['in']['vars']:
if obj['in']['vars']['BUILD_DESKTOP'] == 'yes':
continue
response = client.index(index=index_name, body=obj)
counter += 1
eprint(f"\nRefreshing index after loading {counter}...")
client.indices.refresh(index=index_name)
eprint("\nDone.")