aboutsummaryrefslogtreecommitdiffstats
path: root/contrib/tools/python3/Lib/enum.py
diff options
context:
space:
mode:
authorshadchin <shadchin@yandex-team.com>2024-04-28 21:17:44 +0300
committershadchin <shadchin@yandex-team.com>2024-04-28 21:25:54 +0300
commita55d99a3eb72f90355bc146baeda18aa7eb97352 (patch)
treeb17cfed786effe8b81bba022239d6729f716fbeb /contrib/tools/python3/Lib/enum.py
parent67bf49d08acf1277eff4c336021ac22d964bb4c4 (diff)
downloadydb-a55d99a3eb72f90355bc146baeda18aa7eb97352.tar.gz
Update Python 3 to 3.12.3
7d09de7d8b99ea2be554ef0fc61276942ca9c2e1
Diffstat (limited to 'contrib/tools/python3/Lib/enum.py')
-rw-r--r--contrib/tools/python3/Lib/enum.py82
1 files changed, 54 insertions, 28 deletions
diff --git a/contrib/tools/python3/Lib/enum.py b/contrib/tools/python3/Lib/enum.py
index 1502bfe9158..af5613838d1 100644
--- a/contrib/tools/python3/Lib/enum.py
+++ b/contrib/tools/python3/Lib/enum.py
@@ -166,6 +166,11 @@ def _dedent(text):
lines[j] = l[i:]
return '\n'.join(lines)
+class _not_given:
+ def __repr__(self):
+ return('<not given>')
+_not_given = _not_given()
+
class _auto_null:
def __repr__(self):
return '_auto_null'
@@ -283,9 +288,10 @@ class _proto_member:
enum_member._sort_order_ = len(enum_class._member_names_)
if Flag is not None and issubclass(enum_class, Flag):
- enum_class._flag_mask_ |= value
- if _is_single_bit(value):
- enum_class._singles_mask_ |= value
+ if isinstance(value, int):
+ enum_class._flag_mask_ |= value
+ if _is_single_bit(value):
+ enum_class._singles_mask_ |= value
enum_class._all_bits_ = 2 ** ((enum_class._flag_mask_).bit_length()) - 1
# If another member with the same value was already defined, the
@@ -313,6 +319,7 @@ class _proto_member:
elif (
Flag is not None
and issubclass(enum_class, Flag)
+ and isinstance(value, int)
and _is_single_bit(value)
):
# no other instances found, record this member in _member_names_
@@ -457,10 +464,11 @@ class _EnumDict(dict):
if isinstance(value, auto):
single = True
value = (value, )
- if type(value) is tuple and any(isinstance(v, auto) for v in value):
+ if isinstance(value, tuple) and any(isinstance(v, auto) for v in value):
# insist on an actual tuple, no subclasses, in keeping with only supporting
# top-level auto() usage (not contained in any other data structure)
auto_valued = []
+ t = type(value)
for v in value:
if isinstance(v, auto):
non_auto_store = False
@@ -475,7 +483,12 @@ class _EnumDict(dict):
if single:
value = auto_valued[0]
else:
- value = tuple(auto_valued)
+ try:
+ # accepts iterable as multiple arguments?
+ value = t(auto_valued)
+ except TypeError:
+ # then pass them in singlely
+ value = t(*auto_valued)
self._member_names[key] = None
if non_auto_store:
self._last_values.append(value)
@@ -710,7 +723,7 @@ class EnumType(type):
"""
return True
- def __call__(cls, value, names=None, *values, module=None, qualname=None, type=None, start=1, boundary=None):
+ def __call__(cls, value, names=_not_given, *values, module=None, qualname=None, type=None, start=1, boundary=None):
"""
Either returns an existing member, or creates a new enum class.
@@ -739,18 +752,18 @@ class EnumType(type):
"""
if cls._member_map_:
# simple value lookup if members exist
- if names:
+ if names is not _not_given:
value = (value, names) + values
return cls.__new__(cls, value)
# otherwise, functional API: we're creating a new Enum type
- if names is None and type is None:
+ if names is _not_given and type is None:
# no body? no data-type? possibly wrong usage
raise TypeError(
f"{cls} has no members; specify `names=()` if you meant to create a new, empty, enum"
)
return cls._create_(
class_name=value,
- names=names,
+ names=None if names is _not_given else names,
module=module,
qualname=qualname,
type=type,
@@ -1528,37 +1541,50 @@ class Flag(Enum, boundary=STRICT):
def __bool__(self):
return bool(self._value_)
+ def _get_value(self, flag):
+ if isinstance(flag, self.__class__):
+ return flag._value_
+ elif self._member_type_ is not object and isinstance(flag, self._member_type_):
+ return flag
+ return NotImplemented
+
def __or__(self, other):
- if isinstance(other, self.__class__):
- other = other._value_
- elif self._member_type_ is not object and isinstance(other, self._member_type_):
- other = other
- else:
+ other_value = self._get_value(other)
+ if other_value is NotImplemented:
return NotImplemented
+
+ for flag in self, other:
+ if self._get_value(flag) is None:
+ raise TypeError(f"'{flag}' cannot be combined with other flags with |")
value = self._value_
- return self.__class__(value | other)
+ return self.__class__(value | other_value)
def __and__(self, other):
- if isinstance(other, self.__class__):
- other = other._value_
- elif self._member_type_ is not object and isinstance(other, self._member_type_):
- other = other
- else:
+ other_value = self._get_value(other)
+ if other_value is NotImplemented:
return NotImplemented
+
+ for flag in self, other:
+ if self._get_value(flag) is None:
+ raise TypeError(f"'{flag}' cannot be combined with other flags with &")
value = self._value_
- return self.__class__(value & other)
+ return self.__class__(value & other_value)
def __xor__(self, other):
- if isinstance(other, self.__class__):
- other = other._value_
- elif self._member_type_ is not object and isinstance(other, self._member_type_):
- other = other
- else:
+ other_value = self._get_value(other)
+ if other_value is NotImplemented:
return NotImplemented
+
+ for flag in self, other:
+ if self._get_value(flag) is None:
+ raise TypeError(f"'{flag}' cannot be combined with other flags with ^")
value = self._value_
- return self.__class__(value ^ other)
+ return self.__class__(value ^ other_value)
def __invert__(self):
+ if self._get_value(self) is None:
+ raise TypeError(f"'{self}' cannot be inverted")
+
if self._inverted_ is None:
if self._boundary_ in (EJECT, KEEP):
self._inverted_ = self.__class__(~self._value_)
@@ -1625,7 +1651,7 @@ def global_flag_repr(self):
cls_name = self.__class__.__name__
if self._name_ is None:
return "%s.%s(%r)" % (module, cls_name, self._value_)
- if _is_single_bit(self):
+ if _is_single_bit(self._value_):
return '%s.%s' % (module, self._name_)
if self._boundary_ is not FlagBoundary.KEEP:
return '|'.join(['%s.%s' % (module, name) for name in self.name.split('|')])