Создание набора объектов пользовательского класса в python

Вопрос:
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() добавляет новый объект при каждом вызове и увеличивает значения в таблице.

Это не должно произойти, потому что

  1. В методе добавления я проверяю, что это тот объект в таблице или нет, тогда я добавляю этот конкретный объект.
  2. Таблица представляет собой набор, который является неупорядоченным списком, который не должен иметь повторяющихся объектов.

Справка: есть ли какой-либо способ в этой настройке, я могу добавить объект только тогда, когда он не находится в таблице.

Лучший ответ:

Вам необходимо реализовать __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

Оцените статью
TechArks.Ru
Добавить комментарий