knode.py

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