Trade-offs in AWS OpenSearch

import getpass
import requests

session = requests.Session()
session.auth = (input("Test username: "), getpass.getpass("Test password: "))
base = "https://search-test.example.com"
index = "tradeoff-products-check"

def call(method, path, body=None):
    response = session.request(method, base + path, json=body, timeout=30)
    response.raise_for_status()
    return response.json()

existing = session.head(base + "/" + index, timeout=30)
if existing.status_code != 404:
    raise RuntimeError("Index exists or existence check failed; do not overwrite")
call("PUT", "/" + index, {
    "settings": {"number_of_shards": 1, "number_of_replicas": 0},
    "mappings": {"properties": {
        "sku": {"type": "keyword"},
        "title": {"type": "text", "analyzer": "standard"},
        "category": {"type": "keyword"},
        "available": {"type": "boolean"}
    }}
})
try:
    documents = [
        {"sku": "p-1", "title": "Red shoes", "category": "shoes", "available": True},
        {"sku": "p-2", "title": "RED hat", "category": "hats", "available": True},
        {"sku": "p-3", "title": "Red boots", "category": "shoes", "available": False},
        {"sku": "p-4", "title": "Blue shoes", "category": "shoes", "available": True},
    ]
    for document in documents:
        call("PUT", "/" + index + "/_doc/" + document["sku"], document)
    call("POST", "/" + index + "/_refresh")
    result = call("POST", "/" + index + "/_search", {
        "query": {"bool": {
            "must": [{"match": {"title": "red"}}],
            "filter": [{"term": {"available": True}}]
        }},
        "sort": [{"sku": "asc"}],
        "aggs": {"categories": {"terms": {"field": "category"}}}
    })
    if result.get("timed_out") or result["_shards"]["failed"]:
        raise RuntimeError("Incomplete search response")
    ids = [hit["_id"] for hit in result["hits"]["hits"]]
    counts = {b["key"]: b["doc_count"]
              for b in result["aggregations"]["categories"]["buckets"]}
    if ids != ["p-1", "p-2"] or counts != {"hats": 1, "shoes": 1}:
        raise RuntimeError({"unexpected_ids": ids, "unexpected_counts": counts})
    print({"ids": ids, "category_counts": counts, "fixture": "passed"})
finally:
    call("DELETE", "/" + index)
    session.close()