FIX: policy removal for non-filled models

casbin_tortoise_adapter/adapter.py

@@ -60,7 +60,7 @@ async def add_policies(self, sec: str, ptype: str, rules: List[RuleType]):
 
     async def remove_policy(self, sec: str, ptype: str, rule: RuleType):
         """Removes a policy rule from storage."""
-        vs = {f"v{i}": rule[i] if len(rule) > i else "" for i in range(6)}
+        vs = {f"v{i}": rule[i] for i in range(6) if len(rule) > i}
         r = await self.modelclass.filter(ptype=ptype, **vs).delete()
         return r > 0
 
@@ -70,7 +70,7 @@ async def remove_policies(self, sec, ptype, rules: List[RuleType]):
             return
 
         qs = [
-            Q(**{f"v{i}": rule[i] if len(rule) > i else "" for i in range(6)})
+            Q(**{f"v{i}": rule[i] for i in range(6) if len(rule) > i})
             for i, rule in enumerate(rules)
         ]
         await self.modelclass.filter(Q(*qs, join_type=Q.OR), ptype=ptype).delete()

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "casbin-tortoise-adapter"
-version = "1.0.0.post1"
+version = "1.0.1"
 description = "Tortoise ORM adapter for AsyncCasbin"
 authors = ["Elias Gabriel <me@eliasfgabriel.com>"]
 license = "Apache-2.0"

tests/conftest.py

@@ -24,7 +24,7 @@ async def adapter():
             },
             "apps": {"my_app": {"models": ["casbin_tortoise_adapter"]}},
         },
-        _create_db=True,
+        # _create_db=True,
     )
     await Tortoise.generate_schemas()