Clean-up functools.total_ordering().

This commit is contained in:
Raymond Hettinger 2010-09-05 05:57:35 +00:00
parent bb734c6707
commit 4c7c9af542
2 changed files with 3 additions and 5 deletions

View File

@ -65,7 +65,6 @@ def wraps(wrapped,
return partial(update_wrapper, wrapped=wrapped, return partial(update_wrapper, wrapped=wrapped,
assigned=assigned, updated=updated) assigned=assigned, updated=updated)
_object_defaults = {object.__lt__, object.__le__, object.__gt__, object.__ge__}
def total_ordering(cls): def total_ordering(cls):
"""Class decorator that fills in missing ordering methods""" """Class decorator that fills in missing ordering methods"""
convert = { convert = {
@ -82,9 +81,8 @@ def total_ordering(cls):
('__gt__', lambda self, other: not other >= self), ('__gt__', lambda self, other: not other >= self),
('__lt__', lambda self, other: not self >= other)] ('__lt__', lambda self, other: not self >= other)]
} }
roots = set(dir(cls)) & set(convert) # Find comparisons not inherited from object.
# Remove default comparison operations defined on object. roots = [op for op in convert if getattr(cls, op) is not getattr(object, op)]
roots -= {meth for meth in roots if getattr(cls, meth) in _object_defaults}
if not roots: if not roots:
raise ValueError('must define at least one ordering operation: < > <= >=') raise ValueError('must define at least one ordering operation: < > <= >=')
root = max(roots) # prefer __lt__ to __le__ to __gt__ to __ge__ root = max(roots) # prefer __lt__ to __le__ to __gt__ to __ge__

View File

@ -222,7 +222,7 @@ Library
- Issue #9501: Fixed logging regressions in cleanup code. - Issue #9501: Fixed logging regressions in cleanup code.
- Fix functools.total_ordering() to actually work. - Fix functools.total_ordering() to skip methods inherited from object().
- Issue #9572: Importlib should not raise an exception if a directory it - Issue #9572: Importlib should not raise an exception if a directory it
thought it needed to create was done concurrently by another process. thought it needed to create was done concurrently by another process.