initial commit

core/apps/accounts/views/me.py

@@ -0,0 +1,70 @@
+from rest_framework.viewsets import GenericViewSet  # type: ignore
+from rest_framework.decorators import action  # type: ignore
+from rest_framework import status  # type: ignore
+from rest_framework.request import HttpRequest  # type: ignore
+from rest_framework.response import Response  # type: ignore
+from rest_framework.permissions import (  # type: ignore
+    IsAuthenticated
+)
+
+from django_core.mixins import BaseViewSetMixin  # type: ignore
+
+from core.apps.companies.serializers import (
+    RetrieveCompanySerializer,
+    CreateCompanySerializer,
+)
+from core.apps.companies.models import (
+    CompanyModel,
+    CompanyAccountModel
+)
+
+from django.db import transaction
+
+
+class MeCompanyView(BaseViewSetMixin, GenericViewSet):
+    permission_classes = [IsAuthenticated]
+
+    action_permission_classes = {}
+    action_serializer_class = {
+        "create": CreateCompanySerializer,
+        "list": RetrieveCompanySerializer,
+    }
+
+    def list(
+        self,
+        request: HttpRequest,
+        *args: object,
+        **kwargs: object
+    ) -> Response:
+        
+        companies = CompanyModel.objects.filter(
+            accounts__user=request.user
+        )
+
+        return Response(
+            RetrieveCompanySerializer(instance=companies, many=True).data, 
+            status=status.HTTP_200_OK
+        )
+    
+    def create(
+        self,
+        request: HttpRequest,
+        *args: object,
+        **kwargs: object
+    ) -> Response:
+        
+        with transaction.atomic():
+            serializer = CreateCompanySerializer(data=request.data)  # type: ignore
+            serializer.is_valid(raise_exception=True)
+            company = serializer.save()  # type: ignore
+
+            account = CompanyAccountModel(
+                company=company, 
+                user=request.user
+            )
+            account.save()
+
+        return Response(
+            data=serializer.data, 
+            status=status.HTTP_201_CREATED
+        )