pff/new: Add swarm_id as optional argument to update_session_entry_id.

[cs-p2p-next.git] / ppf / new / config.py

diff --git a/ppf/new/config.py b/ppf/new/config.py
index 84f3a14..bdc4dfd 100644 (file)
--- a/ppf/new/config.py
+++ b/ppf/new/config.py
@@ -87,11 +87,15 @@ class SwarmDescription(object):
             session_list.append(entry)
         return session_list
 
-    def update_session_entry_id(self, session_entry, cs_id):
+    def update_session_entry_id(self, session_entry, cs_id, swarm_id = None):
         """Add or modify client session id in self and session_entry."""
         session_name = session_entry.data.sections()[0]
         self.data.set(session_name, 'client_session_id', str(cs_id))
         session_entry.data.set(session_name, 'client_session_id', str(cs_id))
+        if swarm_id is not None:
+            self.data.set(session_name, 'swarm_id', str(swarm_id))
+            session_entry.data.set(session_name, 'swarm_id', str(swarm_id))
+            self.data.set('swarm', 'swarm_id', str(swarm_id))
 
     def get_file_archives(self):
         """Return a list containing all archives from swarm."""