# 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# regarding copyright ownership. The ASF licenses this file# to you under the Apache License, Version 2.0 (the# "License"); you may not use this file except in compliance# with the License. You may obtain a copy of the License at## http://www.apache.org/licenses/LICENSE-2.0## Unless required by applicable law or agreed to in writing,# software 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.from__future__importannotationsfromtypingimportAny,Callablefromairflow.decorators.baseimportTaskDecoratorfromairflow.decorators.branch_pythonimportbranch_taskfromairflow.decorators.external_pythonimportexternal_python_taskfromairflow.decorators.pythonimportpython_taskfromairflow.decorators.python_virtualenvimportvirtualenv_taskfromairflow.decorators.sensorimportsensor_taskfromairflow.decorators.setup_teardownimportsetup_task,teardown_taskfromairflow.decorators.short_circuitimportshort_circuit_taskfromairflow.decorators.task_groupimporttask_groupfromairflow.models.dagimportdagfromairflow.providers_managerimportProvidersManager# Please keep this in sync with the .pyi's __all__.__all__=["TaskDecorator","TaskDecoratorCollection","dag","task","task_group","python_task","virtualenv_task","external_python_task","branch_task","short_circuit_task","sensor_task","setup","teardown",]
[docs]classTaskDecoratorCollection:"""Implementation to provide the ``@task`` syntax."""
[docs]__call__:Any=python# Alias '@task' to '@task.python'.
[docs]def__getattr__(self,name:str)->TaskDecorator:"""Dynamically get provider-registered task decorators, e.g. ``@task.docker``."""ifname.startswith("__"):raiseAttributeError(f"{type(self).__name__} has no attribute {name!r}")decorators=ProvidersManager().taskflow_decoratorsifnamenotindecorators:raiseAttributeError(f"task decorator {name!r} not found")returndecorators[name]