Add correct image for selected variant

src/dashboard/views.py

@@ -47,6 +47,7 @@ from core import (
     OrderStatus
 )
 from .forms import (
+    ProductVariantUpdateForm,
     OrderLineFulfillForm,
     OrderLineFormset,
     OrderCancelForm,
@@ -438,14 +439,7 @@ class ProductVariantUpdateView(SuccessMessageMixin, UpdateView):
     pk_url_kwarg = 'variant_pk'
     success_message = 'ProductVariant saved.'
     template_name = 'dashboard/variant_form.html'
-    fields = [
-        'name',
-        'sku',
-        'price',
-        'weight',
-        'track_inventory',
-        'stock',
-    ]
+    form_class = ProductVariantUpdateForm
     context_object_name = 'variant'
 
     def get_context_data(self, **kwargs):

src/storefront/templates/storefront/cart_detail.html

@@ -13,7 +13,11 @@
             <div class="cart__item">
             {% with product=item.variant.product %}
                 <figure class="item__figure">
+                    {% if item.variant.image %}
+                    <img class="item__image" src="{{item.variant.image.image.url}}" alt="{{item.variant.image.image}}">
+                    {% else %}
                     <img class="item__image" src="{{product.get_first_img.image.url}}" alt="{{product.get_first_img.image}}">
+                    {% endif %}
                 </figure>
                 <div class="item__info">
                     <h3>{{product.name}}</h3>

src/storefront/templates/storefront/order_detail.html

@@ -23,7 +23,11 @@
                 <tr>
                 {% with product=item.variant.product %}
                     <td>
+                        {% if item.variant.image %}
+                        <img class="line__image" src="{{item.variant.image.image.url}}" alt="{{item.variant.image.image}}">
+                        {% else %}
                         <img class="line__image" src="{{product.get_first_img.image.url}}" alt="{{product.get_first_img.image}}">
+                        {% endif %}
                     </td>
                     <td>
                         <strong>{{ item.variant }}</strong><br>

src/storefront/templates/storefront/order_form.html

@@ -34,7 +34,11 @@
         <div class="cart__item">
         {% with product=item.variant.product %}
             <figure>
+                {% if item.variant.image %}
+                <img src="{{item.variant.image.image.url}}" alt="{{item.variant.image.image}}">
+                {% else %}
                 <img src="{{product.get_first_img.image.url}}" alt="{{product.get_first_img.image}}">
+                {% endif %}
             </figure>
             <div>
                 <h3>{{product.name}}</h3>