1# Licensed to the Apache Software Foundation (ASF) under one
2# or more contributor license agreements. See the NOTICE file
3# distributed with this work for additional information
4# regarding copyright ownership. The ASF licenses this file
5# to you under the Apache License, Version 2.0 (the
6# "License"); you may not use this file except in compliance
7# with the License. You may obtain a copy of the License at
8#
9# http://www.apache.org/licenses/LICENSE-2.0
10#
11# Unless required by applicable law or agreed to in writing,
12# software distributed under the License is distributed on an
13# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
14# KIND, either express or implied. See the License for the
15# specific language governing permissions and limitations
16# under the License.
17from __future__ import annotations
18
19from airflow.utils.state import TaskInstanceState
20
21EXECUTION_STATES = {
22 TaskInstanceState.RUNNING,
23 TaskInstanceState.QUEUED,
24}
25
26# In order to be able to get queued a task must have one of these states
27SCHEDULEABLE_STATES = {
28 None,
29 TaskInstanceState.UP_FOR_RETRY,
30 TaskInstanceState.UP_FOR_RESCHEDULE,
31}
32
33RUNNABLE_STATES = {
34 # For cases like unit tests and run manually
35 None,
36 TaskInstanceState.UP_FOR_RETRY,
37 TaskInstanceState.UP_FOR_RESCHEDULE,
38 # For normal scheduler/backfill cases
39 TaskInstanceState.QUEUED,
40}
41
42QUEUEABLE_STATES = {
43 TaskInstanceState.SCHEDULED,
44}
45
46BACKFILL_QUEUEABLE_STATES = {
47 # For cases like unit tests and run manually
48 None,
49 TaskInstanceState.UP_FOR_RESCHEDULE,
50 TaskInstanceState.UP_FOR_RETRY,
51 # For normal backfill cases
52 TaskInstanceState.SCHEDULED,
53}