added boostrap_mode as API parameter in Pymdht #36

rauljim · Jan 11, 2012 · 2c45199 · 2c45199
1 parent 12f1764
commit 2c45199
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/core/controller.py b/core/controller.py
@@ -51,7 +51,8 @@ def __init__(self, version_label,
                  my_node, state_filename,
                  routing_m_mod, lookup_m_mod,
                  experimental_m_mod,
-                 private_dht_name):
+                 private_dht_name,
+                 bootstrap_mode):
 
         if size_estimation:
             self._size_estimation_file = open('size_estimation.dat', 'w')
@@ -73,7 +74,7 @@ def __init__(self, version_label,
             self._my_node, saved_bootstrap_nodes, self.msg_f)
 
         self._responder = responder.Responder(self._my_id, self._routing_m,
-                                              self.msg_f)
+                                              self.msg_f, boostrap_mode)
         self._tracker = self._responder._tracker
 
         self._lookup_m = lookup_m_mod.LookupManager(self._my_id, self.msg_f)

diff --git a/core/pymdht.py b/core/pymdht.py
@@ -45,15 +45,17 @@ def __init__(self, my_node, conf_path,
                  routing_m_mod, lookup_m_mod,
                  experimental_m_mod,
                  private_dht_name,
-                 debug_level, id_=None):
+                 debug_level, id_=None,
+                 bootsrap_mode=False):
         logging_conf.setup(conf_path, debug_level)
         state_filename = os.path.join(conf_path, controller.STATE_FILENAME)
         self.controller = controller.Controller(VERSION_LABEL,
                                                 my_node, state_filename,
                                                 routing_m_mod,
                                                 lookup_m_mod,
                                                 experimental_m_mod,
-                                                private_dht_name)
+                                                private_dht_name,
+                                                bootsrap_mode)
         self.reactor = minitwisted.ThreadedReactor(
             self.controller.main_loop,
             my_node.addr[1], self.controller.on_datagram_received)
@@ -66,7 +68,8 @@ def stop(self):
         # No need to call_asap because the minitwisted thread is dead by now
         self.controller.on_stop()
 
-    def get_peers(self, lookup_id, info_hash, callback_f, bt_port=0, use_cache=False):
+    def get_peers(self, lookup_id, info_hash, callback_f,
+                  bt_port=0, use_cache=False):
         """ Start a get peers lookup. Return a Lookup object.
         
         The info_hash must be an identifier.Id object.