Update enforcer.py

Author: gemanor

permit/enforcement/enforcer.py

@@ -30,6 +30,7 @@ class CheckQuery(TypedDict):
     user: User
     action: Action
     resource: Resource
+    context: Optional[Context]
 
 
 SETUP_PDP_DOCS_LINK = (
@@ -424,7 +425,7 @@ async def get_user_permissions(
                 ) from err
 
     async def filter_objects(
-        self, user: User, action: Action, _context: Dict[str, str], resources: List[Dict[str, Any]]
+        self, user: User, action: Action, context: Dict[str, str], resources: List[Dict[str, Any]]
     ) -> List[Dict[str, Any]]:
         """
         Filter objects based on permissions using bulk check.
@@ -439,11 +440,7 @@ async def filter_objects(
                 "attributes": resource.get("attributes", {}),
                 "tenant": resource.get("tenant"),
             }
-            check_query: CheckQuery = {
-                "user": user,
-                "action": action,
-                "resource": permit_resource,
-            }
+            check_query: CheckQuery = {"user": user, "action": action, "resource": permit_resource, "context": context}
             requests.append(check_query)
 
         results = await self.bulk_check(requests)