abdurahimov_dev/trustme
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: company/<id>/... views are done

Browse Source
This commit is contained in:
Fazliddin Abdurahimov
2025-08-06 15:29:12 +05:00
parent 0935cfcb1f
commit 8b29e3c659
2 changed files with 38 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

60
core/apps/companies/views/companies.py
View File

@@ -1,4 +1,3 @@
from typing import Any
from django_core.mixins import BaseViewSetMixin # type: ignore
from django.contrib.auth import get_user_model
@@ -70,8 +69,9 @@ class CompanyCrudViewSet(BaseViewSetMixin, ModelViewSet):
######################################################################
# company/<uuid:pk>/contract Views
# company/<uuid:pk>/contract
######################################################################
@extend_schema(tags=["Company Contracts"])
class CompanyContractApiView(BaseApiViewMixin, GenericAPIView): # type: ignore
queryset = CompanyModel.objects.all()
permission_classes = [IsCompanyAccount]
@@ -110,33 +110,26 @@ class CompanyContractApiView(BaseApiViewMixin, GenericAPIView): # type: ignore
######################################################################
# company/<uuid:pk>/accounts Views
# company/<uuid:pk>/accounts
######################################################################
class CompanyAccountView(GenericAPIView):
@extend_schema(tags=["Company Accounts"])
class CompanyAccountApiView(BaseApiViewMixin, GenericAPIView): # type: ignore
queryset = CompanyModel.objects.all()
serializer_class = None
serializer_class = RetrieveCompanyAccountSerializer
permission_classes = [IsCompanyAccount]
action_serializer_class = {
"list_account": RetrieveCompanyAccountSerializer
method_serializer_class = {
"get": RetrieveCompanyAccountSerializer
}
method_permission_classes = {
"get": [IsCompanyAccount]
}
def get_serializer_class(self):
if self.request.method == "GET":
return RetrieveCompanyAccountSerializer
return RetrieveCompanyFolderSerializer
@extend_schema(
summary="List company accounts",
description="List Company Accounts"
)
@action(url_path="accounts", detail=True, methods=["GET"])
def list_account(
self,
request: HttpRequest,
*args: object,
**kwargs: object,
) -> Response:
def get(self, *args: object, **kwargs: object) -> Response:
company = self.get_object()
accounts = CompanyAccountModel.objects.filter(company=company)
ser = self.get_serializer(instance=accounts, many=True) # type: ignore
@@ -144,25 +137,36 @@ class CompanyAccountView(GenericAPIView):
######################################################################
# company/<uuid:pk>/folders Views
# company/<uuid:pk>/folders
######################################################################
class CompanyFolderApiView(GenericAPIView):
@extend_schema(tags=["Company Folders"])
class CompanyFolderApiView(BaseApiViewMixin, GenericAPIView): # type: ignore
queryset = CompanyModel.objects.all()
permission_classes = [IsCompanyAccount]
def get_serializer_class(self): # type: ignore
if self.request.method == "POST":
return CreateFolderForCompanySerializer
return RetrieveCompanyFolderSerializer
method_serializer_class = {
"get": RetrieveCompanyFolderSerializer,
"post": CreateFolderForCompanySerializer,
}
method_permission_classes = {
"get": [IsCompanyAccount],
"get": [IsCompanyAccount],
}
@extend_schema(summary="List Company Folders")
def get(self, request: HttpRequest, *args: object, **kwargs: object) -> Response:
@extend_schema(
summary="List Company Folders",
description="List Company Folders",
)
def get(self, *args: object, **kwargs: object) -> Response:
company = self.get_object()
folders = CompanyFolderModel.objects.filter(company=company)
serializer = self.get_serializer(instance=folders, many=True)
return Response(serializer.data, status=status.HTTP_200_OK)
@extend_schema(summary="Create Folder for Company")
@extend_schema(
summary="Create Folder for Company",
description="Create Folder for Company",
)
def post(self, request: HttpRequest, *args: object, **kwargs: object):
company = self.get_object()
serializer = self.get_serializer(data=request.data, context={"company_id": company.pk})