SWATCH_STRING = '-swatch'
THUMB_STRING = '-thumb'

class ParentSKU(models.Model):
    sku = models.CharField(max_length = 200, unique = True)
    yahoo_id = models.CharField(max_length = 250, blank=True)
    slug = models.SlugField(max_length = 150, blank=True)
    title = models.CharField(max_length = 200, blank=True)
    headline = models.CharField(max_length = 200, blank=True)
    caption1 = models.TextField(blank=True)
    caption2 = models.TextField(blank=True)
    option1_type = models.CharField(max_length = 250, blank=True)
    option2_type = models.CharField(max_length = 250, blank=True)
    galleries = generic.GenericRelation(Gallery)
    def __unicode__(self):
        return '%s' % (self.sku)
    def get_absolute_url(self):
        return reverse('products_parentsku_detail', args = [self.sku])
    def save(self, *args, **kwargs):
        super(ParentSKU, self).save(*args, **kwargs)
        create_application_gallery(self, 'ParentSKU')
    def get_gallery(self):
        try:
            return self.galleries.all()[0]
        except: return None
    def get_default_subsku(self):
        if self.subsku_set:
            try:
                return self.subsku_set.all().get(sku = self.sku)
            except:
                return self.subsku_set.all()[0]
        return None


class SubSKU(models.Model):
    sku = models.CharField(max_length = 200, unique = True)
    parent_sku = models.ForeignKey(ParentSKU, null=True)
    yahoo_id = models.CharField(max_length = 250, blank=True)
    slug = models.SlugField(max_length = 150, blank=True)
    price = models.DecimalField(decimal_places = 2, max_digits = 6, null=True)
    sale_on = models.BooleanField()
    sale_price = models.DecimalField(decimal_places = 2, max_digits = 6, null=True, blank=True)
    qoh = models.IntegerField(null=True)
    barcode = models.CharField(max_length = 250, blank=True)
    weight = models.DecimalField(decimal_places = 2, max_digits = 6, null=True, blank=True)
    option1 = models.CharField(max_length = 250, blank=True)
    option2 = models.CharField(max_length = 250, blank=True)
    image_prefix = models.CharField(max_length = 20, blank=True)
    def __unicode__(self):
        return '%s' % (self.sku)
    def get_absolute_url(self):
        return reverse('products_subsku_detail', args = [self.sku])
    def get_default_image(self):
        parent_gallery = self.parent_sku.get_gallery()
        if parent_gallery:
            default_image = parent_gallery.photo_set.filter(title__icontains = self.image_prefix).order_by('title')
            if default_image:
                return default_image[0]
        return None
    def get_swatch_image(self):
        parent_gallery = self.parent_sku.get_gallery()
        if parent_gallery:
            swatch_image = parent_gallery.photo_set.filter(title__icontains = self.image_prefix + SWATCH_STRING).order_by('title')
            if swatch_image:
                return swatch_image[0]
        return None

    def get_thumb_image(self):
        parent_gallery = self.parent_sku.get_gallery()
        if parent_gallery:
            thumb_image = parent_gallery.photo_set.filter(title__icontains = self.image_prefix + THUMB_STRING).order_by('title')
            if thumb_image:
                return thumb_image[0]
        return None

    def get_detail_images(self):
        parent_gallery = self.parent_sku.get_gallery()
        if parent_gallery:
            return parent_gallery.photo_set.all().exclude(title__icontains = THUMB_STRING).exclude(title__icontains = SWATCH_STRING).order_by('title')
        return None