refactor organization type filter in ProviderCompanyViewSet to use 'AGC'

This commit is contained in:
2025-10-22 09:59:15 +03:30
parent 8355d4af57
commit 31ab784382

View File

@@ -1,32 +1,34 @@
import random
import string
from datetime import timedelta
from apps.product.web.api.v1.viewsets.quota_distribution_api import QuotaDistributionViewSet
from apps.pos_device.web.api.v1.serilaizers import device as device_serializer
from apps.authentication.exceptions import OrganizationBankAccountException
from apps.authorization.api.v1.serializers import UserRelationSerializer
from apps.pos_device.web.api.v1.viewsets.client import POSClientViewSet
from apps.core.mixins.soft_delete_mixin import SoftDeleteMixin
from django.db import transaction
from django.utils.timezone import now
from rest_framework import status
from rest_framework import viewsets
from rest_framework.decorators import action
from rest_framework.exceptions import APIException
from rest_framework.response import Response
from apps.authentication.api.v1.api import (
Organization,
BankAccountInformation,
OrganizationSerializer
)
from apps.core.mixins.search_mixin import DynamicSearchMixin
from apps.core.mixins.admin_mixin import AdminFilterMixin
from apps.authentication.exceptions import OrganizationBankAccountException
from apps.authorization.api.v1.serializers import UserRelationSerializer
from apps.authorization.models import UserRelations
from common.helpers import get_organization_by_user
from rest_framework.exceptions import APIException
from apps.core.mixins.admin_mixin import AdminFilterMixin
from apps.core.mixins.search_mixin import DynamicSearchMixin
from apps.core.mixins.soft_delete_mixin import SoftDeleteMixin
from apps.pos_device import models as pos_models
from rest_framework.response import Response
from rest_framework.decorators import action
from common.tools import CustomOperations
from common.helpers import generate_code
from apps.pos_device.web.api.v1.serilaizers import device as device_serializer
from apps.pos_device.web.api.v1.viewsets.client import POSClientViewSet
from apps.product.models import Broker
from django.utils.timezone import now
from rest_framework import viewsets
from django.db import transaction
from rest_framework import status
from apps.product.web.api.v1.viewsets.quota_distribution_api import QuotaDistributionViewSet
from common.helpers import generate_code
from common.helpers import get_organization_by_user
from common.tools import CustomOperations
class ProviderCompanyViewSet(SoftDeleteMixin, viewsets.ModelViewSet): # noqa
@@ -354,7 +356,7 @@ class StakeHoldersViewSet(SoftDeleteMixin, viewsets.ModelViewSet, DynamicSearchM
stakeholders = self.queryset.filter(
assignment__client__organization=org,
organization__type__key='CMP'
organization__type__key='AGC'
)
# paginate stakeholders
@@ -438,4 +440,3 @@ class StakeHolderShareAmountViewSet(SoftDeleteMixin, viewsets.ModelViewSet, Dyna
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)