Python Node类如何优雅复用基于value的比较方法？

在Python中，要避免重复定义所有比较魔术方法，最Pythonic的方式是使用标准库functools里的total_ordering装饰器。它能帮你自动推导完整的比较方法，只需要定义其中一组必要的基础方法即可。

具体实现步骤 #

1. 导入total_ordering装饰器
2. 给Node类添加该装饰器
3. 仅定义__lt__（小于）和__eq__（等于）两个方法，剩下的__le__、__gt__、__ge__、__ne__会被自动生成

完整示例代码 #

from functools import total_ordering

@total_ordering
class Node:
    def __init__(self, value, name=None, left=None, right=None):
        self.value = value
        self.name = name
        self.right, self.left = right, left
    
    def is_leaf(self):
        return self.right is None and self.left is None
    
    def __lt__(self, other):
        # 先判断类型，避免非Node实例引发错误
        if not isinstance(other, Node):
            return NotImplemented
        return self.value < other.value
    
    def __eq__(self, other):
        if not isinstance(other, Node):
            return NotImplemented
        return self.value == other.value

为什么这比继承value类型更好 #

你之前考虑的继承float或int这类基础类型的方案确实存在风险——会让Node类继承大量无关的方法（比如__add__、__sub__等数值操作），很可能和二叉树节点的逻辑冲突，引发意外行为。而total_ordering只会生成必要的比较方法，完全不会干扰原本的类结构。

验证效果 #

现在可以直接对Node实例列表排序，或者使用所有比较运算符：

nodes = [Node(5), Node(2), Node(8)]
nodes.sort()
# 排序后顺序为Node(2), Node(5), Node(8)

print(Node(3) <= Node(5))  # 输出 True
print(Node(7) > Node(4))   # 输出 True

内容的提问来源于stack exchange，提问作者Tarifazo