# -*- coding: utf-8 -*-
import base64
import datetime
from elasticsearch import Elasticsearch, RequestsHttpConnection
from paclair.ancestries.generic import GenericAncestry, Layer
from paclair.plugins.abstract_plugin import AbstractPlugin
from paclair.exceptions import ResourceNotFoundException
class EsPlugin(AbstractPlugin):
"""
Elasticsearch plugin
"""
__SOURCE_URL = "{}/{}/{}/{}/_source?_source_include=sysinfo_clair"
def __init__(self, clair, hosts, index, doc_type, suffix=None, timedelta=None):
"""
Constructor
:param clair: ClairRequest object
:param hosts: elasticsearch hosts ex:[{'host': '172.18.8.10', 'port': 9200}]
:param index: elasticsearch index
:param doc_type: elasticsearch doc_type
:param suffix: index suffix (ex: one index a day)
:param timedelta: timedelta from today for suffix
"""
super().__init__(clair, "Legacy")
self._es = Elasticsearch(hosts, connection_class=RequestsHttpConnection)
self.index = index
self.doc_type = doc_type
if suffix is not None:
timedelta = timedelta or {}
self.index += (datetime.datetime.today() + datetime.timedelta(**timedelta)).strftime(suffix)
def push(self, name):
ancestry = self.create_ancestry(name)
self.logger.debug("Remove ancestry {} from Clair's Database.".format(name))
try:
self.clair.delete_ancestry(ancestry)
except ResourceNotFoundException:
self.logger.debug("Ancestry {} not yet in Clair's Database.".format(name))
return self.clair.post_ancestry(ancestry)
def create_ancestry(self, name):
# get ID
search = {"size": 1, "sort": {'@timestamp': "desc"}, "_source": False,
"query": {'match_phrase': {'hostname': name}}}
result = self._es.search(index=self.index, doc_type=self.doc_type, body=search)['hits']
# no result
if result['total'] == 0:
raise ResourceNotFoundException("{} not found".format(name))
id_name = result['hits'][0]['_id']
path = self.__SOURCE_URL.format(self._es.transport.get_connection().host, self.index, self.doc_type, id_name)
# Authentication
auth = self._es.transport.get_connection().session.auth
headers = None
if auth is not None:
digest = "{}:{}".format(*auth)
digest = base64.b64encode(digest.encode("utf-8"))
headers = {'Authorization': 'Basic {}'.format(digest.decode('utf-8'))}
return GenericAncestry(name, self.clair_format, [Layer(name, name, path, headers)])
