class DrillOperator(BaseOperator):
"""
Executes the provided SQL in the identified Drill environment.
.. seealso::
For more information on how to use this operator, take a look at the guide:
:ref:`howto/operator:DrillOperator`
:param sql: the SQL code to be executed. (templated)
:type sql: Can receive a str representing a sql statement,
a list of str (sql statements), or a reference to a template file.
Template references are recognized by str ending in '.sql'
:param drill_conn_id: id of the connection config for the target Drill
environment
:type drill_conn_id: str
:param parameters: (optional) the parameters to render the SQL query with.
:type parameters: dict or iterable
"""
template_fields = ('sql', )
template_fields_renderers = {'sql': 'sql'}
template_ext = ('.sql', )
ui_color = '#ededed'
@apply_defaults
def __init__(
self,
*,
sql: str,
drill_conn_id: str = 'drill_default',
parameters: Optional[Union[Mapping, Iterable]] = None,
**kwargs,
) -> None:
super().__init__(**kwargs)
self.sql = sql
self.drill_conn_id = drill_conn_id
self.parameters = parameters
self.hook = None
def execute(self, context):
self.log.info('Executing: %s on %s', self.sql, self.drill_conn_id)
self.hook = DrillHook(drill_conn_id=self.drill_conn_id)
sql = sqlparse.split(sqlparse.format(self.sql, strip_comments=True))
no_term_sql = [s[:-1] for s in sql if s[-1] == ';']
self.hook.run(no_term_sql, parameters=self.parameters)
class DrillOperator(BaseOperator):
"""
Executes the provided SQL in the identified Drill environment.
.. seealso::
For more information on how to use this operator, take a look at the guide:
:ref:`howto/operator:DrillOperator`
:param sql: the SQL code to be executed as a single string, or
a list of str (sql statements), or a reference to a template file.
Template references are recognized by str ending in '.sql'
:param drill_conn_id: id of the connection config for the target Drill
environment
:param parameters: (optional) the parameters to render the SQL query with.
"""
template_fields: Sequence[str] = ('sql', )
template_fields_renderers = {'sql': 'sql'}
template_ext: Sequence[str] = ('.sql', )
ui_color = '#ededed'
def __init__(
self,
*,
sql: str,
drill_conn_id: str = 'drill_default',
parameters: Optional[Union[Iterable, Mapping]] = None,
**kwargs,
) -> None:
super().__init__(**kwargs)
self.sql = sql
self.drill_conn_id = drill_conn_id
self.parameters = parameters
self.hook: Optional[DrillHook] = None
def execute(self, context: 'Context'):
self.log.info('Executing: %s on %s', self.sql, self.drill_conn_id)
self.hook = DrillHook(drill_conn_id=self.drill_conn_id)
self.hook.run(self.sql,
parameters=self.parameters,
split_statements=True)
def execute(self, context):
self.log.info('Executing: %s on %s', self.sql, self.drill_conn_id)
self.hook = DrillHook(drill_conn_id=self.drill_conn_id)
sql = sqlparse.split(sqlparse.format(self.sql, strip_comments=True))
no_term_sql = [s[:-1] for s in sql if s[-1] == ';']
self.hook.run(no_term_sql, parameters=self.parameters)
def execute(self, context: 'Context'):
self.log.info('Executing: %s on %s', self.sql, self.drill_conn_id)
self.hook = DrillHook(drill_conn_id=self.drill_conn_id)
self.hook.run(self.sql,
parameters=self.parameters,
split_statements=True)