wbcompliance 1.59.11__py2.py3-none-any.whl → 1.59.13__py2.py3-none-any.whl

wbcompliance/models/risk_management/backend.py

@@ -117,18 +117,26 @@ class AbstractRuleBackend:
         """
         raise NotImplementedError()
 
-    def check_rule(self, *dto_args, **kwargs) -> Generator[IncidentResult, None, None]:
+    def check_rule(
+        self, *dto_args, ignore_breached_objects: Iterable[models.Model] | None = None, **kwargs
+    ) -> Generator[IncidentResult, None, None]:
         """
         Build and Check the rule against the instantiated backend given its attributes
 
         Returns:
             yield the breach object, its string representation, the incident report and the severity as IncidentResult
         """
+        if not ignore_breached_objects:
+            ignore_breached_objects = []
         if not dto_args:
             # We build the data transfer object if it is not provided.
             dto_args = self._build_dto_args()
         if any(dto_args):
-            yield from self._process_dto(*dto_args, **kwargs)
+            incident = list(self._process_dto(*dto_args, **kwargs))
+            yield from filter(
+                lambda incident: incident.breached_object not in ignore_breached_objects,
+                incident,
+            )
 
     def is_passive_evaluation_valid(self) -> bool:
         """

wbcompliance/models/risk_management/checks.py

@@ -159,7 +159,7 @@ class RiskCheck(ComplexToStringMixin, WBModel):
         )
 
     def evaluate(
-        self, *explicit_dto, override_incident: bool = False, ignore_informational_threshold: bool = False
+        self, *explicit_dto, override_incident: bool = False, ignore_informational_threshold: bool = False, **kwargs
     ) -> list[models.Model]:
         """
         Evaluate the check and returns tuple of incidents information
@@ -177,7 +177,7 @@ class RiskCheck(ComplexToStringMixin, WBModel):
         self.status = self.CheckStatus.SUCCESS
         incidents = []
         report_template = Template(self.rule.rule_backend.incident_report_template)
-        for incident_result in rule_backend.check_rule(*explicit_dto):
+        for incident_result in rule_backend.check_rule(*explicit_dto, **kwargs):
             if (
                 ignore_informational_threshold
                 and incident_result.severity.is_ignorable
@@ -296,11 +296,25 @@ class RiskCheck(ComplexToStringMixin, WBModel):
         return "id"
 
 
-@shared_task(queue=Queue.BACKGROUND.value)
+@shared_task(queue=Queue.HIGH_PRIORITY.value)
 def evaluate_as_task(
-    check_id: int, *dto, override_incident: bool = False, ignore_informational_threshold: bool = False
+    check_id: int,
+    *dto,
+    override_incident: bool = False,
+    ignore_informational_threshold: bool = False,
+    ignore_breached_object_content_types: list[tuple[int, int]] | None = None,
 ):
+    if ignore_breached_object_content_types:
+        ignore_breached_objects = [
+            ContentType.objects.get(id=r[1]).get_object_for_this_type(id=r[0])
+            for r in ignore_breached_object_content_types
+        ]
+    else:
+        ignore_breached_objects = []
     check = RiskCheck.all_objects.get(id=check_id)
     check.evaluate(
-        *dto, override_incident=override_incident, ignore_informational_threshold=ignore_informational_threshold
+        *dto,
+        override_incident=override_incident,
+        ignore_informational_threshold=ignore_informational_threshold,
+        ignore_breached_objects=ignore_breached_objects,
     )

wbcompliance/models/risk_management/mixins.py

@@ -1,5 +1,5 @@
 from datetime import date
-from typing import Any
+from typing import Any, Iterable
 
 from django.contrib.contenttypes.models import ContentType
 from django.db import models, transaction
@@ -84,7 +84,17 @@ class RiskCheckMixin(models.Model):
                 return status
         return RiskCheck.CheckStatus.SUCCESS
 
-    def evaluate_active_rules(self, evaluation_date: date, *dto, asynchronously: bool = True):
+    def evaluate_active_rules(
+        self,
+        evaluation_date: date,
+        *dto,
+        asynchronously: bool = True,
+        ignore_breached_objects: Iterable[models.Model] | None = None,
+    ):
+        # serialize the ignore breached object list into a (id, content_type.id) list of tuple
+        ignore_breached_object_content_types = [
+            (o.id, ContentType.objects.get_for_model(o).id) for o in ignore_breached_objects
+        ]
         for rule in RiskRule.objects.get_active_rules_for_object(self.checked_object):
             check = RiskCheck.all_objects.update_or_create(
                 rule=rule,
@@ -98,11 +108,20 @@ class RiskCheckMixin(models.Model):
             if asynchronously:
                 transaction.on_commit(
                     lambda check_id=check.id: evaluate_as_task.delay(
-                        check_id, *dto, override_incident=True, ignore_informational_threshold=True
+                        check_id,
+                        *dto,
+                        override_incident=True,
+                        ignore_informational_threshold=True,
+                        ignore_breached_object_content_types=ignore_breached_object_content_types,
                     )
                 )
             else:
-                check.evaluate(*dto, override_incident=True, ignore_informational_threshold=True)
+                check.evaluate(
+                    *dto,
+                    override_incident=True,
+                    ignore_informational_threshold=True,
+                    ignore_breached_objects=ignore_breached_objects,
+                )
 
     class Meta:
         abstract = True

{wbcompliance-1.59.11.dist-info → wbcompliance-1.59.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: wbcompliance
-Version: 1.59.11
+Version: 1.59.13
 Summary: A workbench module for managing compliance.
 Author-email: Christopher Wittlinger <c.wittlinger@stainly.com>
 Requires-Dist: wbcore

{wbcompliance-1.59.11.dist-info → wbcompliance-1.59.13.dist-info}/RECORD

@@ -56,11 +56,11 @@ wbcompliance/models/compliance_task.py,sha256=V0gFzkmBlLk2lA0CuMbvObn31M1QL11fZG
 wbcompliance/models/compliance_type.py,sha256=uf48Nl-dqyXOse8pAO7Rq7uzPNvtk955HPHRCKX4m-c,4830
 wbcompliance/models/enums.py,sha256=id1m8trG7b6Sw3Sa9VrLgD104GC2nG9A8ngJzaRqIIQ,424
 wbcompliance/models/risk_management/__init__.py,sha256=DJXoHheA5REDpPYP0uyRmFWyNKP-ir_z5YB61CN6N-E,247
-wbcompliance/models/risk_management/backend.py,sha256=pbszChXMxA36JjK0nIkrz0k7w9ImKUTGMVlokgA1R-w,4736
-wbcompliance/models/risk_management/checks.py,sha256=8puXRaVzmerF1tkWHi1MIxNewe-Io0Ux0iU7EleYOdY,12206
+wbcompliance/models/risk_management/backend.py,sha256=EsGOlBIDTVJdZzCYtFmRiL8sUc0LDQJebHq9mwR_Xm4,5061
+wbcompliance/models/risk_management/checks.py,sha256=drBNhT7kj7TA1EclKX9jpPp-IycNcHvjj3xZsbKqB2E,12670
 wbcompliance/models/risk_management/dispatch.py,sha256=cLi63vSGbk9xhCoVgr4UmVk2kfYmGiyohKINSvCodb0,1423
 wbcompliance/models/risk_management/incidents.py,sha256=9BJP6N2Qbpach7ObymZ1Pz5ih0cLqrzXI9fLfYGb-d4,24382
-wbcompliance/models/risk_management/mixins.py,sha256=KsdjpM1Cl235z8h57USVcYSBKtO2VRH1S0FCuJOEMH0,3762
+wbcompliance/models/risk_management/mixins.py,sha256=vitLClx1QwmhooXselB_MqNoLbxMRp-uNQKsvHDooj8,4449
 wbcompliance/models/risk_management/rules.py,sha256=26zanw7maoi-ODJoDj2FLu8PCN5i2dI9J6RU5QsJ9uU,25917
 wbcompliance/serializers/__init__.py,sha256=OhRIuUrexqQePCISpu-BKry9dfioYGfpO8RwlcwhHtI,1122
 wbcompliance/serializers/compliance_form.py,sha256=OxPILlup_CP4Qwvu864oMqz7QUZsk2QOkOGFifWP3gM,11615
@@ -139,6 +139,6 @@ wbcompliance/viewsets/titles/risk_managment/checks.py,sha256=47DEQpj8HBSa-_TImW-
 wbcompliance/viewsets/titles/risk_managment/incidents.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbcompliance/viewsets/titles/risk_managment/rules.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 wbcompliance/viewsets/titles/risk_managment/tables.py,sha256=x_9mr6JWZ9GVNAEvfdNCeFZtGGYSG0-sXrR9pwTyocU,256
-wbcompliance-1.59.11.dist-info/METADATA,sha256=Jdb0_bQcN5TPdYdbRL2OQYcpUWaTgCNLydKytv_s8w4,197
-wbcompliance-1.59.11.dist-info/WHEEL,sha256=aha0VrrYvgDJ3Xxl3db_g_MDIW-ZexDdrc_m-Hk8YY4,105
-wbcompliance-1.59.11.dist-info/RECORD,,
+wbcompliance-1.59.13.dist-info/METADATA,sha256=mabQUpEGTu2bqj1jYwN2kx8USHcQNJq9qG1cmNzyq3c,197
+wbcompliance-1.59.13.dist-info/WHEEL,sha256=aha0VrrYvgDJ3Xxl3db_g_MDIW-ZexDdrc_m-Hk8YY4,105
+wbcompliance-1.59.13.dist-info/RECORD,,