apache-airflow-providers-amazon 8.16.0rc1__py3-none-any.whl → 8.17.0rc2__py3-none-any.whl

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

@@ -19,14 +19,15 @@ from __future__ import annotations
 
 from typing import TYPE_CHECKING, Sequence
 
-from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.dms import DmsHook
+from airflow.providers.amazon.aws.operators.base_aws import AwsBaseOperator
+from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 
 
-class DmsCreateTaskOperator(BaseOperator):
+class DmsCreateTaskOperator(AwsBaseOperator[DmsHook]):
     """
     Creates AWS DMS replication task.
 
@@ -42,13 +43,19 @@ class DmsCreateTaskOperator(BaseOperator):
     :param migration_type: Migration type ('full-load'|'cdc'|'full-load-and-cdc'), full-load by default.
     :param create_task_kwargs: Extra arguments for DMS replication task creation.
     :param aws_conn_id: The Airflow connection used for AWS credentials.
-        If this is None or empty then the default boto3 behaviour is used. If
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
         running Airflow in a distributed manner and aws_conn_id is None or
         empty, then default boto3 configuration would be used (and must be
         maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
     """
 
-    template_fields: Sequence[str] = (
+    aws_hook_class = DmsHook
+    template_fields: Sequence[str] = aws_template_fields(
         "replication_task_id",
         "source_endpoint_arn",
         "target_endpoint_arn",
@@ -57,7 +64,6 @@ class DmsCreateTaskOperator(BaseOperator):
         "migration_type",
         "create_task_kwargs",
     )
-    template_ext: Sequence[str] = ()
     template_fields_renderers = {
         "table_mappings": "json",
         "create_task_kwargs": "json",
@@ -92,9 +98,7 @@ class DmsCreateTaskOperator(BaseOperator):
 
         :return: replication task arn
         """
-        dms_hook = DmsHook(aws_conn_id=self.aws_conn_id)
-
-        task_arn = dms_hook.create_replication_task(
+        task_arn = self.hook.create_replication_task(
             replication_task_id=self.replication_task_id,
             source_endpoint_arn=self.source_endpoint_arn,
             target_endpoint_arn=self.target_endpoint_arn,
@@ -108,7 +112,7 @@ class DmsCreateTaskOperator(BaseOperator):
         return task_arn
 
 
-class DmsDeleteTaskOperator(BaseOperator):
+class DmsDeleteTaskOperator(AwsBaseOperator[DmsHook]):
     """
     Deletes AWS DMS replication task.
 
@@ -118,26 +122,23 @@ class DmsDeleteTaskOperator(BaseOperator):
 
     :param replication_task_arn: Replication task ARN
     :param aws_conn_id: The Airflow connection used for AWS credentials.
-        If this is None or empty then the default boto3 behaviour is used. If
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
         running Airflow in a distributed manner and aws_conn_id is None or
         empty, then default boto3 configuration would be used (and must be
         maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
     """
 
-    template_fields: Sequence[str] = ("replication_task_arn",)
-    template_ext: Sequence[str] = ()
-    template_fields_renderers: dict[str, str] = {}
+    aws_hook_class = DmsHook
+    template_fields: Sequence[str] = aws_template_fields("replication_task_arn")
 
-    def __init__(
-        self,
-        *,
-        replication_task_arn: str | None = None,
-        aws_conn_id: str = "aws_default",
-        **kwargs,
-    ):
+    def __init__(self, *, replication_task_arn: str | None = None, **kwargs):
         super().__init__(**kwargs)
         self.replication_task_arn = replication_task_arn
-        self.aws_conn_id = aws_conn_id
 
     def execute(self, context: Context):
         """
@@ -145,12 +146,11 @@ class DmsDeleteTaskOperator(BaseOperator):
 
         :return: replication task arn
         """
-        dms_hook = DmsHook(aws_conn_id=self.aws_conn_id)
-        dms_hook.delete_replication_task(replication_task_arn=self.replication_task_arn)
+        self.hook.delete_replication_task(replication_task_arn=self.replication_task_arn)
         self.log.info("DMS replication task(%s) has been deleted.", self.replication_task_arn)
 
 
-class DmsDescribeTasksOperator(BaseOperator):
+class DmsDescribeTasksOperator(AwsBaseOperator[DmsHook]):
     """
     Describes AWS DMS replication tasks.
 
@@ -160,26 +160,24 @@ class DmsDescribeTasksOperator(BaseOperator):
 
     :param describe_tasks_kwargs: Describe tasks command arguments
     :param aws_conn_id: The Airflow connection used for AWS credentials.
-        If this is None or empty then the default boto3 behaviour is used. If
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
         running Airflow in a distributed manner and aws_conn_id is None or
         empty, then default boto3 configuration would be used (and must be
         maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
     """
 
-    template_fields: Sequence[str] = ("describe_tasks_kwargs",)
-    template_ext: Sequence[str] = ()
+    aws_hook_class = DmsHook
+    template_fields: Sequence[str] = aws_template_fields("describe_tasks_kwargs")
     template_fields_renderers: dict[str, str] = {"describe_tasks_kwargs": "json"}
 
-    def __init__(
-        self,
-        *,
-        describe_tasks_kwargs: dict | None = None,
-        aws_conn_id: str = "aws_default",
-        **kwargs,
-    ):
+    def __init__(self, *, describe_tasks_kwargs: dict | None = None, **kwargs):
         super().__init__(**kwargs)
         self.describe_tasks_kwargs = describe_tasks_kwargs or {}
-        self.aws_conn_id = aws_conn_id
 
     def execute(self, context: Context) -> tuple[str | None, list]:
         """
@@ -187,11 +185,10 @@ class DmsDescribeTasksOperator(BaseOperator):
 
         :return: Marker and list of replication tasks
         """
-        dms_hook = DmsHook(aws_conn_id=self.aws_conn_id)
-        return dms_hook.describe_replication_tasks(**self.describe_tasks_kwargs)
+        return self.hook.describe_replication_tasks(**self.describe_tasks_kwargs)
 
 
-class DmsStartTaskOperator(BaseOperator):
+class DmsStartTaskOperator(AwsBaseOperator[DmsHook]):
     """
     Starts AWS DMS replication task.
 
@@ -204,18 +201,23 @@ class DmsStartTaskOperator(BaseOperator):
         ('start-replication'|'resume-processing'|'reload-target')
     :param start_task_kwargs: Extra start replication task arguments
     :param aws_conn_id: The Airflow connection used for AWS credentials.
-        If this is None or empty then the default boto3 behaviour is used. If
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
         running Airflow in a distributed manner and aws_conn_id is None or
         empty, then default boto3 configuration would be used (and must be
         maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
     """
 
-    template_fields: Sequence[str] = (
+    aws_hook_class = DmsHook
+    template_fields: Sequence[str] = aws_template_fields(
         "replication_task_arn",
         "start_replication_task_type",
         "start_task_kwargs",
     )
-    template_ext: Sequence[str] = ()
     template_fields_renderers = {"start_task_kwargs": "json"}
 
     def __init__(
@@ -234,14 +236,8 @@ class DmsStartTaskOperator(BaseOperator):
         self.aws_conn_id = aws_conn_id
 
     def execute(self, context: Context):
-        """
-        Start AWS DMS replication task from Airflow.
-
-        :return: replication task arn
-        """
-        dms_hook = DmsHook(aws_conn_id=self.aws_conn_id)
-
-        dms_hook.start_replication_task(
+        """Start AWS DMS replication task from Airflow."""
+        self.hook.start_replication_task(
             replication_task_arn=self.replication_task_arn,
             start_replication_task_type=self.start_replication_task_type,
             **self.start_task_kwargs,
@@ -249,7 +245,7 @@ class DmsStartTaskOperator(BaseOperator):
         self.log.info("DMS replication task(%s) is starting.", self.replication_task_arn)
 
 
-class DmsStopTaskOperator(BaseOperator):
+class DmsStopTaskOperator(AwsBaseOperator[DmsHook]):
     """
     Stops AWS DMS replication task.
 
@@ -259,33 +255,25 @@ class DmsStopTaskOperator(BaseOperator):
 
     :param replication_task_arn: Replication task ARN
     :param aws_conn_id: The Airflow connection used for AWS credentials.
-        If this is None or empty then the default boto3 behaviour is used. If
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
         running Airflow in a distributed manner and aws_conn_id is None or
         empty, then default boto3 configuration would be used (and must be
         maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.html
     """
 
-    template_fields: Sequence[str] = ("replication_task_arn",)
-    template_ext: Sequence[str] = ()
-    template_fields_renderers: dict[str, str] = {}
+    aws_hook_class = DmsHook
+    template_fields: Sequence[str] = aws_template_fields("replication_task_arn")
 
-    def __init__(
-        self,
-        *,
-        replication_task_arn: str | None = None,
-        aws_conn_id: str = "aws_default",
-        **kwargs,
-    ):
+    def __init__(self, *, replication_task_arn: str | None = None, **kwargs):
         super().__init__(**kwargs)
         self.replication_task_arn = replication_task_arn
-        self.aws_conn_id = aws_conn_id
 
     def execute(self, context: Context):
-        """
-        Stop AWS DMS replication task from Airflow.
-
-        :return: replication task arn
-        """
-        dms_hook = DmsHook(aws_conn_id=self.aws_conn_id)
-        dms_hook.stop_replication_task(replication_task_arn=self.replication_task_arn)
+        """Stop AWS DMS replication task from Airflow."""
+        self.hook.stop_replication_task(replication_task_arn=self.replication_task_arn)
         self.log.info("DMS replication task(%s) is stopping.", self.replication_task_arn)

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

@@ -183,22 +183,36 @@ class EC2CreateInstanceOperator(BaseOperator):
             MaxCount=self.max_count,
             **self.config,
         )["Instances"]
-        instance_ids = []
-        for instance in instances:
-            instance_ids.append(instance["InstanceId"])
-            self.log.info("Created EC2 instance %s", instance["InstanceId"])
+
+        instance_ids = self._on_kill_instance_ids = [instance["InstanceId"] for instance in instances]
+        for instance_id in instance_ids:
+            self.log.info("Created EC2 instance %s", instance_id)
 
             if self.wait_for_completion:
                 ec2_hook.get_waiter("instance_running").wait(
-                    InstanceIds=[instance["InstanceId"]],
+                    InstanceIds=[instance_id],
                     WaiterConfig={
                         "Delay": self.poll_interval,
                         "MaxAttempts": self.max_attempts,
                     },
                 )
 
+        # leave "_on_kill_instance_ids" in place for finishing post-processing
         return instance_ids
 
+    def on_kill(self) -> None:
+        instance_ids = getattr(self, "_on_kill_instance_ids", [])
+
+        if instance_ids:
+            self.log.info("on_kill: Terminating instance/s %s", ", ".join(instance_ids))
+            ec2_hook = EC2Hook(
+                aws_conn_id=self.aws_conn_id,
+                region_name=self.region_name,
+                api_type="client_type",
+            )
+            ec2_hook.conn.terminate_instances(InstanceIds=instance_ids)
+        super().on_kill()
+
 
 class EC2TerminateInstanceOperator(BaseOperator):
     """

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

@@ -716,7 +716,9 @@ class EmrCreateJobFlowOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_countdown has been deprecated to standardize "
                 "naming conventions.  Please use waiter_max_attempts instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
             # waiter_countdown defaults to never timing out, which is not supported
             # by boto waiters, so we will set it here to "a very long time" for now.
@@ -725,7 +727,9 @@ class EmrCreateJobFlowOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_check_interval_seconds has been deprecated to "
                 "standardize naming conventions.  Please use waiter_delay instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
             waiter_delay = waiter_check_interval_seconds
         super().__init__(**kwargs)
@@ -1024,7 +1028,9 @@ class EmrServerlessCreateApplicationOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_check_interval_seconds has been deprecated to standardize "
                 "naming conventions.  Please use waiter_delay instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         if waiter_countdown is NOTSET:
             waiter_max_attempts = 25 if waiter_max_attempts is NOTSET else waiter_max_attempts
@@ -1036,7 +1042,9 @@ class EmrServerlessCreateApplicationOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_countdown has been deprecated to standardize "
                 "naming conventions.  Please use waiter_max_attempts instead. In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         self.aws_conn_id = aws_conn_id
         self.release_label = release_label
@@ -1205,7 +1213,9 @@ class EmrServerlessStartJobOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_check_interval_seconds has been deprecated to standardize "
                 "naming conventions.  Please use waiter_delay instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         if waiter_countdown is NOTSET:
             waiter_max_attempts = 25 if waiter_max_attempts is NOTSET else waiter_max_attempts
@@ -1217,7 +1227,9 @@ class EmrServerlessStartJobOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_countdown has been deprecated to standardize "
                 "naming conventions.  Please use waiter_max_attempts instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         self.aws_conn_id = aws_conn_id
         self.application_id = application_id
@@ -1408,7 +1420,9 @@ class EmrServerlessStopApplicationOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_check_interval_seconds has been deprecated to standardize "
                 "naming conventions.  Please use waiter_delay instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         if waiter_countdown is NOTSET:
             waiter_max_attempts = 25 if waiter_max_attempts is NOTSET else waiter_max_attempts
@@ -1420,7 +1434,9 @@ class EmrServerlessStopApplicationOperator(BaseOperator):
             warnings.warn(
                 "The parameter waiter_countdown has been deprecated to standardize "
                 "naming conventions.  Please use waiter_max_attempts instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         self.aws_conn_id = aws_conn_id
         self.application_id = application_id
@@ -1569,7 +1585,9 @@ class EmrServerlessDeleteApplicationOperator(EmrServerlessStopApplicationOperato
             warnings.warn(
                 "The parameter waiter_check_interval_seconds has been deprecated to standardize "
                 "naming conventions.  Please use waiter_delay instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         if waiter_countdown is NOTSET:
             waiter_max_attempts = 25 if waiter_max_attempts is NOTSET else waiter_max_attempts
@@ -1581,7 +1599,9 @@ class EmrServerlessDeleteApplicationOperator(EmrServerlessStopApplicationOperato
             warnings.warn(
                 "The parameter waiter_countdown has been deprecated to standardize "
                 "naming conventions.  Please use waiter_max_attempts instead.  In the "
-                "future this will default to None and defer to the waiter's default value."
+                "future this will default to None and defer to the waiter's default value.",
+                AirflowProviderDeprecationWarning,
+                stacklevel=2,
             )
         self.wait_for_delete_completion = wait_for_completion
         # super stops the app

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

@@ -16,19 +16,19 @@
 # under the License.
 from __future__ import annotations
 
-from functools import cached_property
 from typing import TYPE_CHECKING, Sequence
 
 from airflow.exceptions import AirflowException
-from airflow.models import BaseOperator
 from airflow.providers.amazon.aws.hooks.eventbridge import EventBridgeHook
+from airflow.providers.amazon.aws.operators.base_aws import AwsBaseOperator
+from airflow.providers.amazon.aws.utils.mixins import aws_template_fields
 from airflow.utils.helpers import prune_dict
 
 if TYPE_CHECKING:
     from airflow.utils.context import Context
 
 
-class EventBridgePutEventsOperator(BaseOperator):
+class EventBridgePutEventsOperator(AwsBaseOperator[EventBridgeHook]):
     """
     Put Events onto Amazon EventBridge.
 
@@ -38,32 +38,25 @@ class EventBridgePutEventsOperator(BaseOperator):
 
     :param entries: the list of events to be put onto EventBridge, each event is a dict (required)
     :param endpoint_id: the URL subdomain of the endpoint
-    :param aws_conn_id: the AWS connection to use
-    :param region_name: the region where events are to be sent
-
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
+        running Airflow in a distributed manner and aws_conn_id is None or
+        empty, then default boto3 configuration would be used (and must be
+        maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.htmlt
     """
 
-    template_fields: Sequence[str] = ("entries", "endpoint_id", "aws_conn_id", "region_name")
+    aws_hook_class = EventBridgeHook
+    template_fields: Sequence[str] = aws_template_fields("entries", "endpoint_id")
 
-    def __init__(
-        self,
-        *,
-        entries: list[dict],
-        endpoint_id: str | None = None,
-        aws_conn_id: str = "aws_default",
-        region_name: str | None = None,
-        **kwargs,
-    ):
+    def __init__(self, *, entries: list[dict], endpoint_id: str | None = None, **kwargs):
         super().__init__(**kwargs)
         self.entries = entries
         self.endpoint_id = endpoint_id
-        self.aws_conn_id = aws_conn_id
-        self.region_name = region_name
-
-    @cached_property
-    def hook(self) -> EventBridgeHook:
-        """Create and return an EventBridgeHook."""
-        return EventBridgeHook(aws_conn_id=self.aws_conn_id, region_name=self.region_name)
 
     def execute(self, context: Context):
         response = self.hook.conn.put_events(
@@ -90,7 +83,7 @@ class EventBridgePutEventsOperator(BaseOperator):
             return [e["EventId"] for e in response["Entries"]]
 
 
-class EventBridgePutRuleOperator(BaseOperator):
+class EventBridgePutRuleOperator(AwsBaseOperator[EventBridgeHook]):
     """
     Create or update a specified EventBridge rule.
 
@@ -106,12 +99,20 @@ class EventBridgePutRuleOperator(BaseOperator):
     :param schedule_expression: the scheduling expression (for example, a cron or rate expression)
     :param state: indicates whether rule is set to be "ENABLED" or "DISABLED"
     :param tags: list of key-value pairs to associate with the rule
-    :param region: the region where rule is to be created or updated
-
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
+        running Airflow in a distributed manner and aws_conn_id is None or
+        empty, then default boto3 configuration would be used (and must be
+        maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.htmlt
     """
 
-    template_fields: Sequence[str] = (
-        "aws_conn_id",
+    aws_hook_class = EventBridgeHook
+    template_fields: Sequence[str] = aws_template_fields(
         "name",
         "description",
         "event_bus_name",
@@ -120,7 +121,6 @@ class EventBridgePutRuleOperator(BaseOperator):
         "schedule_expression",
         "state",
         "tags",
-        "region_name",
     )
 
     def __init__(
@@ -134,8 +134,6 @@ class EventBridgePutRuleOperator(BaseOperator):
         schedule_expression: str | None = None,
         state: str | None = None,
         tags: list | None = None,
-        region_name: str | None = None,
-        aws_conn_id: str = "aws_default",
         **kwargs,
     ):
         super().__init__(**kwargs)
@@ -144,16 +142,9 @@ class EventBridgePutRuleOperator(BaseOperator):
         self.event_bus_name = event_bus_name
         self.event_pattern = event_pattern
         self.role_arn = role_arn
-        self.region_name = region_name
         self.schedule_expression = schedule_expression
         self.state = state
         self.tags = tags
-        self.aws_conn_id = aws_conn_id
-
-    @cached_property
-    def hook(self) -> EventBridgeHook:
-        """Create and return an EventBridgeHook."""
-        return EventBridgeHook(aws_conn_id=self.aws_conn_id, region_name=self.region_name)
 
     def execute(self, context: Context):
         self.log.info('Sending rule "%s" to EventBridge.', self.name)
@@ -170,7 +161,7 @@ class EventBridgePutRuleOperator(BaseOperator):
         )
 
 
-class EventBridgeEnableRuleOperator(BaseOperator):
+class EventBridgeEnableRuleOperator(AwsBaseOperator[EventBridgeHook]):
     """
     Enable an EventBridge Rule.
 
@@ -180,32 +171,25 @@ class EventBridgeEnableRuleOperator(BaseOperator):
 
     :param name: the name of the rule to enable
     :param event_bus_name: the name or ARN of the event bus associated with the rule (default if omitted)
-    :param aws_conn_id: the AWS connection to use
-    :param region_name: the region of the rule to be enabled
-
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
+        running Airflow in a distributed manner and aws_conn_id is None or
+        empty, then default boto3 configuration would be used (and must be
+        maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.htmlt
     """
 
-    template_fields: Sequence[str] = ("name", "event_bus_name", "region_name", "aws_conn_id")
+    aws_hook_class = EventBridgeHook
+    template_fields: Sequence[str] = aws_template_fields("name", "event_bus_name")
 
-    def __init__(
-        self,
-        *,
-        name: str,
-        event_bus_name: str | None = None,
-        region_name: str | None = None,
-        aws_conn_id: str = "aws_default",
-        **kwargs,
-    ):
+    def __init__(self, *, name: str, event_bus_name: str | None = None, **kwargs):
         super().__init__(**kwargs)
         self.name = name
         self.event_bus_name = event_bus_name
-        self.region_name = region_name
-        self.aws_conn_id = aws_conn_id
-
-    @cached_property
-    def hook(self) -> EventBridgeHook:
-        """Create and return an EventBridgeHook."""
-        return EventBridgeHook(aws_conn_id=self.aws_conn_id, region_name=self.region_name)
 
     def execute(self, context: Context):
         self.hook.conn.enable_rule(
@@ -220,7 +204,7 @@ class EventBridgeEnableRuleOperator(BaseOperator):
         self.log.info('Enabled rule "%s"', self.name)
 
 
-class EventBridgeDisableRuleOperator(BaseOperator):
+class EventBridgeDisableRuleOperator(AwsBaseOperator[EventBridgeHook]):
     """
     Disable an EventBridge Rule.
 
@@ -230,32 +214,25 @@ class EventBridgeDisableRuleOperator(BaseOperator):
 
     :param name: the name of the rule to disable
     :param event_bus_name: the name or ARN of the event bus associated with the rule (default if omitted)
-    :param aws_conn_id: the AWS connection to use
-    :param region_name: the region of the rule to be disabled
-
+    :param aws_conn_id: The Airflow connection used for AWS credentials.
+        If this is ``None`` or empty then the default boto3 behaviour is used. If
+        running Airflow in a distributed manner and aws_conn_id is None or
+        empty, then default boto3 configuration would be used (and must be
+        maintained on each worker node).
+    :param region_name: AWS region_name. If not specified then the default boto3 behaviour is used.
+    :param verify: Whether or not to verify SSL certificates. See:
+        https://boto3.amazonaws.com/v1/documentation/api/latest/reference/core/session.html
+    :param botocore_config: Configuration dictionary (key-values) for botocore client. See:
+        https://botocore.amazonaws.com/v1/documentation/api/latest/reference/config.htmlt
     """
 
-    template_fields: Sequence[str] = ("name", "event_bus_name", "region_name", "aws_conn_id")
+    aws_hook_class = EventBridgeHook
+    template_fields: Sequence[str] = aws_template_fields("name", "event_bus_name")
 
-    def __init__(
-        self,
-        *,
-        name: str,
-        event_bus_name: str | None = None,
-        region_name: str | None = None,
-        aws_conn_id: str = "aws_default",
-        **kwargs,
-    ):
+    def __init__(self, *, name: str, event_bus_name: str | None = None, **kwargs):
         super().__init__(**kwargs)
         self.name = name
         self.event_bus_name = event_bus_name
-        self.region_name = region_name
-        self.aws_conn_id = aws_conn_id
-
-    @cached_property
-    def hook(self) -> EventBridgeHook:
-        """Create and return an EventBridgeHook."""
-        return EventBridgeHook(aws_conn_id=self.aws_conn_id, region_name=self.region_name)
 
     def execute(self, context: Context):
         self.hook.conn.disable_rule(