commit 3e3c1cd4eec0a42774a2161ed508f128f6362bf5
author Sapan Bhatia <gwsapan@gmail.com> Sat Jul 15 01:35:44 2017 -0400
committer Sapan Bhatia <sapan@onlab.us> Mon Jul 17 13:18:30 2017 -0700
tree 19fcc4d8e8c820be1ad4f5c4ad544c201dc5145a
parent 5be3a851c1e350259484505091201742407dd30d

CORD-1392: Autogenerate security policies

Change-Id: Ia183f2a84b27923802e62435c82b20b50fb3fcee

lib/xos-genx/xosgenx/xos2jinja.py

diff --git a/lib/xos-genx/xosgenx/xos2jinja.py b/lib/xos-genx/xosgenx/xos2jinja.py
index c58260e..edd80ef 100644
--- a/lib/xos-genx/xosgenx/xos2jinja.py
+++ b/lib/xos-genx/xosgenx/xos2jinja.py
@@ -124,6 +124,16 @@
 
         return True
 
+    def visit_PolicyDefinition(self, obj):
+        if self.package:
+            pname = '.'.join([self.package, obj.name.value.pval])
+        else:
+            pname = obj.name.value.pval
+
+        self.policies[pname] = obj.body
+
+        return True
+
     def visit_PackageStatement(self, obj):
         dotlist = obj.name.value
         dotlist2 = [f.pval for f in dotlist]