set address view and permiosn

backend/order/views.py

@@ -7,13 +7,16 @@ from .serializers import *
 # from cart.models import 
 from rest_framework import status
 from .models import OrderItemModel, OrderModel, DiscountCode
-from .permissons import CanDeleteCartItemPermissions, GetOrderPermission
+from .permissons import CanDeleteCartItemPermissions, GetOrderPermission, SetAddressPermissions
 from azbankgateways import bankfactories, models as bank_models
 from azbankgateways.exceptions import AZBankGatewaysException
 from drf_spectacular.utils import extend_schema, OpenApiParameter, OpenApiTypes
 from utils.pagination import StructurePagination
 from order.models import OrderModel
 from django.urls import reverse
+from account.models import UserAddressModel
+
+
 # try:
 #     pass
 # except DiscountNotAvailableError:
@@ -249,4 +252,29 @@ def callback_view(request):
 
     return HttpResponse(
         "پرداخت با شکست مواجه شده است. اگر پول کم شده است ظرف مدت ۴۸ ساعت پول به حساب شما بازخواهد گشت."
-    )
+    )
+
+
+
+class SetAddressSerilizer(serializers.Serializer):
+    address_id = serializers.IntegerField()
+
+class SetAddressForCartView(APIView):
+    serializer_class = SetAddressSerilizer
+    permission_classes = [IsAuthenticated, SetAddressPermissions]
+    def post(self, request):
+        address_id = request.data.get('address_id', None)
+        if not address_id:
+            return Response({'detail': 'address_id را ارسال کنید'}, status=status.HTTP_400_BAD_REQUEST)
+        address_object = get_object_or_404(UserAddressModel, pk=address_id)
+        permission = SetAddressPermissions()
+        if not permission.has_object_permission(request, self, address_object):
+            return Response({"detail": permission.message}, status=status.HTTP_403_FORBIDDEN)
+        
+        cart_order, created = OrderModel.objects.get_or_create(
+            user=request.user,
+            status='CART'
+        )
+        cart_order.address = address_object
+        cart_order.save()
+        return Response({'detail': 'ادرس با موفقیت انتخاب شد'})