from django.db import models
from django.utils.translation import gettext_lazy as _
from django_core.models.base import AbstractBaseModel
from model_bakery import baker


class Banner(AbstractBaseModel):
    title = models.CharField(max_length=255, verbose_name=_("Title"))
    description = models.TextField(verbose_name=_("Description"))
    mobile_image = models.ImageField(verbose_name=_("Mobile Image"), upload_to="banner/mobile_image/")
    desktop_image = models.ImageField(verbose_name=_("Desktop Image"), upload_to="banner/desktop_image/")
    link = models.URLField(verbose_name=_("Link"))
    bg_color = models.CharField(verbose_name=_("BG Color"), max_length=255)
    text_color = models.CharField(verbose_name=_("Text Color"), max_length=255)

    @classmethod
    def _baker(cls):
        return baker.make(cls)

    def __str__(self):
        return str(self.pk)

    class Meta:
        db_table = "banner"
        verbose_name = _("Banner")
        verbose_name_plural = _("Banners")