from django.db import models
from django.utils.text import slugify


class GroupCompany(models.Model):
    name = models.CharField(max_length=200)
    slug = models.CharField(max_length=256, blank=True, null=True)
    image = models.ImageField(upload_to="groupcompany/images/")
    overview = models.TextField()
    description = models.TextField()

    def __str__(self):
        return self.name

    def save(self, *args, **kwargs):
        if not self.pk:
            self.slug = slugify(self.name)
        return super().save(*args, **kwargs)


class Product(models.Model):
    group_company = models.ForeignKey('GroupCompany', on_delete=models.CASCADE, related_name="products")
    product_image = models.ImageField(upload_to="groupcompany/product/images/")