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


from dw_base.datax.datasources.data_source import DataSource
from dw_base.datax.datax_constants import DS_CLICK_HOUSE_JDBC_URL

# ClickHouse Data Source
DS_TYPE_CLICK_HOUSE = 'clickhouse'
DS_CLICK_HOUSE_KEYS = [DS_CLICK_HOUSE_JDBC_URL, 'username', 'password']


class ClickHouseDataSource(DataSource):
    def __init__(self, ds_file: str):
        super(ClickHouseDataSource, self).__init__(ds_file)
        self.source_type = DS_TYPE_CLICK_HOUSE
        self.keys = DS_CLICK_HOUSE_KEYS