You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
964 B
Python
38 lines
964 B
Python
from datetime import datetime
|
|
|
|
from django.conf import settings
|
|
from elastic_transport import ConnectionError
|
|
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()
|
|
try:
|
|
result = client.index(index=index, body=msg)
|
|
except ConnectionError as e:
|
|
log.error(f"Error indexing '{msg}': {e}")
|
|
return
|
|
if not result["result"] == "created":
|
|
log.error(f"Indexing of '{msg}' failed: {result}")
|