From 8f81b9a01c7708a282271f9afd6b16a91011f105 Mon Sep 17 00:00:00 2001
From: Josh Fell <48934154+josh-fell@users.noreply.github.com>
Date: Tue, 8 Feb 2022 04:23:33 -0500
Subject: [PATCH] Add conditional `template_fields_renderers` check for new SQL
 lexers (#21403)

---
 airflow/providers/amazon/aws/operators/redshift_sql.py |  6 +++++-
 .../providers/apache/hive/transfers/hive_to_mysql.py   | 10 +++++++++-
 .../providers/apache/hive/transfers/mssql_to_hive.py   |  4 +++-
 airflow/providers/microsoft/mssql/operators/mssql.py   |  4 +++-
 airflow/providers/mysql/operators/mysql.py             |  7 ++++++-
 airflow/providers/mysql/transfers/presto_to_mysql.py   |  7 ++++++-
 airflow/providers/mysql/transfers/trino_to_mysql.py    |  7 ++++++-
 airflow/providers/mysql/transfers/vertica_to_mysql.py  |  8 ++++++--
 airflow/providers/postgres/operators/postgres.py       |  6 +++++-
 9 files changed, 49 insertions(+), 10 deletions(-)

diff --git a/airflow/providers/amazon/aws/operators/redshift_sql.py b/airflow/providers/amazon/aws/operators/redshift_sql.py
index d0b4f2b5bd45a..bfdd981d7adc9 100644
--- a/airflow/providers/amazon/aws/operators/redshift_sql.py
+++ b/airflow/providers/amazon/aws/operators/redshift_sql.py
@@ -19,6 +19,7 @@
 
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.redshift_sql import RedshiftSQLHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -44,7 +45,10 @@ class RedshiftSQLOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql',)
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {"sql": "postgresql"}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {
+        "sql": "postgresql" if "postgresql" in wwwutils.get_attr_renderer() else "sql"
+    }
 
     def __init__(
         self,
diff --git a/airflow/providers/apache/hive/transfers/hive_to_mysql.py b/airflow/providers/apache/hive/transfers/hive_to_mysql.py
index 1679471d6f4b3..c7817e5d69551 100644
--- a/airflow/providers/apache/hive/transfers/hive_to_mysql.py
+++ b/airflow/providers/apache/hive/transfers/hive_to_mysql.py
@@ -24,10 +24,14 @@
 from airflow.providers.apache.hive.hooks.hive import HiveServer2Hook
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 from airflow.utils.operator_helpers import context_to_airflow_vars
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 
+# TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+MYSQL_RENDERER = 'mysql' if 'mysql' in wwwutils.get_attr_renderer() else 'sql'
+
 
 class HiveToMySqlOperator(BaseOperator):
     """
@@ -57,7 +61,11 @@ class HiveToMySqlOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql', 'mysql_table', 'mysql_preoperator', 'mysql_postoperator')
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {'sql': 'hql', 'mysql_preoperator': 'mysql', 'mysql_postoperator': 'mysql'}
+    template_fields_renderers = {
+        'sql': 'hql',
+        'mysql_preoperator': MYSQL_RENDERER,
+        'mysql_postoperator': MYSQL_RENDERER,
+    }
     ui_color = '#a0e08c'
 
     def __init__(
diff --git a/airflow/providers/apache/hive/transfers/mssql_to_hive.py b/airflow/providers/apache/hive/transfers/mssql_to_hive.py
index 5668e715758a1..912c2a58a36bd 100644
--- a/airflow/providers/apache/hive/transfers/mssql_to_hive.py
+++ b/airflow/providers/apache/hive/transfers/mssql_to_hive.py
@@ -28,6 +28,7 @@
 from airflow.models import BaseOperator
 from airflow.providers.apache.hive.hooks.hive import HiveCliHook
 from airflow.providers.microsoft.mssql.hooks.mssql import MsSqlHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -65,7 +66,8 @@ class MsSqlToHiveOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql', 'partition', 'hive_table')
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {'sql': 'tsql'}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {'sql': 'tsql' if 'tsql' in wwwutils.get_attr_renderer() else 'sql'}
     ui_color = '#a0e08c'
 
     def __init__(
diff --git a/airflow/providers/microsoft/mssql/operators/mssql.py b/airflow/providers/microsoft/mssql/operators/mssql.py
index c1b853f6bd0a8..7082c7c52d90b 100644
--- a/airflow/providers/microsoft/mssql/operators/mssql.py
+++ b/airflow/providers/microsoft/mssql/operators/mssql.py
@@ -20,6 +20,7 @@
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
 from airflow.providers.microsoft.mssql.hooks.mssql import MsSqlHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.hooks.dbapi import DbApiHook
@@ -49,7 +50,8 @@ class MsSqlOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql',)
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {'sql': 'tsql'}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {'sql': 'tsql' if 'tsql' in wwwutils.get_attr_renderer() else 'sql'}
     ui_color = '#ededed'
 
     def __init__(
diff --git a/airflow/providers/mysql/operators/mysql.py b/airflow/providers/mysql/operators/mysql.py
index b42e5ded9f686..d51a97e6fe38a 100644
--- a/airflow/providers/mysql/operators/mysql.py
+++ b/airflow/providers/mysql/operators/mysql.py
@@ -20,6 +20,7 @@
 
 from airflow.models import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -47,7 +48,11 @@ class MySqlOperator(BaseOperator):
     """
 
     template_fields: Sequence[str] = ('sql', 'parameters')
-    template_fields_renderers = {'sql': 'mysql', 'parameters': 'json'}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {
+        'sql': 'mysql' if 'mysql' in wwwutils.get_attr_renderer() else 'sql',
+        'parameters': 'json',
+    }
     template_ext: Sequence[str] = ('.sql', '.json')
     ui_color = '#ededed'
 
diff --git a/airflow/providers/mysql/transfers/presto_to_mysql.py b/airflow/providers/mysql/transfers/presto_to_mysql.py
index e596117bfc536..45529237729e0 100644
--- a/airflow/providers/mysql/transfers/presto_to_mysql.py
+++ b/airflow/providers/mysql/transfers/presto_to_mysql.py
@@ -20,6 +20,7 @@
 from airflow.models import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 from airflow.providers.presto.hooks.presto import PrestoHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -44,7 +45,11 @@ class PrestoToMySqlOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql', 'mysql_table', 'mysql_preoperator')
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {"sql": "sql", "mysql_preoperator": "mysql"}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {
+        "sql": "sql",
+        "mysql_preoperator": "mysql" if "mysql" in wwwutils.get_attr_renderer() else "sql",
+    }
     ui_color = '#a0e08c'
 
     def __init__(
diff --git a/airflow/providers/mysql/transfers/trino_to_mysql.py b/airflow/providers/mysql/transfers/trino_to_mysql.py
index 93ff183c6c79d..3c013a7eaed2a 100644
--- a/airflow/providers/mysql/transfers/trino_to_mysql.py
+++ b/airflow/providers/mysql/transfers/trino_to_mysql.py
@@ -20,6 +20,7 @@
 from airflow.models import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 from airflow.providers.trino.hooks.trino import TrinoHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -44,7 +45,11 @@ class TrinoToMySqlOperator(BaseOperator):
 
     template_fields: Sequence[str] = ('sql', 'mysql_table', 'mysql_preoperator')
     template_ext: Sequence[str] = ('.sql',)
-    template_fields_renderers = {"sql": "sql", "mysql_preoperator": "mysql"}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {
+        "sql": "sql",
+        "mysql_preoperator": "mysql" if "mysql" in wwwutils.get_attr_renderer() else "sql",
+    }
     ui_color = '#a0e08c'
 
     def __init__(
diff --git a/airflow/providers/mysql/transfers/vertica_to_mysql.py b/airflow/providers/mysql/transfers/vertica_to_mysql.py
index 42016eb698420..595b2cb01b3dc 100644
--- a/airflow/providers/mysql/transfers/vertica_to_mysql.py
+++ b/airflow/providers/mysql/transfers/vertica_to_mysql.py
@@ -26,10 +26,14 @@
 from airflow.models import BaseOperator
 from airflow.providers.mysql.hooks.mysql import MySqlHook
 from airflow.providers.vertica.hooks.vertica import VerticaHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 
+# TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+MYSQL_RENDERER = 'mysql' if 'mysql' in wwwutils.get_attr_renderer() else 'sql'
+
 
 class VerticaToMySqlOperator(BaseOperator):
     """
@@ -57,8 +61,8 @@ class VerticaToMySqlOperator(BaseOperator):
     template_ext: Sequence[str] = ('.sql',)
     template_fields_renderers = {
         "sql": "sql",
-        "mysql_preoperator": "mysql",
-        "mysql_postoperator": "mysql",
+        "mysql_preoperator": MYSQL_RENDERER,
+        "mysql_postoperator": MYSQL_RENDERER,
     }
     ui_color = '#a0e08c'
 
diff --git a/airflow/providers/postgres/operators/postgres.py b/airflow/providers/postgres/operators/postgres.py
index db3156349c179..562fa7a3002fc 100644
--- a/airflow/providers/postgres/operators/postgres.py
+++ b/airflow/providers/postgres/operators/postgres.py
@@ -19,6 +19,7 @@
 
 from airflow.models import BaseOperator
 from airflow.providers.postgres.hooks.postgres import PostgresHook
+from airflow.www import utils as wwwutils
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
@@ -40,7 +41,10 @@ class PostgresOperator(BaseOperator):
     """
 
     template_fields: Sequence[str] = ('sql',)
-    template_fields_renderers = {'sql': 'postgresql'}
+    # TODO: Remove renderer check when the provider has an Airflow 2.3+ requirement.
+    template_fields_renderers = {
+        'sql': 'postgresql' if 'postgresql' in wwwutils.get_attr_renderer() else 'sql'
+    }
     template_ext: Sequence[str] = ('.sql',)
     ui_color = '#ededed'