From 49076f24ad0725842acca87ba380fb972ed93094 Mon Sep 17 00:00:00 2001 From: seik Date: Tue, 6 Dec 2022 22:21:23 +0100 Subject: [PATCH] Add __eq__ for Node --- src/aiobtdht/routing_table/node.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/src/aiobtdht/routing_table/node.py b/src/aiobtdht/routing_table/node.py index dca7c87..0532b89 100644 --- a/src/aiobtdht/routing_table/node.py +++ b/src/aiobtdht/routing_table/node.py @@ -10,3 +10,9 @@ def id(self): @property def addr(self): return self._addr + + def __eq__(self, __o: object) -> bool: + if not isinstance(__o, Node): + return False + + return self.id == __o.id and self.addr == __o.addr