X-Git-Url: https://git.mxchange.org/?a=blobdiff_plain;f=node.py;h=35dadc6eebc52ecff3d11a03616557f33ca4be17;hb=d1507ff7d3cca77f48dffe460097f6ab3cd5d567;hp=22a4dd9b12b69acfe936021bcfb9b459e309f420;hpb=bb75d0a628c92d0e5602fa6be1ef9a37c3fb0d86;p=quix0rs-apt-p2p.git diff --git a/node.py b/node.py index 22a4dd9..35dadc6 100644 --- a/node.py +++ b/node.py @@ -1,79 +1,82 @@ -import hash -import time -from types import * -from xmlrpclib import Binary +## Copyright 2002-2003 Andrew Loewenstern, All Rights Reserved +# see LICENSE.txt for license information + +from time import time +from types import InstanceType + +import khash class Node: - """encapsulate contact info""" - def __init__(self): - self.fails = 0 - self.lastSeen = 0 - self.id = self.host = self.port = '' - - def init(self, id, host, port): - self.id = id - self.num = hash.intify(id) - self.host = host - self.port = port - self._senderDict = {'id': self.id.encode('base64'), 'port' : self.port, 'host' : self.host} - return self - - def initWithDict(self, dict): - self._senderDict = dict - self.id = dict['id'].decode('base64') - self.num = hash.intify(self.id) - self.port = dict['port'] - self.host = dict['host'] - return self - - def updateLastSeen(self): - self.lastSeen = time.time() - self.fails = 0 - - def msgFailed(self): - self.fails = self.fails + 1 - return self.fails - - def senderDict(self): - return self._senderDict - - def __repr__(self): - return `(self.id, self.host, self.port)` - + """encapsulate contact info""" + def __init__(self): + self.fails = 0 + self.lastSeen = 0 + self.id = self.host = self.port = '' + + def init(self, id, host, port): + self.id = id + self.num = khash.intify(id) + self.host = host + self.port = port + self._senderDict = {'id': self.id, 'port' : self.port, 'host' : self.host} + return self + + def initWithDict(self, dict): + self._senderDict = dict + self.id = dict['id'] + self.num = khash.intify(self.id) + self.port = dict['port'] + self.host = dict['host'] + return self + + def updateLastSeen(self): + self.lastSeen = time() + self.fails = 0 + + def msgFailed(self): + self.fails = self.fails + 1 + return self.fails + + def senderDict(self): + return self._senderDict + + def __repr__(self): + return `(self.id, self.host, self.port)` + ## these comparators let us bisect/index a list full of nodes with either a node or an int/long - def __lt__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num < a - def __le__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num <= a - def __gt__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num > a - def __ge__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num >= a - def __eq__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num == a - def __ne__(self, a): - if type(a) == InstanceType: - a = a.num - return self.num != a + def __lt__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num < a + def __le__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num <= a + def __gt__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num > a + def __ge__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num >= a + def __eq__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num == a + def __ne__(self, a): + if type(a) == InstanceType: + a = a.num + return self.num != a import unittest class TestNode(unittest.TestCase): - def setUp(self): - self.node = Node().init(hash.newID(), 'localhost', 2002) - def testUpdateLastSeen(self): - t = self.node.lastSeen - self.node.updateLastSeen() - assert t < self.node.lastSeen - \ No newline at end of file + def setUp(self): + self.node = Node().init(khash.newID(), 'localhost', 2002) + def testUpdateLastSeen(self): + t = self.node.lastSeen + self.node.updateLastSeen() + assert t < self.node.lastSeen + \ No newline at end of file