Homework-4

ecommerce/baskets/serializers.py

@@ -10,6 +10,7 @@ class BasketItemSerializer(serializers.ModelSerializer):
     class Meta:
         model = BasketItem
         fields = ("id", "basket", "product", "quantity", "price")
+        read_only_fields = ["basket", "price"]
 
 
 class BasketSerializer(serializers.ModelSerializer):

ecommerce/baskets/views.py

@@ -1,4 +1,6 @@
 from rest_framework import viewsets
+from rest_framework.decorators import action
+from rest_framework.response import Response
 
 from baskets.filters import BasketItemFilter, BasketFilter
 from baskets.models import BasketItem, Basket
@@ -7,6 +9,7 @@
 
 
 class BasketItemViewSet(DetailedViewSetMixin, viewsets.ModelViewSet):
+    http_method_names = ["get"]
     queryset = BasketItem.objects.all()
     serializer_class = BasketItemSerializer
     filterset_class = BasketItemFilter
@@ -15,12 +18,26 @@ class BasketItemViewSet(DetailedViewSetMixin, viewsets.ModelViewSet):
         "detailed": BasketItemDetailedSerializer,
     }
 
+    def get_queryset(self):
+        queryset = super().get_queryset()
+        user = self.request.user
+        return queryset.filter(basket__customer__id=user)
+
 
 class BasketViewSet(DetailedViewSetMixin, viewsets.ModelViewSet):
+    permission_classes = ()
     queryset = Basket.objects.all()
     serializer_class = BasketSerializer
     filterset_class = BasketFilter
     serializer_action_classes = {
         "detailed_list": BasketDetailedSerializer,
         "detailed": BasketDetailedSerializer,
+        "add_item_to_basket": BasketItemSerializer
     }
+
+    @action(detail=True, methods=['post'])
+    def add_item_to_basket(self, request, pk=None):
+        serializer = BasketItemSerializer(data=request.data)
+        serializer.save()
+
+        return Response(serializer.data)

ecommerce/customers/serializers.py

@@ -13,6 +13,24 @@ class Meta:
         fields = ("id", "first_name", "last_name", "email", "is_staff", "is_active", "date_joined")
 
 
+class CustomerRegisterSerializer(serializers.ModelSerializer):
+    password = serializers.CharField(write_only=True)
+
+    def create(self, validated_data):
+        customer = Customer.objects.create_user(
+            email=validated_data['email'],
+            password=validated_data["password"]
+        )
+        customer.save()
+        return customer
+
+    class Meta:
+        model = Customer
+        fields = ("id", "email", "first_name", "last_name", "password",)
+
+
+
+
 class ProfileSerializer(serializers.ModelSerializer):
 
     class Meta:

ecommerce/customers/views.py

@@ -8,7 +8,14 @@
 from customers.models import Customer, Address, City, Country
 from customers.serializers import CustomerSerializer, AddressSerializer, CitySerializer, \
     CountrySerializer, \
-    AddressDetailedSerializer, CityDetailedSerializer, ProfileSerializer
+    AddressDetailedSerializer, CityDetailedSerializer, ProfileSerializer,  CustomerRegisterSerializer
+
+
+class CustomerRegisterViewSet(viewsets.ModelViewSet, mixins.CreateModelMixin):
+    permission_classes = ()
+    http_method_names = ['post']
+    queryset = Customer.objects.all()
+    serializer_class = CustomerRegisterSerializer
 
 
 class AdminCustomerViewSet(viewsets.ModelViewSet):

ecommerce/ecommerce/urls.py

@@ -21,7 +21,7 @@
 from baskets.views import BasketItemViewSet, BasketViewSet
 from core.views import APITokenObtainPairView
 from customers.views import AddressViewSet, CityViewSet, \
-    CountryViewSet, AdminCustomerViewSet, MyProfileViewSet
+    CountryViewSet, AdminCustomerViewSet, MyProfileViewSet, CustomerRegisterViewSet
 from ecommerce.router import router
 from orders.views import OrderItemViewSet, OrderViewSet, BillingAddressViewSet, ShippingAddressViewSet, \
     OrderBankAccountViewSet
@@ -46,6 +46,7 @@
 router.register("banks", BankViewSet)
 router.register("admin-products", AdminProductViewSet, basename="admin-product")
 router.register("admin-customers", AdminCustomerViewSet, basename="admin-customer")
+router.register("register", CustomerRegisterViewSet)
 
 
 urlpatterns = [