knode.py

   1 ## Copyright 2002-2003 Andrew Loewenstern, All Rights Reserved
   2 # see LICENSE.txt for license information
   3
   4 from node import Node
   5 from twisted.internet.defer import Deferred
   6 from const import reactor, NULL_ID
   7
   8
   9 class IDChecker:
  10     def __init__(id):
  11         self.id = id
  12
  13 class KNode(Node):
  14     def checkSender(self, dict):
  15         try:
  16             senderid = dict['rsp']['id']
  17         except KeyError:
  18             print ">>>> No peer id in response"
  19             raise Exception, "No peer id in response."
  20         else:
  21             if self.id != NULL_ID and senderid != self.id:
  22                 print "Got response from different node than expected."
  23                 raise Exception, "Got response from different node than expected."
  24         return dict
  25
  26     def errBack(self, err):
  27         print ">>> ", err
  28         return err
  29
  30     def ping(self, id):
  31         df = self.conn.sendRequest('ping', {"id":id})
  32         df.addErrback(self.errBack)
  33         df.addCallback(self.checkSender)
  34         return df
  35     def findNode(self, target, id):
  36         df = self.conn.sendRequest('find_node', {"target" : target, "id": id})
  37         df.addErrback(self.errBack)
  38         df.addCallback(self.checkSender)
  39         return df
  40     def storeValue(self, key, value, id):
  41         df = self.conn.sendRequest('store_value', {"key" : key, "value" : value, "id": id})
  42         df.addErrback(self.errBack)
  43         df.addCallback(self.checkSender)
  44         return df
  45     def findValue(self, key, id):
  46         df =  self.conn.sendRequest('find_value', {"key" : key, "id" : id})
  47         df.addErrback(self.errBack)
  48         df.addCallback(self.checkSender)
  49         return df