node.py

   1 import hash
   2 import time
   3 from types import *
   4 from xmlrpclib import Binary
   5
   6 class Node:
   7         """encapsulate contact info"""
   8         def __init__(self):
   9                 self.fails = 0
  10                 self.lastSeen = 0
  11                 self.id = self.host = self.port = ''
  12
  13         def init(self, id, host, port):
  14                 self.id = id
  15                 self.num = hash.intify(id)
  16                 self.host = host
  17                 self.port = port
  18                 self._senderDict = {'id': self.id.encode('base64'), 'port' : self.port, 'host' : self.host}
  19                 return self
  20
  21         def initWithDict(self, dict):
  22                 self._senderDict = dict
  23                 self.id = dict['id'].decode('base64')
  24                 self.num = hash.intify(self.id)
  25                 self.port = dict['port']
  26                 self.host = dict['host']
  27                 return self
  28
  29         def updateLastSeen(self):
  30                 self.lastSeen = time.time()
  31                 self.fails = 0
  32
  33         def msgFailed(self):
  34                 self.fails = self.fails + 1
  35                 return self.fails
  36
  37         def senderDict(self):
  38                 return self._senderDict
  39
  40         def __repr__(self):
  41                 return `(self.id, self.host, self.port)`
  42
  43     ## these comparators let us bisect/index a list full of nodes with either a node or an int/long
  44         def __lt__(self, a):
  45                 if type(a) == InstanceType:
  46                         a = a.num
  47                 return self.num < a
  48         def __le__(self, a):
  49                 if type(a) == InstanceType:
  50                         a = a.num
  51                 return self.num <= a
  52         def __gt__(self, a):
  53                 if type(a) == InstanceType:
  54                         a = a.num
  55                 return self.num > a
  56         def __ge__(self, a):
  57                 if type(a) == InstanceType:
  58                         a = a.num
  59                 return self.num >= a
  60         def __eq__(self, a):
  61                 if type(a) == InstanceType:
  62                         a = a.num
  63                 return self.num == a
  64         def __ne__(self, a):
  65                 if type(a) == InstanceType:
  66                         a = a.num
  67                 return self.num != a
  68
  69
  70 import unittest
  71
  72 class TestNode(unittest.TestCase):
  73         def setUp(self):
  74                 self.node = Node().init(hash.newID(), 'localhost', 2002)
  75         def testUpdateLastSeen(self):
  76                 t = self.node.lastSeen
  77                 self.node.updateLastSeen()
  78                 assert t < self.node.lastSeen
  79