Question

table = set([])

class GlobeLearningTable(object):
    def __init__(self,mac,port,dpid):

        self.mac = mac
        self.port = port
        self.dpid = dpid

    def add(self):

        global table
        if self not in table:
            table.add(self)

class LearningSwitch(object):
    def __init__ (self, connection, transparent):
       self.connection = connection
       self.transparent = transparent
       self.macToPort = {}
       connection.addListeners(self)
       self.hold_down_expired = _flood_delay == 0

    def _handle_PacketIn (self, event):
       packet = event.parsed
       self.macToPort[packet.src] = event.port # 1
       packet_src = str(packet.src)
       packet_mac = packet_src.upper()
       entry = GlobeLearningTable(packet_mac, event.port, dpid_to_str(self.connection.dpid))
       entry.add()

问题：entry.add()方法每次调用时都会添加新对象，并递增表中的项目。

这不应该发生，因为

在add方法中，我正在检查表中是否有该对象，然后我添加了该特定对象。
表是无序列表的集合，不应该有重复的对象。

帮助：在这个设置中是否有任何方法我只能在不在表中时添加对象。

Answer 1

您需要实施__eq__和__hash__方法来教授Python如何识别唯一的GlobeLearningTable实例。

class GlobeLearningTable(object):
    def __init__(self,mac,port,dpid):
        self.mac = mac
        self.port = port
        self.dpid = dpid

    def __hash__(self):
        return hash((self.mac, self.port, self.dpid))

    def __eq__(self, other):
        if not isinstance(other, type(self)): return NotImplemented
        return self.mac == other.mac and self.port == other.port and self.dpid == other.dpid

现在您的对象具有可比性，并且相等的对象也将返回__hash__的相等值。这样，set和dict对象可以有效地存储您的对象，并检测它是否已存在：

>>> demo = set([GlobeLearningTable('a', 10, 'b')])
>>> GlobeLearningTable('a', 10, 'b') in demo
True

在python中创建用户定义类的对象集

1 个答案: