wbcompliance 1.59.4__tar.gz → 1.59.5rc0__tar.gz

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/PKG-INFO

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

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/wbcompliance/models/compliance_type.py

@@ -13,6 +13,7 @@ from slugify import slugify
 from wbcore.contrib.documents.models import Document, DocumentType
 from wbcore.contrib.documents.models.mixins import DocumentMixin
 from wbcore.models import WBModel
+from wbcore.workers import Queue
 
 User = get_user_model()
 SelfComplianceType = TypeVar("SelfComplianceType", bound="ComplianceType")
@@ -83,7 +84,7 @@ class ComplianceType(WBModel):
         return user.has_perm("wbcompliance.administrate_compliance")
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def update_or_create_compliance_document(user_id: int, content_type_id: int, object_id: int, send_email: bool = True):
     user = get_user_model().objects.get(id=user_id)
     content_type = ContentType.objects.get(id=content_type_id)

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/wbcompliance/models/risk_management/checks.py

@@ -15,6 +15,7 @@ from wbcore.contrib.color.enums import WBColor
 from wbcore.contrib.icons import WBIcon
 from wbcore.models import WBModel
 from wbcore.utils.models import ComplexToStringMixin
+from wbcore.workers import Queue
 
 from .incidents import CheckedObjectIncidentRelationship, RiskIncident, RiskIncidentType
 
@@ -295,7 +296,7 @@ class RiskCheck(ComplexToStringMixin, WBModel):
         return "id"
 
 
-@shared_task
+@shared_task(queue=Queue.BACKGROUND.value)
 def evaluate_as_task(
     check_id: int, *dto, override_incident: bool = False, ignore_informational_threshold: bool = False
 ):

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/wbcompliance/models/risk_management/incidents.py

@@ -30,6 +30,7 @@ from wbcore.metadata.configs.display.instance_display.shortcuts import (
     create_simple_display,
 )
 from wbcore.utils.models import ComplexToStringMixin
+from wbcore.workers import Queue
 
 User: BaseUser = get_user_model()
 
@@ -617,7 +618,7 @@ def post_save_incident_relationship(sender, instance, created, raw, **kwargs):
         instance.incident.save()
 
 
-@shared_task
+@shared_task(queue=Queue.DEFAULT.value)
 def resolve_all_incidents_as_task(
     resolved_by_id,
     reviewer_comment,

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/wbcompliance/models/risk_management/rules.py

@@ -29,6 +29,7 @@ from wbcore.contrib.notifications.dispatch import send_notification
 from wbcore.models import WBModel
 from wbcore.utils.models import ComplexToStringMixin
 from wbcore.utils.rrules import convert_rrulestr_to_dict
+from wbcore.workers import Queue
 
 from .backend import AbstractRuleBackend
 from .checks import RiskCheck
@@ -612,7 +613,7 @@ def pre_delete_risk_threshold(sender, instance, **kwargs):
     CheckedObjectIncidentRelationship.objects.filter(incident__rule=instance.rule, severity=instance.severity).delete()
 
 
-@shared_task
+@shared_task(queue=Queue.BACKGROUND.value)
 def process_rule_as_task(rule_id: int, evaluation_date: date, override_incident: bool | None = False):
     """
     Async task to process rule

{wbcompliance-1.59.4 → wbcompliance-1.59.5rc0}/wbcompliance/tasks.py

@@ -5,6 +5,7 @@ from celery import shared_task
 from django.contrib.contenttypes.models import ContentType
 from django.db.models import Q
 from tqdm import tqdm
+from wbcore.workers import Queue
 
 from wbcompliance.models import ComplianceTask, ReviewComplianceTask
 from wbcompliance.models.risk_management.rules import (
@@ -13,7 +14,7 @@ from wbcompliance.models.risk_management.rules import (
 )
 
 
-@shared_task
+@shared_task(queue=Queue.EXTENDED_BACKGROUND.value)
 def check_passive_rules(
     from_date: date | None = None,
     to_date: date | None = None,
@@ -56,7 +57,7 @@ def check_passive_rules(
                 rule.notify(checked_date)
 
 
-@shared_task
+@shared_task(queue=Queue.BACKGROUND.value)
 def clean_dynamic_rule_relationships(debug: bool = False):
     """
     Periodic function to check reverse generic object that don't exist anymore. Furthermore, get the queryset representing all the active relationship for all backend and ensure that every object have a relationship (e.g. new object creation)
@@ -79,7 +80,7 @@ def clean_dynamic_rule_relationships(debug: bool = False):
             leftover_relationship.delete()
 
 
-@shared_task
+@shared_task(queue=Queue.BACKGROUND.value)
 def periodic_quaterly_or_monthly_compliance_task():
     today = datetime.now()
     qs = ComplianceTask.objects.filter(active=True)