Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement