Coverage for src/metador_core/util/enum.py: 0%

1from __future__ import annotations 

2 

3import itertools 

4from enum import Enum, EnumMeta 

5from typing import Any, Dict, Optional, Set, Union 

6 

7 

8class NestedEnumMeta(EnumMeta): 

9 """Enum subclass for Enums generated from a semantic taxonomy. 

10 

11 Approach is to collect values and treat them as sets. 

12 """ 

13 

14 __self__: Optional[Union[int, str]] = None 

15 __values__: Set[Any] = set() 

16 __nested__: Dict[str, NestedEnumMeta] = {} 

17 

18 name: str 

19 value: Union[int, str] 

20 

21 def __call__(metacls, clsname, dct, *args, **kwargs): 

22 self = kwargs.pop("self", None) 

23 mod_dct = {} 

24 nested = {} 

25 for k, v in dct.items(): 

26 if isinstance(v, NestedEnumMeta): 

27 if v.__self__ is None: 

28 raise ValueError(f"Nested enum {v} must have a self value!") 

29 nested[k] = v 

30 else: 

31 mod_dct[k] = v 

32 

33 ret = super().__call__(clsname, mod_dct, *args, **kwargs) 

34 

35 ret.__self__ = self 

36 nested_values = set.union( 

37 set(), *map(lambda x: {x.__self__}.union(x.__values__), nested.values()) 

38 ) 

39 ret.__values__ = set(map(lambda x: x.value, iter(ret))).union(nested_values) 

40 ret.__nested__ = nested 

41 for name, nst in nested.items(): 

42 setattr(ret, name, nst) 

43 

44 return ret 

45 

46 def __contains__(self, other): 

47 print(other, "in", self, "?") 

48 if isinstance(other, NestedEnum): 

49 return self.__contains__(other.value) 

50 if isinstance(other, type) and issubclass(other.__class__, NestedEnumMeta): 

51 return self.__contains__(other.__self__) 

52 # lookup plain value 

53 return other in self.__values__ 

54 

55 def __iter__(self): 

56 return itertools.chain( 

57 super().__iter__(), 

58 *(itertools.chain(iter((x,)), iter(x)) for x in self.__nested__.values()), 

59 ) 

60 

61 def __dir__(self): 

62 return itertools.chain(super().__dir__(), self.__nested__.keys()) 

63 

64 def __repr__(self): 

65 if self.__self__ is not None: 

66 return f"<enum {self.__name__}: {self.__self__}>" 

67 else: 

68 return super().__repr__() 

69 

70 

71class NestedEnum(Enum, metaclass=NestedEnumMeta): 

72 """Base class for hierarchical enumerations."""