Update models.py

add Oracle SQL support through the OracleHook
praveenaki · Sep 7, 2015 · 8eff6e4 · 8eff6e4
1 parent 2a2dfc7
commit 8eff6e4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/airflow/models.py b/airflow/models.py
@@ -388,6 +388,8 @@ def get_hook(self):
                 return hooks.JdbcHook(jdbc_conn_id=self.conn_id)
             elif self.conn_type == 'mssql':
                 return hooks.MsSqlHook(mssql_conn_id=self.conn_id)
+            elif self.conn_type == 'oracle':
+                return hooks.OracleHook(oracle_conn_id=self.conn_id)
         except:
             return None