From 799ae9d01c8eeb8ed8357ed83d6c8aff5e70259c Mon Sep 17 00:00:00 2001
From: Scott Piper <scott@summitroute.com>
Date: Wed, 13 May 2020 11:10:59 -0600
Subject: [PATCH] Track statement references

---
 parliament/statement.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/parliament/statement.py b/parliament/statement.py
index 18d1136..3bde768 100644
--- 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