fix: KeyError: 'g' when build_role_links

casbin · Jun 24, 2024 · bf9cb44 · bf9cb44
1 parent b03e65a
commit bf9cb44
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/casbin/model/policy.py b/casbin/model/policy.py
@@ -44,8 +44,9 @@ def build_role_links(self, rm_map):
             return
 
         for ptype, ast in self["g"].items():
-            rm = rm_map[ptype]
-            ast.build_role_links(rm)
+            rm = rm_map.get(ptype)
+            if rm:
+                ast.build_role_links(rm)
 
     def build_incremental_role_links(self, rm, op, sec, ptype, rules):
         if sec == "g":