ConstraintExpression: Add __iter__.

Signed-off-by: Chris PeBenito <pebenito@ieee.org>
pebenito · Oct 14, 2020 · aceedd8 · aceedd8
1 parent 582f883
commit aceedd8
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/setools/policyrep.pyi b/setools/policyrep.pyi
@@ -333,6 +333,7 @@ class ConstraintExpression(PolicyObject):
     def __ge__(self, other) -> bool: ...
     def __getitem__(self, index) -> Union[str, FrozenSet[TypeOrAttr], FrozenSet[Role], FrozenSet[User]]: ...
     def __gt__(self, other) -> bool: ...
+    def __iter__(self) -> Iterator[Union[FrozenSet[Role], FrozenSet[Type], FrozenSet[User], str]]: ...
     def __le__(self, other) -> bool: ...
     def __lt__(self, other) -> bool: ...
     def __ne__(self, other) -> bool: ...

diff --git a/setools/policyrep/constraint.pxi b/setools/policyrep/constraint.pxi
@@ -174,6 +174,9 @@ cdef class ConstraintExpression(PolicyObject):
     def __getitem__(self, idx):
         return self._postfix[idx]
 
+    def __iter__(self):
+        return iter(self._postfix)
+
     def __eq__(self, other):
         return self._postfix == other