Track statement references

duo-labs · May 13, 2020 · 799ae9d · 799ae9d
1 parent 09121b9
commit 799ae9d
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/parliament/statement.py b/parliament/statement.py
@@ -263,9 +263,13 @@ class Statement:
     findings = []
     effect_allow = True
     stmt = None
+    policy_id = None
 
     _is_valid = True
 
+    def set_policy_identifier(self, policy_id):
+        self.policy_id = policy_id
+
     def __init__(self, stmt, analyze=True):
         self.findings = []
         self.stmt = stmt