October 13, 2016 17:02 · Oct 13, 2016
diff --git a/Wagtial integration with saleor b/Wagtial integration with saleor
@@ -0,0 +1,118 @@
+from django.db import models
+from wagtail.wagtailcore.models import Page
+from satchless.item import StockedItem
+from wagtail.wagtailimages.edit_handlers import ImageChooserPanel
+from wagtail.wagtailadmin.edit_handlers import (TabbedInterface, ObjectList,
+                                                PageChooserPanel, FieldPanel,
+                                                InlinePanel, MultiFieldPanel)
+#from wagtail.wagtailadmin.views.pages import PAGE_EDIT_HANDLERS
+from wagtail.wagtailcore.fields import RichTextField
+from wagtail.wagtailcore.models import Orderable
+from wagtail.wagtailadmin.edit_handlers import TabbedInterface, ObjectList
+
+from django_prices.models import PriceField
+from modelcluster.fields import ParentalKey
+
+
+class LinkFields(models.Model):
+    link_external = models.URLField("External link", blank=True)
+    link_page = models.ForeignKey(
+        'wagtailcore.Page',
+        null=True,
+        blank=True,
+        related_name='+'
+    )
+
+    @property
+    def link(self):
+        if self.link_page:
+            return self.link_page.url
+        else:
+            return self.link_external
+
+    panels = [
+        FieldPanel('link_external'),
+        PageChooserPanel('link_page'),
+    ]
+
+    class Meta:
+        abstract = True
+
+
+class Picture(LinkFields):
+    image = models.ForeignKey(
+        'wagtailimages.Image',
+        null=True,
+        blank=True,
+        on_delete=models.SET_NULL,
+        related_name='+'
+    )
+    embed_url = models.URLField("Embed URL", blank=True)
+    caption = models.CharField(max_length=255, blank=True)
+
+    panels = [
+        ImageChooserPanel('image'),
+        FieldPanel('embed_url'),
+        FieldPanel('caption'),
+        MultiFieldPanel(LinkFields.panels, "Link"),
+    ]
+
+    class Meta:
+        abstract = True
+
+
+class ProductPicture(Orderable, Picture):
+    page = ParentalKey('products.Product', related_name='pictures')
+
+
+class Variant(StockedItem, models.Model):
+    name = models.CharField(max_length=200)
+    description = RichTextField()
+    price = PriceField(currency='USD', max_digits=5, decimal_places=2)
+    size = models.CharField(max_length=3, null=True, blank=True)
+    color = models.CharField(max_length=10, null=True, blank=True)
+    product = ParentalKey('products.Product', related_name='variants')
+    stock = models.IntegerField('In stock')
+
+    def __str__(self):
+        return self.name
+
+    def get_stock(self):
+        return self.stock
+
+
+class Product(Page):
+    description = RichTextField()
+
+    @property
+    def name(self):
+        return self.title
+
+    def __str__(self):
+        return self.name
+
+
+Product.content_panels = [
+    FieldPanel('title'),
+    FieldPanel('description'),
+    #InlinePanel(Product, 'pictures', label='Pictures')
+]
+
+
+Product.content_panels += [
+	InlinePanel('pictures', label="Related pictures"),
+ ]
+
+
+Product.variant_panels = [
+    InlinePanel('variants', label='Variants')
+]
+
+
+edit_handlers = TabbedInterface([
+    ObjectList(Product.content_panels, heading='Product'),
+    ObjectList(Product.promote_panels, heading='Promote'),
+    ObjectList(Product.variant_panels, heading='Variants')
+])
+
+Product.edit_handler = edit_handlers