perf: License

apps/accounts/api/automations/check_account.py

@@ -18,9 +18,13 @@ from accounts.models import (
 )
 from assets.models import Asset
 from common.api import JMSModelViewSet
+from common.permissions import IsValidLicense
 from common.utils import many_get
 from orgs.mixins.api import OrgBulkModelViewSet
+from rbac.permissions import RBACPermission
 from .base import AutomationExecutionViewSet
+from ...filters import NodeFilterBackend
+from ...risk_handlers import RiskHandler
 
 __all__ = [
     "CheckAccountAutomationViewSet",
@@ -29,15 +33,12 @@ __all__ = [
     "CheckAccountEngineViewSet",
 ]
 
-from ...filters import NodeFilterBackend
-
-from ...risk_handlers import RiskHandler
-
 
 class CheckAccountAutomationViewSet(OrgBulkModelViewSet):
     model = CheckAccountAutomation
     filterset_fields = ("name",)
     search_fields = filterset_fields
+    permission_classes = [RBACPermission, IsValidLicense]
     serializer_class = serializers.CheckAccountAutomationSerializer
 
 
@@ -51,6 +52,7 @@ class CheckAccountExecutionViewSet(AutomationExecutionViewSet):
     )
     ordering = ("-date_created",)
     tp = AutomationTypes.check_account
+    permission_classes = [RBACPermission, IsValidLicense]
 
     def get_queryset(self):
         queryset = super().get_queryset()
@@ -84,6 +86,7 @@ class AccountRiskViewSet(OrgBulkModelViewSet):
     search_fields = ["username", "asset__name"]
     filterset_fields = ("risk", "status", "asset_id")
     extra_filter_backends = [NodeFilterBackend]
+    permission_classes = [RBACPermission, IsValidLicense]
     serializer_classes = {
         "default": serializers.AccountRiskSerializer,
         "assets": serializers.AssetRiskSerializer,
@@ -140,7 +143,7 @@ class AccountRiskViewSet(OrgBulkModelViewSet):
 class CheckAccountEngineViewSet(JMSModelViewSet):
     search_fields = ("name",)
     serializer_class = serializers.CheckAccountEngineSerializer
-
+    permission_classes = [RBACPermission, IsValidLicense]
     perm_model = CheckAccountEngine
 
     def get_queryset(self):