Merge branch 'master' into backend/notifications/44797__email_notifications_not_sent_group_user

Author: EBirkenfeld

backend/src/accounts/tests/services/test_transfer.py

@@ -14,6 +14,7 @@
 from src.authentication.enums import AuthTokenType
 from src.payment.stripe.service import StripeService
 from src.processes.enums import (
+    OwnerRole,
     OwnerType,
     PerformerType,
 )
@@ -748,10 +749,12 @@ def test_accept_transfer__template_owner_in_template__ok(
     token['new_user_id'] = account_2_new_user.id
     request_template_owners = [
         {
+            'role': OwnerRole.OWNER,
             'type': OwnerType.USER,
             'source_id': f'{account_2_owner.id}',
         },
         {
+            'role': OwnerRole.OWNER,
             'type': OwnerType.USER,
             'source_id': account_2_new_user.id,
         },

backend/src/accounts/tests/views/test_users/test_count_workflows.py

@@ -4,6 +4,7 @@
     UserInvite,
 )
 from src.processes.enums import (
+    OwnerRole,
     OwnerType,
     PerformerType,
 )
@@ -170,6 +171,7 @@ def test_count_templates__raw_performer__ok(api_client):
     api_client.token_authenticate(user)
     request_template_owners = [
         {
+            'role': OwnerRole.OWNER,
             'type': OwnerType.USER,
             'source_id': f'{user.id}',
         },

backend/src/authentication/serializers.py

@@ -1,5 +1,6 @@
 from django.contrib.auth import authenticate, get_user_model
 from django.core.validators import RegexValidator
+from django.db.models import Exists, Q
 from drf_recaptcha.fields import ReCaptchaV2Field
 from rest_framework import serializers
 from typing import Any, Dict, Optional
@@ -15,6 +16,8 @@
 from src.generics.fields import DateFormatField, TimeStampField
 from src.generics.mixins.services import EncryptionMixin
 from src.generics.serializers import CustomValidationErrorMixin
+from src.processes.enums import OwnerRole, OwnerType
+from src.processes.models.templates.owner import TemplateOwner
 
 UserModel = get_user_model()
 
@@ -182,13 +185,59 @@ class Meta:
             'timezone',
             'date_fmt',
             'date_fdw',
+            'has_workflow_viewer_access',
+            'has_workflow_starter_access',
         )
 
     account = ContextAccountSerializer()
     is_supermode = serializers.BooleanField(required=False)
+    has_workflow_viewer_access = serializers.SerializerMethodField()
+    has_workflow_starter_access = serializers.SerializerMethodField()
     date_joined_tsp = TimeStampField(source='date_joined', read_only=True)
     date_fmt = DateFormatField(read_only=True)
 
+    def get_has_workflow_viewer_access(self, obj) -> bool:
+        access = self._get_template_access(obj)
+        return access['viewer']
+
+    def get_has_workflow_starter_access(self, obj) -> bool:
+        access = self._get_template_access(obj)
+        return access['starter']
+
+    def _get_template_access(self, obj) -> dict:
+        if hasattr(self, '_template_access_cache'):
+            return self._template_access_cache
+
+        user_id = obj.id
+        account_id = obj.account_id
+
+        base_qs = TemplateOwner.objects.filter(
+            template__account_id=account_id,
+            is_deleted=False,
+        ).filter(
+            Q(type=OwnerType.USER, user_id=user_id) |
+            Q(type=OwnerType.GROUP, group__users__id=user_id),
+        )
+        owner_subq = base_qs.filter(role=OwnerRole.OWNER)
+        viewer_subq = base_qs.filter(role=OwnerRole.VIEWER)
+        starter_subq = base_qs.filter(role=OwnerRole.STARTER)
+
+        result = UserModel.objects.filter(pk=user_id).annotate(
+            has_owner=Exists(owner_subq),
+            has_viewer=Exists(viewer_subq),
+            has_starter=Exists(starter_subq),
+        ).values('has_owner', 'has_viewer', 'has_starter').first()
+
+        has_owner = result['has_owner'] if result else False
+        has_viewer = result['has_viewer'] if result else False
+        has_starter = result['has_starter'] if result else False
+
+        self._template_access_cache = {
+            'viewer': has_owner or has_viewer,
+            'starter': has_owner or has_viewer or has_starter,
+        }
+        return self._template_access_cache
+
     def to_representation(self, data):
         data = super().to_representation(data)
         data['is_supermode'] = self.context.get('is_supermode', False)