Repository: incubator-airflow Updated Branches: refs/heads/master f21e97e99 -> 0f9112daf
[AIRFLOW-657] Add AutoCommit Parameter for MSSQL Closes #1908 from robin-miller-ow/MsSqlAutoCommit Project: http://git-wip-us.apache.org/repos/asf/incubator-airflow/repo Commit: http://git-wip-us.apache.org/repos/asf/incubator-airflow/commit/0f9112da Tree: http://git-wip-us.apache.org/repos/asf/incubator-airflow/tree/0f9112da Diff: http://git-wip-us.apache.org/repos/asf/incubator-airflow/diff/0f9112da Branch: refs/heads/master Commit: 0f9112daf826d12c3c408d78e66000573344175c Parents: f21e97e Author: robin_miller_ow <[email protected]> Authored: Sun Dec 25 14:47:02 2016 +0100 Committer: Bolke de Bruin <[email protected]> Committed: Sun Dec 25 14:47:05 2016 +0100 ---------------------------------------------------------------------- airflow/operators/mssql_operator.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/incubator-airflow/blob/0f9112da/airflow/operators/mssql_operator.py ---------------------------------------------------------------------- diff --git a/airflow/operators/mssql_operator.py b/airflow/operators/mssql_operator.py index ed9bdf4..0590454 100644 --- a/airflow/operators/mssql_operator.py +++ b/airflow/operators/mssql_operator.py @@ -36,13 +36,14 @@ class MsSqlOperator(BaseOperator): @apply_defaults def __init__( self, sql, mssql_conn_id='mssql_default', parameters=None, - *args, **kwargs): + autocommit=False, *args, **kwargs): super(MsSqlOperator, self).__init__(*args, **kwargs) self.mssql_conn_id = mssql_conn_id self.sql = sql self.parameters = parameters + self.autocommit = autocommit def execute(self, context): logging.info('Executing: ' + str(self.sql)) hook = MsSqlHook(mssql_conn_id=self.mssql_conn_id) - hook.run(self.sql, parameters=self.parameters) + hook.run(self.sql, autocommit=self.autocommit, parameters=self.parameters)
