def _get_jdbc_params(self) -> Dict[str, str]:
"""Get JDBC parameters."""
jdbc_config = read_config(JDBC_CONFIG)
try:
db = jdbc_config[self.source_system.upper()]
except KeyError as e:
raise KeyError('Section %s is missing in jdbc config.' % e)
try:
jdbc_params = {
'TYPE':
db['type'],
'URI':
db['uri'],
'PORT':
db['port'],
'USER':
db['user'],
'PASSWORD':
self._get_jdbc_pass(db),
'DATABASE':
db.get('database').format(
schema=self.get_source_subsystem().upper()),
'SCHEMA':
self.get_null_param('schema', self.get_source_subsystem()),
'DRIVER_PATH':
JDBC_DRIVER_PATH.get(db['type']),
'FETCH_ROWS':
self.get_null_param('fetch_rows', self.fetch_rows)
}
return jdbc_params
except KeyError as e:
raise KeyError(
f'In section {self.source_system.upper()} attrubute {e} is missing in jdbc config.'
)
def _get_es_params(self) -> Dict[str, str]:
"""Get JDBC parameters."""
es_conf = read_config(ES_CONFIG)
try:
es = es_conf[self.source_system.upper()]
except KeyError as e:
raise KeyError('Section %s is missing in es config.' % e)
try:
es_params = {
'ES_URI': es['uri'],
'ES_PORT': es['port'],
'ES_USER': es['user'],
'ES_PASSWORD': self._get_es_pass(es),
'ES_SCHEME': es['scheme'],
}
return es_params
except KeyError as e:
raise KeyError(
f'In section {self.source_system.upper()} attrubute {e} is missing in jdbc config.'
)
def __init__(self, config_file: str):
self.cfg = read_config(config_file)