From: Evan Broder <broder@mit.edu>
Date: Mon, 23 Nov 2009 04:05:42 +0000 (-0500)
Subject: Update PTRelationSet._add and PTRelationSet._discard to not try to
X-Git-Tag: 0.1.1~3
X-Git-Url: http://xvm.mit.edu/gitweb/invirt/packages/python-afs.git/commitdiff_plain/76b00ab5bf3750a131294674c402ee17930b4309

Update PTRelationSet._add and PTRelationSet._discard to not try to
mutate the membership set if it hasn't been loaded yet.

Signed-off-by: Evan Broder <broder@mit.edu>
---

diff --git a/afs/pts.py b/afs/pts.py
index 9c44f3a..4d3e17c 100644
--- a/afs/pts.py
+++ b/afs/pts.py
@@ -45,7 +45,8 @@ class PTRelationSet(collections.MutableSet):
         Args:
             elt: The element to add.
         """
-        self._set.add(self._ent._pts.getEntry(elt))
+        if hasattr(self, '_set'):
+            self._set.add(self._ent._pts.getEntry(elt))
 
     def _discard(self, elt):
         """Remove a PTEntry to this instance's internal representation.
@@ -57,7 +58,8 @@ class PTRelationSet(collections.MutableSet):
         Args:
             elt: The element to discard.
         """
-        self._set.discard(self._ent._pts.getEntry(elt))
+        if hasattr(self, '_set'):
+            self._set.discard(self._ent._pts.getEntry(elt))
 
     def __len__(self):
         """Count the members/groups in this set.