Add-to and remove-from cart

mattfreire · mattfreire · commit e7a4e46df71b · 2019-04-26T10:47:45.000+02:00
diff --git a/core/migrations/0002_auto_20190426_0659.py b/core/migrations/0002_auto_20190426_0659.py
@@ -0,0 +1,25 @@
+# Generated by Django 2.2 on 2019-04-26 06:59
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='item',
+            name='category',
+            field=models.CharField(choices=[('S', 'Shirt'), ('SW', 'Sport wear'), ('OW', 'Outwear')], default='S', max_length=2),
+            preserve_default=False,
+        ),
+        migrations.AddField(
+            model_name='item',
+            name='label',
+            field=models.CharField(choices=[('P', 'primary'), ('S', 'secondary'), ('D', 'danger')], default='P', max_length=1),
+            preserve_default=False,
+        ),
+    ]
diff --git a/core/migrations/0003_item_slug.py b/core/migrations/0003_item_slug.py
@@ -0,0 +1,19 @@
+# Generated by Django 2.2 on 2019-04-26 07:15
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0002_auto_20190426_0659'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='item',
+            name='slug',
+            field=models.SlugField(default='test-product'),
+            preserve_default=False,
+        ),
+    ]
diff --git a/core/migrations/0004_item_discount_price.py b/core/migrations/0004_item_discount_price.py
@@ -0,0 +1,18 @@
+# Generated by Django 2.2 on 2019-04-26 07:17
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0003_item_slug'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='item',
+            name='discount_price',
+            field=models.FloatField(blank=True, null=True),
+        ),
+    ]
diff --git a/core/migrations/0005_item_description.py b/core/migrations/0005_item_description.py
@@ -0,0 +1,19 @@
+# Generated by Django 2.2 on 2019-04-26 07:21
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0004_item_discount_price'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='item',
+            name='description',
+            field=models.TextField(default='This is a test description fefwefweifjwief jeiwfjeiwfj iwef jiwef weif ewj'),
+            preserve_default=False,
+        ),
+    ]
diff --git a/core/migrations/0006_item_quantity.py b/core/migrations/0006_item_quantity.py
@@ -0,0 +1,18 @@
+# Generated by Django 2.2 on 2019-04-26 07:31
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0005_item_description'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='item',
+            name='quantity',
+            field=models.IntegerField(default=1),
+        ),
+    ]
diff --git a/core/migrations/0007_auto_20190426_0737.py b/core/migrations/0007_auto_20190426_0737.py
@@ -0,0 +1,22 @@
+# Generated by Django 2.2 on 2019-04-26 07:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0006_item_quantity'),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name='item',
+            name='quantity',
+        ),
+        migrations.AddField(
+            model_name='orderitem',
+            name='quantity',
+            field=models.IntegerField(default=1),
+        ),
+    ]
diff --git a/core/migrations/0008_auto_20190426_0756.py b/core/migrations/0008_auto_20190426_0756.py
@@ -0,0 +1,26 @@
+# Generated by Django 2.2 on 2019-04-26 07:56
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        migrations.swappable_dependency(settings.AUTH_USER_MODEL),
+        ('core', '0007_auto_20190426_0737'),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name='orderitem',
+            name='ordered',
+            field=models.BooleanField(default=False),
+        ),
+        migrations.AddField(
+            model_name='orderitem',
+            name='user',
+            field=models.ForeignKey(blank=True, null=True, on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL),
+        ),
+    ]
diff --git a/core/migrations/0009_auto_20190426_0757.py b/core/migrations/0009_auto_20190426_0757.py
@@ -0,0 +1,20 @@
+# Generated by Django 2.2 on 2019-04-26 07:57
+
+from django.conf import settings
+from django.db import migrations, models
+import django.db.models.deletion
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('core', '0008_auto_20190426_0756'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='orderitem',
+            name='user',
+            field=models.ForeignKey(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL),
+        ),
+    ]
diff --git a/core/models.py b/core/models.py
@@ -1,20 +1,57 @@
 from django.conf import settings
 from django.db import models
+from django.shortcuts import reverse
+
+CATEGORY_CHOICES = (
+    ('S', 'Shirt'),
+    ('SW', 'Sport wear'),
+    ('OW', 'Outwear')
+)
+
+LABEL_CHOICES = (
+    ('P', 'primary'),
+    ('S', 'secondary'),
+    ('D', 'danger')
+)
 
 
 class Item(models.Model):
     title = models.CharField(max_length=100)
     price = models.FloatField()
+    discount_price = models.FloatField(blank=True, null=True)
+    category = models.CharField(choices=CATEGORY_CHOICES, max_length=2)
+    label = models.CharField(choices=LABEL_CHOICES, max_length=1)
+    slug = models.SlugField()
+    description = models.TextField()
 
     def __str__(self):
         return self.title
 
+    def get_absolute_url(self):
+        return reverse("core:product", kwargs={
+            'slug': self.slug
+        })
+
+    def get_add_to_cart_url(self):
+        return reverse("core:add-to-cart", kwargs={
+            'slug': self.slug
+        })
+
+    def get_remove_from_cart_url(self):
+        return reverse("core:remove-from-cart", kwargs={
+            'slug': self.slug
+        })
+
 
 class OrderItem(models.Model):
+    user = models.ForeignKey(settings.AUTH_USER_MODEL,
+                             on_delete=models.CASCADE)
+    ordered = models.BooleanField(default=False)
     item = models.ForeignKey(Item, on_delete=models.CASCADE)
+    quantity = models.IntegerField(default=1)
 
     def __str__(self):
-        return self.title
+        return f"{self.quantity} of {self.item.title}"
 
 
 class Order(models.Model):
diff --git a/core/urls.py b/core/urls.py
@@ -1,14 +1,18 @@
 from django.urls import path
 from .views import (
-    products,
+    ItemDetailView,
     checkout,
-    home
+    HomeView,
+    add_to_cart,
+    remove_from_cart
 )
 
 app_name = 'core'
 
 urlpatterns = [
-    path('', home, name='home'),
+    path('', HomeView.as_view(), name='home'),
     path('checkout/', checkout, name='checkout'),
-    path('products/', products, name='products')
+    path('product/<slug>/', ItemDetailView.as_view(), name='product'),
+    path('add-to-cart/<slug>/', add_to_cart, name='add-to-cart'),
+    path('remove-from-cart/<slug>/', remove_from_cart, name='remove-from-cart')
 ]
diff --git a/core/views.py b/core/views.py
@@ -1,5 +1,9 @@
-from django.shortcuts import render
-from .models import Item
+from django.contrib import messages
+from django.shortcuts import render, get_object_or_404
+from django.views.generic import ListView, DetailView
+from django.shortcuts import redirect
+from django.utils import timezone
+from .models import Item, OrderItem, Order
 
 
 def products(request):
@@ -13,5 +17,66 @@ def checkout(request):
     return render(request, "checkout.html")
 
 
-def home(request):
-    return render(request, "home.html")
+class HomeView(ListView):
+    model = Item
+    template_name = "home.html"
+
+
+class ItemDetailView(DetailView):
+    model = Item
+    template_name = "product.html"
+
+
+def add_to_cart(request, slug):
+    item = get_object_or_404(Item, slug=slug)
+    order_item, created = OrderItem.objects.get_or_create(
+        item=item,
+        user=request.user,
+        ordered=False
+    )
+    order_qs = Order.objects.filter(user=request.user, ordered=False)
+    if order_qs.exists():
+        order = order_qs[0]
+        # check if the order item is in the order
+        if order.items.filter(item__slug=item.slug).exists():
+            order_item.quantity += 1
+            order_item.save()
+            messages.info(request, "This item quantity was updated.")
+            return redirect("core:product", slug=slug)
+        else:
+            order.items.add(order_item)
+            messages.info(request, "This item was added to your cart.")
+            return redirect("core:product", slug=slug)
+    else:
+        ordered_date = timezone.now()
+        order = Order.objects.create(
+            user=request.user, ordered_date=ordered_date)
+        order.items.add(order_item)
+        messages.info(request, "This item was added to your cart.")
+        return redirect("core:product", slug=slug)
+
+
+def remove_from_cart(request, slug):
+    item = get_object_or_404(Item, slug=slug)
+    order_qs = Order.objects.filter(
+        user=request.user,
+        ordered=False
+    )
+    if order_qs.exists():
+        order = order_qs[0]
+        # check if the order item is in the order
+        if order.items.filter(item__slug=item.slug).exists():
+            order_item = OrderItem.objects.filter(
+                item=item,
+                user=request.user,
+                ordered=False
+            )[0]
+            order.items.remove(order_item)
+            messages.info(request, "This item was removed from your cart.")
+            return redirect("core:product", slug=slug)
+        else:
+            messages.info(request, "This item was not in your cart")
+            return redirect("core:product", slug=slug)
+    else:
+        messages.info(request, "You do not have an active order")
+        return redirect("core:product", slug=slug)
diff --git a/db.sqlite3 b/db.sqlite3
diff --git a/templates/base.html b/templates/base.html
@@ -45,6 +45,20 @@
 {% include "navbar.html" %}
 
 <body>
+
+    <div class="mt-5 pt-4">
+    {% if messages %}
+        {% for message in messages %}
+        <div class="alert alert-{{ message.tags }} alert-dismissible fade show" role="alert">
+          {{ message }}
+          <button type="button" class="close" data-dismiss="alert" aria-label="Close">
+            <span aria-hidden="true">&times;</span>
+          </button>
+        </div>
+        {% endfor %}
+    {% endif %}
+    </div>
+
     {% block content %}
     {% endblock content %}
 
diff --git a/templates/home.html b/templates/home.html
diff --git a/templates/product.html b/templates/product.html
