| 12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849 |
- import ssl
- from elasticsearch import Elasticsearch, ApiError
- from elasticsearch import helpers
- from config.es_mappings import index_name
- class ElasticSearchClient:
- def __init__(self, index_=index_name):
- self.password = "nkvvASQuQ0XUGRq5OLvm"
- self.hosts = ["https://192.168.205.85:9200", "https://192.168.205.85:9300"]
- self.ctx = ssl.create_default_context(cafile="config/es_certs.crt")
- self.es = Elasticsearch(
- self.hosts, basic_auth=("elastic", self.password), ssl_context=self.ctx
- )
- self.index_name = index_
- def create_index(self, settings, mappings):
- if self.es.indices.exists(index=self.index_name):
- self.es.indices.delete(index=self.index_name)
- try:
- self.es.indices.create(
- index=self.index_name, settings=settings, mappings=mappings
- )
- print("Index created successfully")
- except ApiError as e:
- print(f"❌ Failed: {e.meta.error['type']} – {e.meta.error['reason']}")
- def get_max_article_id(self):
- response = self.es.search(
- index=self.index_name,
- size=1,
- sort="article_id:desc",
- _source=["article_id"],
- )
- return response["hits"]["hits"][0]["_source"]["article_id"]
- def search(self, search_keys, size=10):
- query = {
- "query": {"match": {"title": search_keys}},
- "_source": ["article_id", "title"],
- "size": size
- }
- resp = self.es.search(index=index_name, body=query)
- return [i["_source"] for i in resp["hits"]["hits"]]
- def bulk_insert(self, docs):
- helpers.bulk(self.es, docs)
|
