Update PTRelationSet._add and PTRelationSet._discard to not try to

[invirt/packages/python-afs.git] / afs / pts.py

diff --git a/afs/pts.py b/afs/pts.py
index 9c44f3a..4d3e17c 100644 (file)
--- 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.