# -*- coding:utf-8 -*-


from dw_base.datax.datasources.data_source import DataSource

# Elasticsearch Data Source
DS_TYPE_ELASTICSEARCH = 'elasticsearch'
DS_ELASTICSEARCH_KEYS = [
    'nodes',
    'port',
    'user',
    'password'
]


class ElasticsearchDataSource(DataSource):

    def __init__(self, ds_file: str):
        super(ElasticsearchDataSource, self).__init__(ds_file)
        self.source_type = DS_TYPE_ELASTICSEARCH
        self.keys = DS_ELASTICSEARCH_KEYS