Coverage for /pythoncovmergedfiles/medio/medio/src/airflow/airflow/utils/weight_rule.py: 88%

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 

19 

20from enum import Enum 

21 

22import methodtools 

23 

24 

25class WeightRule(str, Enum): 

26 """Weight rules.""" 

27 

28 DOWNSTREAM = "downstream" 

29 UPSTREAM = "upstream" 

30 ABSOLUTE = "absolute" 

31 

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() 

36 

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()) 

42 

43 def __str__(self) -> str: 

44 return self.value