Fix template_fields type to have MyPy friendly Sequence type (#20571)

Part of #19891

Author: potiuk

airflow/providers/airbyte/operators/airbyte.py

@@ -15,7 +15,7 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Sequence
 
 from airflow.models import BaseOperator
 from airflow.providers.airbyte.hooks.airbyte import AirbyteHook
@@ -51,7 +51,7 @@ class AirbyteTriggerSyncOperator(BaseOperator):
     :type timeout: float
     """
 
-    template_fields = ('connection_id',)
+    template_fields: Sequence[str] = ('connection_id',)
 
     def __init__(
         self,

airflow/providers/airbyte/sensors/airbyte.py

@@ -16,7 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains a Airbyte Job sensor."""
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Sequence
 
 from airflow.exceptions import AirflowException
 from airflow.providers.airbyte.hooks.airbyte import AirbyteHook
@@ -39,7 +39,7 @@ class AirbyteJobSensor(BaseSensorOperator):
     :type api_version: str
     """
 
-    template_fields = ('airbyte_job_id',)
+    template_fields: Sequence[str] = ('airbyte_job_id',)
     ui_color = '#6C51FD'
 
     def __init__(

airflow/providers/alibaba/cloud/sensors/oss_key.py

@@ -22,7 +22,7 @@
 else:
     from cached_property import cached_property
 
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Sequence
 from urllib.parse import urlparse
 
 from airflow.exceptions import AirflowException
@@ -51,7 +51,7 @@ class OSSKeySensor(BaseSensorOperator):
     :type oss_conn_id: Optional[str]
     """
 
-    template_fields = ('bucket_key', 'bucket_name')
+    template_fields: Sequence[str] = ('bucket_key', 'bucket_name')
 
     def __init__(
         self,

airflow/providers/amazon/aws/operators/athena.py

@@ -18,7 +18,7 @@
 #
 import sys
 import warnings
-from typing import TYPE_CHECKING, Any, Dict, Optional
+from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence
 from uuid import uuid4
 
 if sys.version_info >= (3, 8):
@@ -64,7 +64,7 @@ class AthenaOperator(BaseOperator):
     """
 
     ui_color = '#44b5e2'
-    template_fields = ('query', 'database', 'output_location')
+    template_fields: Sequence[str] = ('query', 'database', 'output_location')
     template_ext = ('.sql',)
     template_fields_renderers = {"query": "sql"}
 

airflow/providers/amazon/aws/operators/batch.py

@@ -26,7 +26,7 @@
     - http://boto3.readthedocs.io/en/latest/reference/services/batch.html
     - https://docs.aws.amazon.com/batch/latest/APIReference/Welcome.html
 """
-from typing import TYPE_CHECKING, Any, Optional
+from typing import TYPE_CHECKING, Any, Optional, Sequence
 
 from airflow.exceptions import AirflowException
 from airflow.models import BaseOperator
@@ -97,7 +97,7 @@ class AwsBatchOperator(BaseOperator):
 
     ui_color = "#c3dae0"
     arn = None  # type: Optional[str]
-    template_fields = (
+    template_fields: Sequence[str] = (
         "job_name",
         "overrides",
         "parameters",

airflow/providers/amazon/aws/operators/cloud_formation.py

@@ -16,7 +16,7 @@
 # specific language governing permissions and limitations
 # under the License.
 """This module contains CloudFormation create/delete stack operators."""
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, Optional, Sequence
 
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.cloud_formation import CloudFormationHook
@@ -40,7 +40,7 @@ class CloudFormationCreateStackOperator(BaseOperator):
     :type aws_conn_id: str
     """
 
-    template_fields: List[str] = ['stack_name']
+    template_fields: Sequence[str] = ('stack_name',)
     template_ext = ()
     ui_color = '#6b9659'
 
@@ -72,7 +72,7 @@ class CloudFormationDeleteStackOperator(BaseOperator):
     :type aws_conn_id: str
     """
 
-    template_fields: List[str] = ['stack_name']
+    template_fields: Sequence[str] = ('stack_name',)
     template_ext = ()
     ui_color = '#1d472b'
     ui_fgcolor = '#FFF'

airflow/providers/amazon/aws/operators/datasync.py

@@ -20,7 +20,7 @@
 import logging
 import random
 import warnings
-from typing import TYPE_CHECKING, List, Optional
+from typing import TYPE_CHECKING, List, Optional, Sequence
 
 from airflow.exceptions import AirflowException, AirflowTaskTimeout
 from airflow.models import BaseOperator
@@ -111,7 +111,7 @@ class DataSyncOperator(BaseOperator):
     :raises AirflowException: If Task creation, update, execution or delete fails.
     """
 
-    template_fields = (
+    template_fields: Sequence[str] = (
         "task_arn",
         "source_location_uri",
         "destination_location_uri",

airflow/providers/amazon/aws/operators/dms.py

@@ -17,7 +17,7 @@
 # under the License.
 
 
-from typing import TYPE_CHECKING, Dict, Optional
+from typing import TYPE_CHECKING, Dict, Optional, Sequence
 
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.dms import DmsHook
@@ -56,7 +56,7 @@ class DmsCreateTaskOperator(BaseOperator):
     :type aws_conn_id: Optional[str]
     """
 
-    template_fields = (
+    template_fields: Sequence[str] = (
         'replication_task_id',
         'source_endpoint_arn',
         'target_endpoint_arn',
@@ -134,7 +134,7 @@ class DmsDeleteTaskOperator(BaseOperator):
     :type aws_conn_id: Optional[str]
     """
 
-    template_fields = ('replication_task_arn',)
+    template_fields: Sequence[str] = ('replication_task_arn',)
     template_ext = ()
     template_fields_renderers: Dict[str, str] = {}
 
@@ -174,7 +174,7 @@ class DmsDescribeTasksOperator(BaseOperator):
     :type aws_conn_id: Optional[str]
     """
 
-    template_fields = ('describe_tasks_kwargs',)
+    template_fields: Sequence[str] = ('describe_tasks_kwargs',)
     template_ext = ()
     template_fields_renderers: Dict[str, str] = {'describe_tasks_kwargs': 'json'}
 
@@ -223,7 +223,7 @@ class DmsStartTaskOperator(BaseOperator):
     :type aws_conn_id: Optional[str]
     """
 
-    template_fields = (
+    template_fields: Sequence[str] = (
         'replication_task_arn',
         'start_replication_task_type',
         'start_task_kwargs',
@@ -276,7 +276,7 @@ class DmsStopTaskOperator(BaseOperator):
     :type aws_conn_id: Optional[str]
     """
 
-    template_fields = ('replication_task_arn',)
+    template_fields: Sequence[str] = ('replication_task_arn',)
     template_ext = ()
     template_fields_renderers: Dict[str, str] = {}
 

airflow/providers/amazon/aws/operators/ec2.py

@@ -17,7 +17,7 @@
 # under the License.
 #
 
-from typing import TYPE_CHECKING, Optional
+from typing import TYPE_CHECKING, Optional, Sequence
 
 from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.ec2 import EC2Hook
@@ -41,7 +41,7 @@ class EC2StartInstanceOperator(BaseOperator):
     :type check_interval: float
     """
 
-    template_fields = ("instance_id", "region_name")
+    template_fields: Sequence[str] = ("instance_id", "region_name")
     ui_color = "#eeaa11"
     ui_fgcolor = "#ffffff"
 
@@ -87,7 +87,7 @@ class EC2StopInstanceOperator(BaseOperator):
     :type check_interval: float
     """
 
-    template_fields = ("instance_id", "region_name")
+    template_fields: Sequence[str] = ("instance_id", "region_name")
     ui_color = "#eeaa11"
     ui_fgcolor = "#ffffff"
 

airflow/providers/amazon/aws/operators/ecs.py

@@ -22,7 +22,7 @@
 from datetime import datetime, timedelta
 from logging import Logger
 from threading import Event, Thread
-from typing import Dict, Generator, Optional
+from typing import Dict, Generator, Optional, Sequence
 
 from botocore.exceptions import ClientError
 from botocore.waiter import Waiter
@@ -225,7 +225,7 @@ class ECSOperator(BaseOperator):
     """
 
     ui_color = '#f0ede4'
-    template_fields = ('overrides',)
+    template_fields: Sequence[str] = ('overrides',)
     template_fields_renderers = {
         "overrides": "json",
         "network_configuration": "json",