Improve add_policy api

Author: leeqvip

casbin/model/policy.py

@@ -61,6 +61,15 @@ def has_policy(self, sec, ptype, rule):
 
         return False
 
+    def add_policy(self, sec, ptype, rule):
+        """adds a policy rule to the model."""
+
+        if not self.has_policy(sec, ptype, rule):
+            self.model[sec][ptype].policy.append(rule)
+            return True
+
+        return False
+
     def get_values_for_field_in_policy(self, sec, ptype, field_index):
         """gets all values for a field for all rules in a policy, duplicated values are removed."""
 

setup.py

@@ -7,7 +7,7 @@
 
 setuptools.setup(
     name="casbin",
-    version="0.4",
+    version="0.5",
     author="TechLee",
     author_email="techlee@qq.com",
     description="An authorization library that supports access control models like ACL, RBAC, ABAC in Python",

tests/test_management_api.py

@@ -60,3 +60,28 @@ def test_get_policy_api(self):
         self.assertEqual(e.get_filtered_grouping_policy(0, '', 'data2_admin'), [['alice', 'data2_admin']])
         self.assertTrue(e.has_grouping_policy(['alice', 'data2_admin']))
         self.assertFalse(e.has_grouping_policy(['bob', 'data2_admin']))
+
+    def test_modify_policy_api(self):
+        e = get_enforcer(
+            get_examples("rbac_model.conf"),
+            get_examples("rbac_policy.csv"),
+            # True,
+        )
+
+        self.assertEqual(e.get_policy(), [
+            ['alice', 'data1', 'read'],
+            ['bob', 'data2', 'write'],
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+        ])
+
+        e.add_policy('eve', 'data3', 'read')
+        e.add_named_policy('p', ['eve', 'data3', 'write'])
+        self.assertEqual(e.get_policy(), [
+            ['alice', 'data1', 'read'],
+            ['bob', 'data2', 'write'],
+            ['data2_admin', 'data2', 'read'],
+            ['data2_admin', 'data2', 'write'],
+            ['eve', 'data3', 'read'],
+            ['eve', 'data3', 'write'],
+        ])