class Company(models.Model):
    title = models.CharField(verbose_name="Название", max_length=100)
    description = models.TextField(verbose_name="Описание")
    category = models.ForeignKey(Category, verbose_name="Категория", on_delete=models.CASCADE)
    date_added = models.DateTimeField(auto_now_add=True)

    def __str__(self):
        return self.title

    def get_absolute_url(self):
        return reverse("company_detail", kwargs={"category_slug": self.category.slug, "company_id": self.pk })

    def get_main_image(self):
        try:
            main_image = self.images.get(is_main=True)
            return main_image.image.url
        except:
            return ''

    def get_main_image_thumbnail(self):
        try:
            main_image = self.images.get(is_main=True)
            return main_image.thumbnail.url
        except:
            return ''

    def get_all_images(self):
        try:
            return self.images.all()
        except:
            return []

class ImagesGalery(models.Model):
    company = models.ForeignKey(Company, verbose_name="Компания", on_delete=models.CASCADE, related_name="images")

    image = models.ImageField(verbose_name='Изображение', upload_to='media')

    thumbnail = ImageSpecField(source="image", processors=[ResizeToFill(414,220)], format='PNG')
    smart = ImageSpecField(source='image', processors=[SmartResize(50,50)], format='PNG')

    is_main = models.BooleanField(verbose_name="Главное изображение", default=False)