from db.es_connector import ElasticsearchConnector


__all__ = ["ESMixin"]


class ESMixin:
    def __init__(self, **kwargs):
        super().__init__(**kwargs)
        self.es_client: ElasticsearchConnector = ElasticsearchConnector()

    def _pre_run(self):
        super()._pre_run()
        self.es_client.init()

    def _post_run(self):
        super()._post_run()
        self.es_client.close()