feat: add add_policies_Ex and add_named_policies_Ex APIs

Author: Chimou0

casbin/internal_enforcer.py

@@ -59,6 +59,27 @@ def _add_policies(self, sec, ptype, rules):
 
         return rules_added
 
+    def _add_policies_Ex(self, sec, ptype, rules):
+        """adds rules to the current policy."""
+        rules_added = self.model.add_policies_Ex(sec, ptype, rules)
+        if not rules_added:
+            return rules_added
+        
+        if self.adapter and self.auto_save:
+            if hasattr(self.adapter, "add_policies_Ex") is False:
+                return False
+
+            if self.adapter.add_policies_Ex(sec, ptype, rules) is False:
+                return False
+
+            if self.watcher and self.auto_notify_watcher:
+                if callable(getattr(self.watcher, "update_for_add_policies_Ex", None)):
+                    self.watcher.update_for_add_policies_Ex(sec, ptype, rules)
+                else:
+                    self.watcher.update()
+
+        return rules_added
+
     def _update_policy(self, sec, ptype, old_rule, new_rule):
         """updates a rule from the current policy."""
         rule_updated = self.model.update_policy(sec, ptype, old_rule, new_rule)

casbin/management_enforcer.py

@@ -117,6 +117,14 @@ def add_policies(self, rules):
         """
         return self.add_named_policies("p", rules)
 
+    def add_policies_Ex(self, rules):
+        """add_policies_Ex adds authorization rules to the current policy.
+
+        If the rule already exists, the rule will not be added.
+        But unlike add_policies, other non-existent rules are added instead of returning false directly.
+        """
+        return self.add_named_policies_Ex("p", rules)
+
     def add_named_policy(self, ptype, *params):
         """adds an authorization rule to the current named policy.
 
@@ -139,6 +147,14 @@ def add_named_policies(self, ptype, rules):
         Otherwise the function returns true for the corresponding by adding the new rule."""
         return self._add_policies("p", ptype, rules)
 
+    def add_named_policies_Ex(self, ptype, rules):
+        """add_named_policies_Ex adds authorization rules to the current policy.
+
+        If the rule already exists, the rule will not be added.
+        But unlike add_named_policies, other non-existent rules are added instead of returning false directly.
+        """
+        return self._add_policies_Ex("p", ptype, rules)
+
     def update_policy(self, old_rule, new_rule):
         """updates an authorization rule from the current policy."""
         return self.update_named_policy("p", old_rule, new_rule)

casbin/model/policy.py

@@ -158,6 +158,20 @@ def add_policies(self, sec, ptype, rules):
 
         return True
 
+    def add_policies_Ex(self, sec, ptype, rules):
+        """
+        add_policies_Ex adds authorization rules to the current policy.
+        If the rule already exists, the rule will not be added.
+        But unlike add_policies, other non-existent rules are added instead of returning false directly.
+        """
+        rules_added = False
+        for rule in rules:
+            if self.has_policy(sec, ptype, rule):
+                continue
+            self.add_policy(sec, ptype, rule)
+            rules_added = True
+        return rules_added
+
     def update_policy(self, sec, ptype, old_rule, new_rule):
         """update a policy rule from the model."""