apache-airflow-providers-standard 0.1.0rc1__py3-none-any.whl → 1.0.0.dev1__py3-none-any.whl

airflow/providers/standard/LICENSE

@@ -199,3 +199,55 @@ distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 See the License for the specific language governing permissions and
 limitations under the License.
+
+============================================================================
+   APACHE AIRFLOW SUBCOMPONENTS:
+
+   The Apache Airflow project contains subcomponents with separate copyright
+   notices and license terms. Your use of the source code for the these
+   subcomponents is subject to the terms and conditions of the following
+   licenses.
+
+
+========================================================================
+Third party Apache 2.0 licenses
+========================================================================
+
+The following components are provided under the Apache 2.0 License.
+See project link for details. The text of each license is also included
+at 3rd-party-licenses/LICENSE-[project].txt.
+
+    (ALv2 License) hue v4.3.0 (https://github.com/cloudera/hue/)
+    (ALv2 License) jqclock v2.3.0 (https://github.com/JohnRDOrazio/jQuery-Clock-Plugin)
+    (ALv2 License) bootstrap3-typeahead v4.0.2 (https://github.com/bassjobsen/Bootstrap-3-Typeahead)
+    (ALv2 License) connexion v2.7.0 (https://github.com/zalando/connexion)
+
+========================================================================
+MIT licenses
+========================================================================
+
+The following components are provided under the MIT License. See project link for details.
+The text of each license is also included at 3rd-party-licenses/LICENSE-[project].txt.
+
+    (MIT License) jquery v3.5.1 (https://jquery.org/license/)
+    (MIT License) dagre-d3 v0.6.4 (https://github.com/cpettitt/dagre-d3)
+    (MIT License) bootstrap v3.4.1 (https://github.com/twbs/bootstrap/)
+    (MIT License) d3-tip v0.9.1 (https://github.com/Caged/d3-tip)
+    (MIT License) dataTables v1.10.25 (https://datatables.net)
+    (MIT License) normalize.css v3.0.2 (http://necolas.github.io/normalize.css/)
+    (MIT License) ElasticMock v1.3.2 (https://github.com/vrcmarcos/elasticmock)
+    (MIT License) MomentJS v2.24.0 (http://momentjs.com/)
+    (MIT License) eonasdan-bootstrap-datetimepicker v4.17.49 (https://github.com/eonasdan/bootstrap-datetimepicker/)
+
+========================================================================
+BSD 3-Clause licenses
+========================================================================
+The following components are provided under the BSD 3-Clause license. See project links for details.
+The text of each license is also included at 3rd-party-licenses/LICENSE-[project].txt.
+
+    (BSD 3 License) d3 v5.16.0 (https://d3js.org)
+    (BSD 3 License) d3-shape v2.1.0 (https://github.com/d3/d3-shape)
+    (BSD 3 License) cgroupspy 0.2.1 (https://github.com/cloudsigma/cgroupspy)
+
+========================================================================
+See 3rd-party-licenses/LICENSES-ui.txt for packages used in `/airflow/www`

airflow/providers/standard/__init__.py

@@ -1,3 +1,4 @@
+#
 # Licensed to the Apache Software Foundation (ASF) under one
 # or more contributor license agreements.  See the NOTICE file
 # distributed with this work for additional information
@@ -14,26 +15,3 @@
 # KIND, either express or implied.  See the License for the
 # specific language governing permissions and limitations
 # under the License.
-#
-# NOTE! THIS FILE IS AUTOMATICALLY GENERATED AND WILL BE
-# OVERWRITTEN WHEN PREPARING DOCUMENTATION FOR THE PACKAGES.
-#
-# IF YOU WANT TO MODIFY THIS FILE, YOU SHOULD MODIFY THE TEMPLATE
-# `PROVIDER__INIT__PY_TEMPLATE.py.jinja2` IN the `dev/breeze/src/airflow_breeze/templates` DIRECTORY
-#
-from __future__ import annotations
-
-import packaging.version
-
-from airflow import __version__ as airflow_version
-
-__all__ = ["__version__"]
-
-__version__ = "0.1.0"
-
-if packaging.version.parse(packaging.version.parse(airflow_version).base_version) < packaging.version.parse(
-    "2.9.0"
-):
-    raise RuntimeError(
-        f"The package `apache-airflow-providers-standard:{__version__}` needs Apache Airflow 2.9.0+"
-    )

airflow/providers/standard/get_provider_info.py

@@ -15,7 +15,8 @@
 # specific language governing permissions and limitations
 # under the License.
 
-# NOTE! THIS FILE IS AUTOMATICALLY GENERATED AND WILL BE OVERWRITTEN!
+# NOTE! THIS FILE IS AUTOMATICALLY GENERATED AND WILL BE
+# OVERWRITTEN WHEN PREPARING PACKAGES.
 #
 # IF YOU WANT TO MODIFY THIS FILE, YOU SHOULD MODIFY THE TEMPLATE
 # `get_provider_info_TEMPLATE.py.jinja2` IN the `dev/breeze/src/airflow_breeze/templates` DIRECTORY
@@ -26,19 +27,16 @@ def get_provider_info():
         "package-name": "apache-airflow-providers-standard",
         "name": "Standard",
         "description": "Airflow Standard Provider\n",
-        "state": "ready",
-        "source-date-epoch": 1739964539,
-        "versions": ["0.1.0", "0.0.3", "0.0.2", "0.0.1"],
+        "state": "not-ready",
+        "source-date-epoch": 1718603992,
+        "versions": ["1.0.0"],
+        "dependencies": ["apache-airflow>=2.10.0"],
         "integrations": [
             {
                 "integration-name": "Standard",
                 "external-doc-url": "https://airflow.apache.org/",
                 "tags": ["apache"],
-                "how-to-guide": [
-                    "/docs/apache-airflow-providers-standard/operators/bash.rst",
-                    "/docs/apache-airflow-providers-standard/operators/python.rst",
-                    "/docs/apache-airflow-providers-standard/operators/datetime.rst",
-                ],
+                "how-to-guide": ["/docs/apache-airflow-providers-standard/operators.rst"],
             }
         ],
         "operators": [
@@ -48,11 +46,6 @@ def get_provider_info():
                     "airflow.providers.standard.operators.datetime",
                     "airflow.providers.standard.operators.weekday",
                     "airflow.providers.standard.operators.bash",
-                    "airflow.providers.standard.operators.python",
-                    "airflow.providers.standard.operators.empty",
-                    "airflow.providers.standard.operators.generic_transfer",
-                    "airflow.providers.standard.operators.trigger_dagrun",
-                    "airflow.providers.standard.operators.latest_only",
                 ],
             }
         ],
@@ -65,45 +58,7 @@ def get_provider_info():
                     "airflow.providers.standard.sensors.time",
                     "airflow.providers.standard.sensors.weekday",
                     "airflow.providers.standard.sensors.bash",
-                    "airflow.providers.standard.sensors.python",
-                    "airflow.providers.standard.sensors.filesystem",
-                    "airflow.providers.standard.sensors.external_task",
-                ],
-            }
-        ],
-        "hooks": [
-            {
-                "integration-name": "Standard",
-                "python-modules": [
-                    "airflow.providers.standard.hooks.filesystem",
-                    "airflow.providers.standard.hooks.package_index",
-                    "airflow.providers.standard.hooks.subprocess",
                 ],
             }
         ],
-        "triggers": [
-            {
-                "integration-name": "Standard",
-                "python-modules": [
-                    "airflow.providers.standard.triggers.external_task",
-                    "airflow.providers.standard.triggers.file",
-                    "airflow.providers.standard.triggers.temporal",
-                ],
-            }
-        ],
-        "config": {
-            "standard": {
-                "description": "Options for the standard provider operators.",
-                "options": {
-                    "venv_install_method": {
-                        "description": "Which python tooling should be used to install the virtual environment.\n\nThe following options are available:\n- ``auto``: Automatically select, use ``uv`` if available, otherwise use ``pip``.\n- ``pip``: Use pip to install the virtual environment.\n- ``uv``: Use uv to install the virtual environment. Must be available in environment PATH.\n",
-                        "version_added": None,
-                        "type": "string",
-                        "example": "uv",
-                        "default": "auto",
-                    }
-                },
-            }
-        },
-        "dependencies": ["apache-airflow>=2.9.0", "apache-airflow-providers-common-sql>=1.20.0"],
     }

airflow/providers/standard/operators/bash.py

@@ -19,26 +19,19 @@ from __future__ import annotations
 
 import os
 import shutil
-import tempfile
-from collections.abc import Container, Sequence
+import warnings
 from functools import cached_property
-from typing import TYPE_CHECKING, Any, Callable, cast
+from typing import TYPE_CHECKING, Any, Callable, Container, Sequence, cast
 
 from airflow.exceptions import AirflowException, AirflowSkipException
+from airflow.hooks.subprocess import SubprocessHook
 from airflow.models.baseoperator import BaseOperator
-from airflow.providers.standard.hooks.subprocess import SubprocessHook, SubprocessResult, working_directory
 from airflow.utils.operator_helpers import context_to_airflow_vars
-from airflow.utils.session import NEW_SESSION, provide_session
 from airflow.utils.types import ArgNotSet
 
 if TYPE_CHECKING:
-    from sqlalchemy.orm import Session as SASession
-
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.models.taskinstance import TaskInstance
+    from airflow.utils.context import Context
 
 
 class BashOperator(BaseOperator):
@@ -69,10 +62,7 @@ class BashOperator(BaseOperator):
     :param cwd: Working directory to execute the command in (templated).
         If None (default), the command is run in a temporary directory.
         To use current DAG folder as the working directory,
-        you might set template ``{{ task.dag.folder }}``.
-        When bash_command is a '.sh' or '.bash' file, Airflow must have write
-        access to the working directory. The script will be rendered (Jinja
-        template) into a new temporary file in this directory.
+        you might set template ``{{ dag_run.dag.folder }}``.
     :param output_processor: Function to further process the output of the bash script
         (default is lambda output: output).
 
@@ -103,18 +93,14 @@ class BashOperator(BaseOperator):
 
         .. code-block:: python
 
-            bash_command = "set -e; python3 script.py '{{ data_interval_end }}'"
+            bash_command = "set -e; python3 script.py '{{ next_execution_date }}'"
 
     .. note::
 
-        To simply execute a ``.sh`` or ``.bash`` script (without any Jinja template), add a space after the
-        script name ``bash_command`` argument -- for example ``bash_command="my_script.sh "``. This
-        is because Airflow tries to load this file and process it as a Jinja template when
-        it ends with ``.sh`` or ``.bash``.
-
-        If you have Jinja template in your script, do not put any blank space. And add the script's directory
-        in the DAG's ``template_searchpath``. If you specify a ``cwd``, Airflow must have write access to
-        this directory. The script will be rendered (Jinja template) into a new temporary file in this directory.
+        Add a space after the script name when directly calling a ``.sh`` script with the
+        ``bash_command`` argument -- for example ``bash_command="my_script.sh "``.  This
+        is because Airflow tries to apply load this file and process it as a Jinja template to
+        it ends with ``.sh``, which will likely not be what most users want.
 
     .. warning::
 
@@ -163,6 +149,7 @@ class BashOperator(BaseOperator):
         env: dict[str, str] | None = None,
         append_env: bool = False,
         output_encoding: str = "utf-8",
+        skip_exit_code: int | None = None,
         skip_on_exit_code: int | Container[int] | None = 99,
         cwd: str | None = None,
         output_processor: Callable[[str], Any] = lambda result: result,
@@ -172,6 +159,11 @@ class BashOperator(BaseOperator):
         self.bash_command = bash_command
         self.env = env
         self.output_encoding = output_encoding
+        if skip_exit_code is not None:
+            warnings.warn(
+                "skip_exit_code is deprecated. Please use skip_on_exit_code", DeprecationWarning, stacklevel=2
+            )
+            skip_on_exit_code = skip_exit_code
         self.skip_on_exit_code = (
             skip_on_exit_code
             if isinstance(skip_on_exit_code, Container)
@@ -188,20 +180,13 @@ class BashOperator(BaseOperator):
         # determine whether the bash_command value needs to re-rendered.
         self._init_bash_command_not_set = isinstance(self.bash_command, ArgNotSet)
 
-        # Keep a copy of the original bash_command, without the Jinja template rendered.
-        # This is later used to determine if the bash_command is a script or an inline string command.
-        # We do this later, because the bash_command is not available in __init__ when using @task.bash.
-        self._unrendered_bash_command: str | ArgNotSet = bash_command
-
     @cached_property
     def subprocess_hook(self):
         """Returns hook for running the bash command."""
         return SubprocessHook()
 
-    # TODO: This should be replaced with Task SDK API call
     @staticmethod
-    @provide_session
-    def refresh_bash_command(ti, session: SASession = NEW_SESSION) -> None:
+    def refresh_bash_command(ti: TaskInstance) -> None:
         """
         Rewrite the underlying rendered bash_command value for a task instance in the metadatabase.
 
@@ -213,13 +198,9 @@ class BashOperator(BaseOperator):
         """
         from airflow.models.renderedtifields import RenderedTaskInstanceFields
 
-        """Update rendered task instance fields for cases where runtime evaluated, not templated."""
-
-        rtif = RenderedTaskInstanceFields(ti)
-        RenderedTaskInstanceFields.write(rtif, session=session)
-        RenderedTaskInstanceFields.delete_old_records(ti.task_id, ti.dag_id, session=session)
+        RenderedTaskInstanceFields._update_runtime_evaluated_template_fields(ti)
 
-    def get_env(self, context) -> dict:
+    def get_env(self, context):
         """Build the set of environment variables to be exposed for the bash command."""
         system_env = os.environ.copy()
         env = self.env
@@ -239,7 +220,7 @@ class BashOperator(BaseOperator):
         return env
 
     def execute(self, context: Context):
-        bash_path: str = shutil.which("bash") or "bash"
+        bash_path = shutil.which("bash") or "bash"
         if self.cwd is not None:
             if not os.path.exists(self.cwd):
                 raise AirflowException(f"Can not find the cwd: {self.cwd}")
@@ -253,17 +234,15 @@ class BashOperator(BaseOperator):
         # Both will ensure the correct Bash command is executed and that the Rendered Template view in the UI
         # displays the executed command (otherwise it will display as an ArgNotSet type).
         if self._init_bash_command_not_set:
-            is_inline_command = self._is_inline_command(bash_command=cast(str, self.bash_command))
-            ti = context["ti"]
+            ti = cast("TaskInstance", context["ti"])
             self.refresh_bash_command(ti)
-        else:
-            is_inline_command = self._is_inline_command(bash_command=cast(str, self._unrendered_bash_command))
-
-        if is_inline_command:
-            result = self._run_inline_command(bash_path=bash_path, env=env)
-        else:
-            result = self._run_rendered_script_file(bash_path=bash_path, env=env)
 
+        result = self.subprocess_hook.run_command(
+            command=[bash_path, "-c", self.bash_command],
+            env=env,
+            output_encoding=self.output_encoding,
+            cwd=self.cwd,
+        )
         if result.exit_code in self.skip_on_exit_code:
             raise AirflowSkipException(f"Bash command returned exit code {result.exit_code}. Skipping.")
         elif result.exit_code != 0:
@@ -273,38 +252,5 @@ class BashOperator(BaseOperator):
 
         return self.output_processor(result.output)
 
-    def _run_inline_command(self, bash_path: str, env: dict) -> SubprocessResult:
-        """Pass the bash command as string directly in the subprocess."""
-        return self.subprocess_hook.run_command(
-            command=[bash_path, "-c", self.bash_command],
-            env=env,
-            output_encoding=self.output_encoding,
-            cwd=self.cwd,
-        )
-
-    def _run_rendered_script_file(self, bash_path: str, env: dict) -> SubprocessResult:
-        """
-        Save the bash command into a file and execute this file.
-
-        This allows for longer commands, and prevents "Argument list too long error".
-        """
-        with working_directory(cwd=self.cwd) as cwd:
-            with tempfile.NamedTemporaryFile(mode="w", dir=cwd, suffix=".sh") as file:
-                file.write(cast(str, self.bash_command))
-                file.flush()
-
-                bash_script = os.path.basename(file.name)
-                return self.subprocess_hook.run_command(
-                    command=[bash_path, bash_script],
-                    env=env,
-                    output_encoding=self.output_encoding,
-                    cwd=cwd,
-                )
-
-    @classmethod
-    def _is_inline_command(cls, bash_command: str) -> bool:
-        """Return True if the bash command is an inline string. False if it's a bash script file."""
-        return not bash_command.endswith(tuple(cls.template_ext))
-
     def on_kill(self) -> None:
         self.subprocess_hook.send_sigterm()

airflow/providers/standard/operators/datetime.py

@@ -17,19 +17,14 @@
 from __future__ import annotations
 
 import datetime
-from collections.abc import Iterable
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Iterable
 
 from airflow.exceptions import AirflowException
 from airflow.operators.branch import BaseBranchOperator
 from airflow.utils import timezone
 
 if TYPE_CHECKING:
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.utils.context import Context
 
 
 class BranchDateTimeOperator(BaseBranchOperator):
@@ -49,7 +44,7 @@ class BranchDateTimeOperator(BaseBranchOperator):
     :param target_lower: target lower bound.
     :param target_upper: target upper bound.
     :param use_task_logical_date: If ``True``, uses task's logical date to compare with targets.
-        Logical date is useful for backfilling. If ``False``, uses system's date.
+        Execution date is useful for backfilling. If ``False``, uses system's date.
     """
 
     def __init__(

airflow/providers/standard/operators/weekday.py

@@ -17,19 +17,14 @@
 # under the License.
 from __future__ import annotations
 
-from collections.abc import Iterable
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Iterable
 
 from airflow.operators.branch import BaseBranchOperator
 from airflow.utils import timezone
 from airflow.utils.weekday import WeekDay
 
 if TYPE_CHECKING:
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.utils.context import Context
 
 
 class BranchDayOfWeekOperator(BaseBranchOperator):
@@ -43,8 +38,7 @@ class BranchDayOfWeekOperator(BaseBranchOperator):
 
     .. code-block:: python
 
-        from airflow.providers.standard.operators.empty import EmptyOperator
-        from airflow.operators.weekday import BranchDayOfWeekOperator
+        from airflow.operators.empty import EmptyOperator
 
         monday = EmptyOperator(task_id="monday")
         other_day = EmptyOperator(task_id="other_day")
@@ -64,8 +58,7 @@ class BranchDayOfWeekOperator(BaseBranchOperator):
 
         # import WeekDay Enum
         from airflow.utils.weekday import WeekDay
-        from airflow.providers.standard.operators.empty import EmptyOperator
-        from airflow.operators.weekday import BranchDayOfWeekOperator
+        from airflow.operators.empty import EmptyOperator
 
         workday = EmptyOperator(task_id="workday")
         weekend = EmptyOperator(task_id="weekend")

airflow/providers/standard/sensors/bash.py

@@ -18,20 +18,15 @@
 from __future__ import annotations
 
 import os
-from collections.abc import Sequence
 from subprocess import PIPE, STDOUT, Popen
 from tempfile import NamedTemporaryFile, TemporaryDirectory, gettempdir
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Sequence
 
 from airflow.exceptions import AirflowFailException
 from airflow.sensors.base import BaseSensorOperator
 
 if TYPE_CHECKING:
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.utils.context import Context
 
 
 class BashSensor(BaseSensorOperator):
@@ -75,10 +70,9 @@ class BashSensor(BaseSensorOperator):
         """Execute the bash command in a temporary directory."""
         bash_command = self.bash_command
         self.log.info("Tmp dir root location: %s", gettempdir())
-        with (
-            TemporaryDirectory(prefix="airflowtmp") as tmp_dir,
-            NamedTemporaryFile(dir=tmp_dir, prefix=self.task_id) as f,
-        ):
+        with TemporaryDirectory(prefix="airflowtmp") as tmp_dir, NamedTemporaryFile(
+            dir=tmp_dir, prefix=self.task_id
+        ) as f:
             f.write(bytes(bash_command, "utf_8"))
             f.flush()
             fname = f.name

airflow/providers/standard/sensors/date_time.py

@@ -18,37 +18,15 @@
 from __future__ import annotations
 
 import datetime
-from collections.abc import Sequence
-from dataclasses import dataclass
-from typing import TYPE_CHECKING, Any, NoReturn
+from typing import TYPE_CHECKING, Any, NoReturn, Sequence
 
-from airflow.providers.standard.triggers.temporal import DateTimeTrigger
-from airflow.providers.standard.version_compat import AIRFLOW_V_3_0_PLUS
 from airflow.sensors.base import BaseSensorOperator
-
-try:
-    from airflow.triggers.base import StartTriggerArgs
-except ImportError:
-    # TODO: Remove this when min airflow version is 2.10.0 for standard provider
-    @dataclass
-    class StartTriggerArgs:  # type: ignore[no-redef]
-        """Arguments required for start task execution from triggerer."""
-
-        trigger_cls: str
-        next_method: str
-        trigger_kwargs: dict[str, Any] | None = None
-        next_kwargs: dict[str, Any] | None = None
-        timeout: datetime.timedelta | None = None
-
-
+from airflow.triggers.base import StartTriggerArgs
+from airflow.triggers.temporal import DateTimeTrigger
 from airflow.utils import timezone
 
 if TYPE_CHECKING:
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.utils.context import Context
 
 
 class DateTimeSensor(BaseSensorOperator):
@@ -59,7 +37,7 @@ class DateTimeSensor(BaseSensorOperator):
     It handles some cases for which ``TimeSensor`` and ``TimeDeltaSensor`` are not suited.
 
     **Example** 1 :
-        If a task needs to wait for 11am on each ``logical_date``. Using
+        If a task needs to wait for 11am on each ``execution_date``. Using
         ``TimeSensor`` or ``TimeDeltaSensor``, all backfill tasks started at
         1am have to wait for 10 hours. This is unnecessary, e.g. a backfill
         task with ``{{ ds }} = '1970-01-01'`` does not need to wait because
@@ -74,7 +52,7 @@ class DateTimeSensor(BaseSensorOperator):
 
             DateTimeSensor(
                 task_id="wait_for_0100",
-                target_time="{{ data_interval_end.tomorrow().replace(hour=1) }}",
+                target_time="{{ next_execution_date.tomorrow().replace(hour=1) }}",
             )
 
     :param target_time: datetime after which the job succeeds. (templated)
@@ -115,7 +93,7 @@ class DateTimeSensorAsync(DateTimeSensor):
     """
 
     start_trigger_args = StartTriggerArgs(
-        trigger_cls="airflow.providers.standard.triggers.temporal.DateTimeTrigger",
+        trigger_cls="airflow.triggers.temporal.DateTimeTrigger",
         trigger_kwargs={"moment": "", "end_from_trigger": False},
         next_method="execute_complete",
         next_kwargs=None,
@@ -147,9 +125,7 @@ class DateTimeSensorAsync(DateTimeSensor):
             trigger=DateTimeTrigger(
                 moment=timezone.parse(self.target_time),
                 end_from_trigger=self.end_from_trigger,
-            )
-            if AIRFLOW_V_3_0_PLUS
-            else DateTimeTrigger(moment=timezone.parse(self.target_time)),
+            ),
         )
 
     def execute_complete(self, context: Context, event: Any = None) -> None:

airflow/providers/standard/sensors/time.py

@@ -18,36 +18,15 @@
 from __future__ import annotations
 
 import datetime
-from dataclasses import dataclass
 from typing import TYPE_CHECKING, Any, NoReturn
 
-from airflow.providers.standard.triggers.temporal import DateTimeTrigger
-from airflow.providers.standard.version_compat import AIRFLOW_V_2_10_PLUS
 from airflow.sensors.base import BaseSensorOperator
-
-try:
-    from airflow.triggers.base import StartTriggerArgs
-except ImportError:
-    # TODO: Remove this when min airflow version is 2.10.0 for standard provider
-    @dataclass
-    class StartTriggerArgs:  # type: ignore[no-redef]
-        """Arguments required for start task execution from triggerer."""
-
-        trigger_cls: str
-        next_method: str
-        trigger_kwargs: dict[str, Any] | None = None
-        next_kwargs: dict[str, Any] | None = None
-        timeout: datetime.timedelta | None = None
-
-
+from airflow.triggers.base import StartTriggerArgs
+from airflow.triggers.temporal import DateTimeTrigger
 from airflow.utils import timezone
 
 if TYPE_CHECKING:
-    try:
-        from airflow.sdk.definitions.context import Context
-    except ImportError:
-        # TODO: Remove once provider drops support for Airflow 2
-        from airflow.utils.context import Context
+    from airflow.utils.context import Context
 
 
 class TimeSensor(BaseSensorOperator):
@@ -89,7 +68,7 @@ class TimeSensorAsync(BaseSensorOperator):
     """
 
     start_trigger_args = StartTriggerArgs(
-        trigger_cls="airflow.providers.standard.triggers.temporal.DateTimeTrigger",
+        trigger_cls="airflow.triggers.temporal.DateTimeTrigger",
         trigger_kwargs={"moment": "", "end_from_trigger": False},
         next_method="execute_complete",
         next_kwargs=None,
@@ -123,9 +102,7 @@ class TimeSensorAsync(BaseSensorOperator):
 
     def execute(self, context: Context) -> NoReturn:
         self.defer(
-            trigger=DateTimeTrigger(moment=self.target_datetime, end_from_trigger=self.end_from_trigger)
-            if AIRFLOW_V_2_10_PLUS
-            else DateTimeTrigger(moment=self.target_datetime),
+            trigger=DateTimeTrigger(moment=self.target_datetime, end_from_trigger=self.end_from_trigger),
             method_name="execute_complete",
         )