wbcompliance 2.2.1__py2.py3-none-any.whl

wbcompliance/viewsets/endpoints/compliance_task.py

@@ -0,0 +1,193 @@
+from rest_framework.reverse import reverse
+from wbcompliance.models import ComplianceType, ReviewComplianceTask
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class ComplianceTaskGroupEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None
+
+
+class ComplianceTaskEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None
+
+
+class ComplianceTaskComplianceTaskGroupEndpointConfig(ComplianceTaskEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if group_id := self.view.kwargs.get("group_id", None):
+            return reverse(
+                "wbcompliance:compliancetaskgroup-compliancetask-list", args=[group_id], request=self.request
+            )
+        return None
+
+
+class ComplianceTaskInstanceEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        if self.request.user.is_superuser:
+            return super().get_delete_endpoint()
+        return None
+
+
+class ComplianceTaskInstanceComplianceTaskEndpointConfig(ComplianceTaskInstanceEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return reverse(
+                "wbcompliance:compliancetask-compliancetaskinstance-list",
+                args=[self.view.kwargs["task_id"]],
+                request=self.request,
+            )
+        return None
+
+
+class ComplianceTaskMatrixEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return reverse("wbcompliance:compliancetaskmatrix-list", [], request=self.request)
+        return None
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_instance_endpoint(self, **kwargs):
+        return None
+
+
+class ComplianceActionEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None
+
+
+class ComplianceEventEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint()
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            obj = self.view.get_object()
+            if not ComplianceType.is_administrator(self.request.user) and obj.creator != self.request.user.profile:
+                return None
+        return super().get_instance_endpoint()
+
+    def get_delete_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_delete_endpoint()
+        return None
+
+
+class ReviewComplianceTaskEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None
+
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            if not ComplianceType.is_administrator(self.request.user):
+                return None
+            if obj_id := self.view.kwargs.get("pk", None):
+                obj = ReviewComplianceTask.objects.get(id=obj_id)
+                if obj.status != ReviewComplianceTask.Status.DRAFT:
+                    return None
+        return super().get_instance_endpoint()
+
+    def get_delete_endpoint(self, **kwargs):
+        if not ComplianceType.is_administrator(self.request.user):
+            return None
+        if obj_id := self.view.kwargs.get("pk", None):
+            obj = ReviewComplianceTask.objects.get(id=obj_id)
+            if obj.status != ReviewComplianceTask.Status.DRAFT or obj.is_instance:
+                return None
+        return super().get_delete_endpoint()
+
+
+class AbstractComplianceTaskReviewEndpointConfig(EndpointViewConfig):
+    def get_instance_endpoint(self, **kwargs):
+        if self.instance:
+            if not ComplianceType.is_administrator(self.request.user):
+                return None
+            if review_id := self.view.kwargs.get("review_id", None):
+                obj = ReviewComplianceTask.objects.get(id=review_id)
+                if obj.status != ReviewComplianceTask.Status.DRAFT:
+                    return None
+        return super().get_instance_endpoint()
+
+    def get_create_endpoint(self, **kwargs):
+        if not ComplianceType.is_administrator(self.request.user):
+            return None
+        if review_id := self.view.kwargs.get("review_id", None):
+            obj = ReviewComplianceTask.objects.get(id=review_id)
+            if obj.status != ReviewComplianceTask.Status.DRAFT:
+                return None
+        return super().get_create_endpoint()
+
+    def get_delete_endpoint(self, **kwargs):
+        if not ComplianceType.is_administrator(self.request.user):
+            return None
+        if review_id := self.view.kwargs.get("review_id", None):
+            obj = ReviewComplianceTask.objects.get(id=review_id)
+            if obj.status != ReviewComplianceTask.Status.DRAFT:
+                return None
+        return super().get_delete_endpoint()
+
+
+class ComplianceTaskReviewNoGroupEndpointConfig(AbstractComplianceTaskReviewEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            if review_id := self.view.kwargs.get("review_id", None):
+                return reverse(
+                    "wbcompliance:review-compliancetasknogroup-list", args=[review_id], request=self.request
+                )
+        return None
+
+
+class ComplianceTaskReviewGroupEndpointConfig(AbstractComplianceTaskReviewEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if (
+            ComplianceType.is_administrator(self.request.user)
+            and (review_id := self.view.kwargs.get("review_id", None))
+            and (group_id := self.view.kwargs.get("group_id", None))
+        ):
+            return reverse(
+                "wbcompliance:review-compliancetaskgroup-list", args=[review_id, group_id], request=self.request
+            )
+        return None
+
+
+class ComplianceTaskInstanceReviewNoGroupEndpointConfig(ComplianceTaskInstanceEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user) and (
+            review_id := self.view.kwargs.get("review_id", None)
+        ):
+            return reverse(
+                "wbcompliance:review-compliancetaskinstancenogroup-list", args=[review_id], request=self.request
+            )
+        return None
+
+
+class ComplianceTaskInstanceReviewGroupEndpointConfig(ComplianceTaskInstanceEndpointConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            if (review_id := self.view.kwargs.get("review_id", None)) and (
+                group_id := self.view.kwargs.get("group_id", None)
+            ):
+                return reverse(
+                    "wbcompliance:review-compliancetaskinstancegroup-list",
+                    args=[review_id, group_id],
+                    request=self.request,
+                )
+        return None

wbcompliance/viewsets/endpoints/compliance_type.py

@@ -0,0 +1,9 @@
+from wbcompliance.models import ComplianceType
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class ComplianceTypeEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if ComplianceType.is_administrator(self.request.user):
+            return super().get_endpoint()
+        return None

wbcompliance/viewsets/endpoints/risk_managment/__init__.py

@@ -0,0 +1,12 @@
+from .checks import RiskCheckEndpointConfig
+from .incidents import (
+    CheckedObjectIncidentRelationshipEndpointConfig,
+    CheckedObjectIncidentRelationshipRiskRuleEndpointConfig,
+    RiskIncidentEndpointConfig,
+)
+from .rules import (
+    RiskRuleEndpointConfig,
+    RuleCheckedObjectRelationshipRiskRuleEndpointConfig,
+    RuleThresholdRiskRuleEndpointConfig,
+)
+from .tables import RiskManagementIncidentTableEndpointConfig

wbcompliance/viewsets/endpoints/risk_managment/checks.py

@@ -0,0 +1,16 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class RiskCheckEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+    def get_list_endpoint(self, **kwargs):
+        if rule_id := self.view.kwargs.get("rule_id"):
+            return reverse(
+                "wbcompliance:riskrule-check-list",
+                args=[rule_id],
+                request=self.request,
+            )
+        return super().get_endpoint(**kwargs)

wbcompliance/viewsets/endpoints/risk_managment/incidents.py

@@ -0,0 +1,36 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class RiskIncidentEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if rule_id := self.view.kwargs.get("rule_id"):
+            return reverse(
+                "wbcompliance:riskrule-incident-list",
+                args=[rule_id],
+                request=self.request,
+            )
+        return super().get_endpoint(**kwargs)
+
+    def get_create_endpoint(self, **kwargs):
+        return None
+
+    def get_delete_endpoint(self, **kwargs):
+        return None
+
+
+class CheckedObjectIncidentRelationshipEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+    def get_list_endpoint(self, **kwargs):
+        return reverse("wbcompliance:checkedobjectincidentrelationship-list", request=self.request)
+
+
+class CheckedObjectIncidentRelationshipRiskRuleEndpointConfig(CheckedObjectIncidentRelationshipEndpointConfig):
+    def get_list_endpoint(self, **kwargs):
+        return reverse(
+            "wbcompliance:riskincident-relationship-list",
+            args=[self.view.kwargs["incident_id"]],
+            request=self.request,
+        )

wbcompliance/viewsets/endpoints/risk_managment/rules.py

@@ -0,0 +1,32 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class RuleThresholdRiskRuleEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if rule_id := self.view.kwargs.get("rule_id"):
+            return reverse(
+                "wbcompliance:riskrule-threshold-list",
+                args=[rule_id],
+                request=self.request,
+            )
+        return super().get_endpoint(**kwargs)
+
+
+class RuleCheckedObjectRelationshipRiskRuleEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        if rule_id := self.view.kwargs.get("rule_id"):
+            return reverse(
+                "wbcompliance:riskrule-relationship-list",
+                args=[rule_id],
+                request=self.request,
+            )
+        return super().get_endpoint(**kwargs)
+
+
+class RiskRuleEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return super().get_endpoint(**kwargs)
+
+    def get_delete_endpoint(self, **kwargs):
+        return None

wbcompliance/viewsets/endpoints/risk_managment/tables.py

@@ -0,0 +1,14 @@
+from rest_framework.reverse import reverse
+from wbcore.metadata.configs.endpoints import EndpointViewConfig
+
+
+class RiskManagementIncidentTableEndpointConfig(EndpointViewConfig):
+    def get_endpoint(self, **kwargs):
+        return None
+
+    def get_list_endpoint(self, **kwargs):
+        return reverse(
+            "wbcompliance:riskmanagementincidenttable-list",
+            args=[],
+            request=self.request,
+        )

wbcompliance/viewsets/menu/__init__.py

@@ -0,0 +1,17 @@
+from wbcompliance.permissions import (
+    is_admin_compliance,
+    is_internal_employee_compliance,
+)
+
+from .compliance_form import COMPLIANCEFORM_MENU, COMPLIANCEFORMSIGNATURE_MENUITEM
+from .compliance_task import (
+    COMPLIANCE_SETTINGS_MENU,
+    COMPLIANCEEVENT_MENUITEM,
+    COMPLIANCEREPORT_MENU,
+)
+from .compliance_type import COMPLIANCETYPE_MENUITEM
+from .risk_management import (
+    RISK_INCIDENT_MENUITEM,
+    RISKINCIDENTTABLE_MENUITEM,
+    RULE_MENUITEM,
+)

wbcompliance/viewsets/menu/compliance_form.py

@@ -0,0 +1,49 @@
+from django.utils.translation import gettext as _
+from wbcompliance.permissions import is_admin_compliance
+from wbcore.menus import ItemPermission, Menu, MenuItem
+
+COMPLIANCEFORM_MENUITEM = MenuItem(
+    label=_("Forms"),
+    endpoint="wbcompliance:complianceform-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_complianceform", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Form"),
+        endpoint="wbcompliance:complianceform-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)
+
+
+COMPLIANCEFORMSIGNATURE_MENUITEM = MenuItem(
+    label=_("Form Signatures"),
+    endpoint="wbcompliance:complianceformsignature-list",
+    permission=ItemPermission(permissions=["wbcompliance.view_complianceformsignature"]),
+)
+
+ADMIN_COMPLIANCEFORMSIGNATURE_MENUITEM = MenuItem(
+    label=_("Form Signatures"),
+    endpoint="wbcompliance:complianceformsignature-list",
+    permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.view_complianceformsignature"]),
+)
+
+COMPLIANCEFORMTYPE_MENUITEM = MenuItem(
+    label=_("Form Types"),
+    endpoint="wbcompliance:complianceformtype-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_complianceformtype", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Form Type"),
+        endpoint="wbcompliance:complianceformtype-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)
+
+COMPLIANCEFORM_MENU = Menu(
+    label=_("Forms"),
+    items=[COMPLIANCEFORM_MENUITEM, ADMIN_COMPLIANCEFORMSIGNATURE_MENUITEM, COMPLIANCEFORMTYPE_MENUITEM],
+)

wbcompliance/viewsets/menu/compliance_task.py

@@ -0,0 +1,130 @@
+from django.utils.translation import gettext as _
+from wbcompliance.permissions import (
+    is_admin_compliance,
+    is_internal_employee_compliance,
+)
+from wbcore.menus import ItemPermission, Menu, MenuItem
+
+from .compliance_type import COMPLIANCETYPE_MENUITEM
+
+COMPLIANCETASKGROUP_MENUITEM = MenuItem(
+    label=_("Groups Indicator"),
+    endpoint="wbcompliance:compliancetaskgroup-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_compliancetaskgroup", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Group Indicator"),
+        endpoint="wbcompliance:compliancetaskgroup-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)
+
+COMPLIANCETASK_MENUITEM = MenuItem(
+    label=_("Indicators"),
+    endpoint="wbcompliance:compliancetask-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_compliancetask", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Indicator"),
+        endpoint="wbcompliance:compliancetask-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)
+
+COMPLIANCETASKINSTANCE_MENUITEM = MenuItem(
+    label=_("Instances Indicator"),
+    endpoint="wbcompliance:compliancetaskinstance-list",
+    permission=ItemPermission(
+        method=lambda request: request.user.is_superuser,
+        permissions=["wbcompliance.view_compliancetaskinstance", "wbcompliance.administrate_compliance"],
+    ),
+)
+
+COMPLIANCETASKMATRIX_MENUITEM = MenuItem(
+    label=_("Indicator Matrix"),
+    endpoint="wbcompliance:compliancetaskmatrix-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=[
+            "wbcompliance.view_compliancetaskinstance",
+            "wbcompliance.view_compliancetask",
+            "wbcompliance.administrate_compliance",
+        ],
+    ),
+)
+
+COMPLIANCEACTION_MENUITEM = MenuItem(
+    label=_("Actions"),
+    endpoint="wbcompliance:complianceaction-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_complianceaction", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Compliance Action"),
+        endpoint="wbcompliance:complianceaction-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)
+
+COMPLIANCEEVENT_MENUITEM = MenuItem(
+    label=_("Events"),
+    endpoint="wbcompliance:complianceevent-list",
+    permission=ItemPermission(
+        method=is_internal_employee_compliance, permissions=["wbcompliance.view_complianceevent"]
+    ),
+    add=MenuItem(
+        label=_("Add Compliance Event"),
+        endpoint="wbcompliance:complianceevent-list",
+        permission=ItemPermission(
+            method=is_internal_employee_compliance, permissions=["wbcompliance.add_complianceevent"]
+        ),
+    ),
+)
+
+ADMIN_COMPLIANCEEVENT_MENUITEM = MenuItem(
+    label=_("Events"),
+    endpoint="wbcompliance:complianceevent-list",
+    permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.view_complianceevent"]),
+    add=MenuItem(
+        label=_("Add Compliance Event"),
+        endpoint="wbcompliance:complianceevent-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.add_complianceevent"]),
+    ),
+)
+
+REVIEWCOMPLIANCETASK_MENUITEM = MenuItem(
+    label=_("Reports"),
+    endpoint="wbcompliance:reviewcompliancetask-list",
+    permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.view_reviewcompliancetask"]),
+    add=MenuItem(
+        label=_("Add Report Indicator"),
+        endpoint="wbcompliance:reviewcompliancetask-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.add_reviewcompliancetask"]),
+    ),
+)
+
+
+COMPLIANCEREPORT_MENU = Menu(
+    label=_("Reports"),
+    items=[
+        REVIEWCOMPLIANCETASK_MENUITEM,
+        COMPLIANCEACTION_MENUITEM,
+        ADMIN_COMPLIANCEEVENT_MENUITEM,
+        COMPLIANCETASKMATRIX_MENUITEM,
+    ],
+)
+
+COMPLIANCE_SETTINGS_MENU = Menu(
+    label=_("Settings"),
+    items=[
+        COMPLIANCETYPE_MENUITEM,
+        COMPLIANCETASKGROUP_MENUITEM,
+        COMPLIANCETASK_MENUITEM,
+        COMPLIANCETASKINSTANCE_MENUITEM,
+    ],
+)

wbcompliance/viewsets/menu/compliance_type.py

@@ -0,0 +1,17 @@
+from django.utils.translation import gettext as _
+from wbcompliance.permissions import is_admin_compliance
+from wbcore.menus import ItemPermission, MenuItem
+
+COMPLIANCETYPE_MENUITEM = MenuItem(
+    label=_("Administrators"),
+    endpoint="wbcompliance:compliancetype-list",
+    permission=ItemPermission(
+        method=is_admin_compliance,
+        permissions=["wbcompliance.view_compliancetype", "wbcompliance.administrate_compliance"],
+    ),
+    add=MenuItem(
+        label=_("Add Compliance Type"),
+        endpoint="wbcompliance:compliancetype-list",
+        permission=ItemPermission(method=is_admin_compliance, permissions=["wbcompliance.administrate_compliance"]),
+    ),
+)

wbcompliance/viewsets/menu/risk_management.py

@@ -0,0 +1,56 @@
+from contextlib import suppress
+
+from django.utils.translation import gettext as _
+from wbcompliance.models.risk_management.rules import RuleGroup
+from wbcore.menus import ItemPermission, Menu, MenuItem
+from wbcore.permissions.shortcuts import is_internal_user
+
+RULE_MENUITEM = MenuItem(
+    label=_("Rules"),
+    endpoint="wbcompliance:riskrule-list",
+    permission=ItemPermission(
+        method=lambda request: is_internal_user(request.user), permissions=["wbcompliance.view_riskrule"]
+    ),
+    add=MenuItem(
+        label=_("Add Rule"),
+        endpoint="wbcompliance:riskrule-list",
+        permission=ItemPermission(
+            method=lambda request: is_internal_user(request.user), permissions=["wbcompliance.add_riskrule"]
+        ),
+    ),
+)
+
+
+def _get_incident_menu():
+    items = []
+    with suppress(Exception):
+        for rule_group in RuleGroup.objects.all():
+            items.append(
+                MenuItem(
+                    label=rule_group.name,
+                    endpoint="wbcompliance:riskincident-list",
+                    permission=ItemPermission(
+                        method=lambda request: is_internal_user(request.user),
+                        permissions=["wbcompliance.view_riskincident"],
+                    ),
+                    endpoint_get_parameters={"rule_group": rule_group.id},
+                )
+            )
+    if items:
+        return Menu(label=_("Incidents"), items=items)
+    return MenuItem(
+        label=_("Incidents"),
+        endpoint="wbcompliance:riskincident-list",
+        permission=ItemPermission(
+            method=lambda request: is_internal_user(request.user), permissions=["wbcompliance.view_riskincident"]
+        ),
+    )
+
+
+RISK_INCIDENT_MENUITEM = _get_incident_menu()
+
+RISKINCIDENTTABLE_MENUITEM = MenuItem(
+    label=_("Incident Table"),
+    endpoint="wbcompliance:riskmanagementincidenttable-list",
+    permission=ItemPermission(permissions=["wbcompliance.administrate_riskrule"]),
+)

wbcompliance/viewsets/risk_management/__init__.py

@@ -0,0 +1,21 @@
+from .checks import RiskCheckModelViewSet, RiskCheckRepresentationViewSet
+from .incidents import (
+    CheckedObjectIncidentRelationshipModelViewSet,
+    CheckedObjectIncidentRelationshipRepresentationViewSet,
+    CheckedObjectIncidentRelationshipRiskRuleModelViewSet,
+    RiskIncidentModelViewSet,
+    RiskIncidentRepresentationViewSet,
+    RiskIncidentRiskRuleModelViewSet,
+    RiskIncidentTypeRepresentationViewSet,
+)
+from .rules import (
+    RuleGroupRepresentationViewSet,
+    RiskRuleModelViewSet,
+    RiskRuleRepresentationViewSet,
+    RuleBackendRepresentationViewSet,
+    RuleCheckedObjectRelationshipRepresentationViewSet,
+    RuleCheckedObjectRelationshipRiskRuleModelViewSet,
+    RuleThresholdRepresentationViewSet,
+    RuleThresholdRiskRuleModelViewSet,
+)
+from .tables import RiskManagementIncidentTableView

wbcompliance/viewsets/risk_management/checks.py

@@ -0,0 +1,49 @@
+from django.contrib.auth import get_user_model
+from django.db import models
+from django.db.models import F
+from wbcompliance.filters.risk_management.checks import RiskCheckFilterSet
+from wbcompliance.models.risk_management import RiskCheck
+from wbcompliance.serializers import (
+    RiskCheckModelSerializer,
+    RiskCheckRepresentationSerializer,
+)
+from wbcompliance.viewsets.buttons import RiskCheckButtonConfig
+from wbcompliance.viewsets.display import RiskCheckDisplayConfig
+from wbcompliance.viewsets.endpoints import RiskCheckEndpointConfig
+from wbcore import viewsets
+
+User = get_user_model()
+
+
+class RiskCheckRepresentationViewSet(viewsets.RepresentationViewSet):
+    queryset = RiskCheck.objects.all()
+    serializer_class = RiskCheckRepresentationSerializer
+
+
+class RiskCheckModelViewSet(viewsets.ModelViewSet):
+    button_config_class = RiskCheckButtonConfig
+    display_config_class = RiskCheckDisplayConfig
+    endpoint_config_class = RiskCheckEndpointConfig
+    queryset = RiskCheck.objects.all()
+    serializer_class = RiskCheckModelSerializer
+
+    filterset_class = RiskCheckFilterSet
+
+    def get_queryset(self):
+        return (
+            RiskCheck.objects.annotate(
+                rule_repr=F("rule_checked_object_relationship__rule__name"),
+                status_icon=models.Case(
+                    *[
+                        models.When(status=value, then=models.Value(RiskCheck.CheckStatus[value].icon))
+                        for value in RiskCheck.CheckStatus.values
+                    ],
+                    default=models.Value("none")
+                ),
+            )
+            .select_related(
+                "rule_checked_object_relationship",
+                "trigger_content_type",
+            )
+            .prefetch_related("incidents")
+        )