Move all "old" SQL operators to common.sql providers

Previously, in apache#24836 we moved Hooks and added some new operators to the common.sql package. Now we are salso moving the operators and sensors to common.sql.
potiuk · Jul 27, 2022 · d7b1860 · d7b1860
1 parent 5d4abbd
commit d7b1860
Show file tree

Hide file tree

Showing 22 changed files with 613 additions and 565 deletions.
diff --git a/.pre-commit-config.yaml b/.pre-commit-config.yaml
@@ -376,7 +376,7 @@ repos:
         name: Update cross-dependencies for providers packages
         entry: ./scripts/ci/pre_commit/pre_commit_build_providers_dependencies.py
         language: python
-        files: ^airflow/providers/.*\.py$|^tests/providers/.*\.py$|^tests/system/providers/.*\.py$|$airflow/providers/.*/provider.yaml$
+        files: ^airflow/providers/.*\.py$|^tests/providers/.*\.py$|^tests/system/providers/.*\.py$|^airflow/providers/.*/provider.yaml$
         pass_filenames: false
         additional_dependencies: ['setuptools', 'rich>=12.4.4', 'pyyaml']
       - id: update-extras

diff --git a/airflow/operators/check_operator.py b/airflow/operators/check_operator.py
@@ -20,7 +20,7 @@
 
 import warnings
 
-from airflow.operators.sql import (
+from airflow.providers.common.sql.operators.sql import (
     SQLCheckOperator,
     SQLIntervalCheckOperator,
     SQLThresholdCheckOperator,

diff --git a/airflow/operators/presto_check_operator.py b/airflow/operators/presto_check_operator.py
@@ -19,7 +19,11 @@
 
 import warnings
 
-from airflow.operators.sql import SQLCheckOperator, SQLIntervalCheckOperator, SQLValueCheckOperator
+from airflow.providers.common.sql.operators.sql import (
+    SQLCheckOperator,
+    SQLIntervalCheckOperator,
+    SQLValueCheckOperator,
+)
 
 warnings.warn(
     "This module is deprecated. Please use `airflow.operators.sql`.", DeprecationWarning, stacklevel=2