Add Elasticsearch support

core/lib/elastic.py

@@ -0,0 +1,32 @@
+from datetime import datetime
+
+from django.conf import settings
+from elasticsearch import Elasticsearch
+
+from core.util import logs
+
+log = logs.get_logger(__name__)
+
+client = None
+
+
+def initialise_elasticsearch():
+    """
+    Initialise the Elasticsearch client.
+    """
+    auth = (settings.ELASTICSEARCH_USERNAME, settings.ELASTICSEARCH_PASSWORD)
+    client = Elasticsearch(
+        settings.ELASTICSEARCH_HOST, http_auth=auth, verify_certs=False
+    )
+    return client
+
+
+def store_msg(index, msg):
+    global client
+    if not client:
+        client = initialise_elasticsearch()
+    if "ts" not in msg:
+        msg["ts"] = datetime.utcnow().isoformat()
+    result = client.index(index=index, body=msg)
+    if not result["result"] == "created":
+        log.error(f"Indexing of '{msg}' failed: {result}")