knode.py

   1 from node import Node
   2 from twisted.internet.defer import Deferred
   3 from const import reactor, NULL_ID
   4
   5
   6 class IDChecker:
   7     def __init__(id):
   8         self.id = id
   9
  10 class KNode(Node):
  11     def checkSender(self, dict):
  12         try:
  13             senderid = dict['rsp']['sender']['id']
  14         except KeyError:
  15             print ">>>> No peer id in response"
  16             raise Exception, "No peer id in response."
  17         else:
  18             if self.id != NULL_ID and senderid != self.id:
  19                 print "Got response from different node than expected."
  20                 raise Exception, "Got response from different node than expected."
  21         return dict
  22
  23     def ping(self, sender):
  24         df = self.conn.protocol.sendRequest('ping', {"sender":sender})
  25         df.addCallback(self.checkSender)
  26         return df
  27     def findNode(self, target, sender):
  28         df = self.conn.protocol.sendRequest('find_node', {"target" : target, "sender": sender})
  29         df.addCallback(self.checkSender)
  30         return df
  31     def storeValue(self, key, value, sender):
  32         df = self.conn.protocol.sendRequest('store_value', {"key" : key, "value" : value, "sender": sender})
  33         df.addCallback(self.checkSender)
  34         return df
  35     def findValue(self, key, sender):
  36         df =  self.conn.protocol.sendRequest('find_value', {"key" : key, "sender" : sender})
  37         df.addCallback(self.checkSender)
  38         return df