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

fix: company/<uuid:pk>/folders code changed

Browse Source
This commit is contained in:
Fazliddin Abdurahimov
2025-08-06 10:14:52 +05:00
parent 11834f3072
commit f3566f4e8b
1 changed files with 4 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -169,12 +169,9 @@ class CompanyFolderViewSet(BaseViewSetMixin, GenericViewSet):
description="List Company Folders" description="List Company Folders"
) )
@action(methods=["GET"], detail=True, url_path="folders") @action(methods=["GET"], detail=True, url_path="folders")
def list_folder(self, *args: object, **kwargs: object) -> Response: def list_folder(self, pk: uuid.UUID, *args: object, **kwargs: object) -> Response:
company = self.get_object() folders = CompanyFolderModel.objects.filter(company__id=pk)
folders = CompanyFolderModel.objects.filter(company=company) ser = self.get_serializer(instance=folders, many=True) # type: ignore
ser = self.get_serializer(
instance=folders, many=True, context={"company_id": company.pk}
) # type: ignore
return Response(data=ser.data, status=status.HTTP_200_OK) return Response(data=ser.data, status=status.HTTP_200_OK)
@extend_schema( @extend_schema(
@@ -184,7 +181,6 @@ class CompanyFolderViewSet(BaseViewSetMixin, GenericViewSet):
@action(url_path="folders", detail=True, methods=["POST"]) @action(url_path="folders", detail=True, methods=["POST"])
def create_folder(self, request: HttpRequest, *args: object, **kwargs: object) -> Response: def create_folder(self, request: HttpRequest, *args: object, **kwargs: object) -> Response:
company = self.get_object() company = self.get_object()
data = request.data.copy() | dict(company=company.id) # type: ignore ser = self.get_serializer(data=data, context={"company": company}) # type: ignore
ser = self.get_serializer(data=data) # type: ignore
ser.is_valid(raise_exception=True) ser.is_valid(raise_exception=True)
return Response(data=ser.data, status=status.HTTP_201_CREATED) return Response(data=ser.data, status=status.HTTP_201_CREATED)