Python：继承内置类型

Question

我有一个关于内置类型的子类型及其构造函数的问题。 我想要一个类从元组和自定义类继承。

让我举一个具体的例子。 我使用图表工作很多，这意味着与边缘连接的节点。 我开始在自己的图形框架上做一些工作。

有一个类Edge，它有自己的属性和方法。 它还应该继承自GraphElement类。 （GraphElement是在特定图形的上下文之外没有任何意义的每个对象。）但在最基本的级别，边缘只是包含两个节点的元组。 如果您可以执行以下操作，那将是很好的语法糖：

edge = graph.create_edge("Spam","Eggs")
(u, v) = edge

所以（你，v）将包含“垃圾邮件”和“鸡蛋”。 它也支持迭代

for node in edge: ...

我希望你明白为什么我想要子类型元组（或其他基本类型，如集）。

所以这是我的Edge类及其init ：

class Edge(GraphElement, tuple):

def __init__(self, graph, (source, target)):
    GraphElement.__init__(self, graph)
    tuple.__init__((source, target))

我打电话的时候

Edge(aGraph, (source, target))

我得到一个TypeError：tuple（）最多需要1个参数（给定2个）。 我究竟做错了什么？

Answer 1

由于元组是不可变的，因此您还需要覆盖__new__方法。 见http://www.python.org/download/releases/2.2.3/descrintro/#__new__

class GraphElement:
    def __init__(self, graph):
        pass

class Edge(GraphElement, tuple):
    def __new__(cls, graph, (source, target)):
        return tuple.__new__(cls, (source, target))
    def __init__(self, graph, (source, target)):
        GraphElement.__init__(self, graph)

Answer 2

根据您的需要，我会避免多重继承，并使用生成器实现迭代器：

class GraphElement:
    def __init__(self, graph):
        pass

class Edge(GraphElement):
    def __init__(self, graph, (source, target)):
        GraphElement.__init__(self, graph)
        self.source = source
        self.target = target

    def __iter__(self):
        yield self.source
        yield self.target

在这种情况下，两种用法都可以正常工作：

e = Edge(None, ("Spam","Eggs"))
(s, t) = e
print s, t
for p in e:
    print p

Answer 3

你需要覆盖__new__ - 当前tuple.__new__被调用（因为你没有覆盖它）与你传递给Edge所有参数。