Remove Provider Deprecations in Trino (apache#44717)

* remove provider deprecations in trino * remove trino.py and test_trino.py and fix mypy test * add trino.py and test_trino.py to fix static checks * fix static checks --------- Co-authored-by: pratiksha rajendrabhai badheka <pratiksha@DESKTOP-T5HUA05>
astronomer · Dec 10, 2024 · 43ad2ca · 43ad2ca
1 parent 1fbd889
commit 43ad2ca
Show file tree

Hide file tree

Showing 7 changed files with 11 additions and 232 deletions.
diff --git a/providers/src/airflow/providers/trino/CHANGELOG.rst b/providers/src/airflow/providers/trino/CHANGELOG.rst
@@ -27,6 +27,15 @@
 Changelog
 ---------
 
+main
+....
+
+.. warning::
+  All deprecated classes, parameters and features have been removed from the {provider_name} provider package.
+  The following breaking changes were introduced:
+
+  * Remove ``TrinoOperator`` class from airflow.providers.trino.operators.trino. Please use ``airflow.providers.common.sql.operators.sql.SQLExecuteQueryOperator```
+
 5.9.0
 .....
 

diff --git a/providers/src/airflow/providers/trino/operators/__init__.py b/providers/src/airflow/providers/trino/operators/__init__.py
diff --git a/providers/src/airflow/providers/trino/operators/trino.py b/providers/src/airflow/providers/trino/operators/trino.py
diff --git a/providers/src/airflow/providers/trino/provider.yaml b/providers/src/airflow/providers/trino/provider.yaml
@@ -83,11 +83,6 @@ integrations:
       - /docs/apache-airflow-providers-trino/operators/trino.rst
     tags: [software]
 
-operators:
-  - integration-name: Trino
-    python-modules:
-      - airflow.providers.trino.operators.trino
-
 asset-uris:
   - schemes: [trino]
     handler: airflow.providers.trino.assets.trino.sanitize_uri

diff --git a/providers/tests/integration/trino/hooks/test_trino.py b/providers/tests/integration/trino/hooks/test_trino.py
@@ -21,8 +21,8 @@
 
 import pytest
 
+from airflow.providers.common.sql.operators.sql import SQLExecuteQueryOperator
 from airflow.providers.trino.hooks.trino import TrinoHook
-from airflow.providers.trino.operators.trino import TrinoOperator
 
 
 @pytest.mark.integration("trino")
@@ -50,7 +50,7 @@ def test_should_record_records_with_kerberos_auth(self):
 
     @mock.patch.dict("os.environ", AIRFLOW_CONN_TRINO_DEFAULT="trino://airflow@trino:8080/")
     def test_openlineage_methods(self):
-        op = TrinoOperator(task_id="trino_test", sql="SELECT name FROM tpch.sf1.customer LIMIT 3")
+        op = SQLExecuteQueryOperator(task_id="trino_test", sql="SELECT name FROM tpch.sf1.customer LIMIT 3")
         op.execute({})
         lineage = op.get_openlineage_facets_on_start()
         assert lineage.inputs[0].namespace == "trino://trino:8080"

diff --git a/providers/tests/trino/operators/__init__.py b/providers/tests/trino/operators/__init__.py
diff --git a/providers/tests/trino/operators/test_trino.py b/providers/tests/trino/operators/test_trino.py