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


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

# PostgreSQL Data Source
DS_TYPE_POSTGRE_SQL = 'postgresql'
DS_POSTGRE_SQL_KEYS = [DS_POSTGRE_SQL_JDBC_URL, 'username', 'password']


class PostgreSQLDataSource(DataSource):
    def __init__(self, ds_file: str):
        super(PostgreSQLDataSource, self).__init__(ds_file)
        self.source_type = DS_TYPE_POSTGRE_SQL
        self.keys = DS_POSTGRE_SQL_KEYS