From: Brian Warner <warner@lothar.com>
Date: Sun, 27 Feb 2011 02:11:11 +0000 (-0700)
Subject: happinessutil.py: server-vs-tracker cleanup
X-Git-Url: https://git.rkrishnan.org/pf/content/frontends/somewhere?a=commitdiff_plain;h=459b9d9e7c8c7c14bedf0c5473325ac44b4230d8;p=tahoe-lafs%2Ftahoe-lafs.git

happinessutil.py: server-vs-tracker cleanup
---

diff --git a/src/allmydata/util/happinessutil.py b/src/allmydata/util/happinessutil.py
index 9a5d74b2..11d36791 100644
--- a/src/allmydata/util/happinessutil.py
+++ b/src/allmydata/util/happinessutil.py
@@ -54,7 +54,7 @@ def shares_by_server(servermap):
             ret.setdefault(peerid, set()).add(shareid)
     return ret
 
-def merge_peers(servermap, upload_servers=None):
+def merge_peers(servermap, upload_trackers=None):
     """
     I accept a dict of shareid -> set(peerid) mappings, and optionally a
     set of PeerTrackers. If no set of PeerTrackers is provided, I return
@@ -66,15 +66,15 @@ def merge_peers(servermap, upload_servers=None):
     # context where it is okay to do that, make a copy of servermap and
     # work with it.
     servermap = deepcopy(servermap)
-    if not upload_servers:
+    if not upload_trackers:
         return servermap
 
     assert(isinstance(servermap, dict))
-    assert(isinstance(upload_servers, set))
+    assert(isinstance(upload_trackers, set))
 
-    for peer in upload_servers:
-        for shnum in peer.buckets:
-            servermap.setdefault(shnum, set()).add(peer.serverid)
+    for tracker in upload_trackers:
+        for shnum in tracker.buckets:
+            servermap.setdefault(shnum, set()).add(tracker.serverid)
     return servermap
 
 def servers_of_happiness(sharemap):