Coverage for /pythoncovmergedfiles/medio/medio/src/airflow/airflow/utils/weight_rule.py: 88%
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
Shortcuts on this page
r m x toggle line displays
j k next/prev highlighted chunk
0 (zero) top of page
1 (one) first highlighted chunk
1#
2# Licensed to the Apache Software Foundation (ASF) under one
3# or more contributor license agreements. See the NOTICE file
4# distributed with this work for additional information
5# regarding copyright ownership. The ASF licenses this file
6# to you under the Apache License, Version 2.0 (the
7# "License"); you may not use this file except in compliance
8# with the License. You may obtain a copy of the License at
9#
10# http://www.apache.org/licenses/LICENSE-2.0
11#
12# Unless required by applicable law or agreed to in writing,
13# software distributed under the License is distributed on an
14# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
15# KIND, either express or implied. See the License for the
16# specific language governing permissions and limitations
17# under the License.
18from __future__ import annotations
20from enum import Enum
22import methodtools
25class WeightRule(str, Enum):
26 """Weight rules."""
28 DOWNSTREAM = "downstream"
29 UPSTREAM = "upstream"
30 ABSOLUTE = "absolute"
32 @classmethod
33 def is_valid(cls, weight_rule: str) -> bool:
34 """Check if weight rule is valid."""
35 return weight_rule in cls.all_weight_rules()
37 @methodtools.lru_cache(maxsize=None)
38 @classmethod
39 def all_weight_rules(cls) -> set[str]:
40 """Return all weight rules."""
41 return set(cls.__members__.values())
43 def __str__(self) -> str:
44 return self.value