Передать параметры (высота и ширина) в класс. models.ImageField?

Question

[mmailcompany]

Начал изучать Django. Понимаю, что вперед надо бы python... ну не суть.

Имеется класс для уменьшения и конвертации изображения:

fields.py

class WEBPThumbnailImg(ImageFieldFile):
    # def __init__(self, w, h):
        # self.width = w
        # self.height = h

    def save(self, name, content, save=True, max_width=200, max_height=200):
        content.file.seek(0)
        image = Image.open(content.file)
        # max_width  = 200
        # max_height = 200
        (w, h) = image.size
        if float(image.size[0]) > max_width:
            
            ......


class WEBPThumbnail(models.ImageField):
		attr_class = WEBPThumbnailImg

Как передать в модель высоту и ширину?

models.py

...

class Category(models.Model):
    name         = models.CharField("Категория", max_length=100, db_index=True)
    slug         = models.SlugField("URL", max_length=44, unique=True, db_index=True)
    # thumbnail    = models.ImageField("Миниатюра", upload_to=upload_cats_thumbnail, blank=True)
    thumbnail    = WEBPThumbnail("Миниатюра", upload_to=upload_cats_thumbnail, blank=True ) #добавить ширину и высоту?

...

Можно разумеется указать явные, но планируется использовать класс и для другого проекта, но с другими величинами.

Добавлено к вопросу:

Временное решение оставил в комментариях к ответу Dev-op. Если есть идеи и наставления, прошу отписаться...

Comments

[Vindicar]

Я с джанго не работал почти, но... вот это не похоже?

Answer 1

[Dev-op]

Если вы хотите использовать класс в других проектах то первое, это не надо жестко кодировать max_width и max_height
def save(self, name, content, save=True, max_width=200, max_height=200)
Передайте через поле модели и сохраните.

class WEBPThumbnail(models.ImageField):
    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 200)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)

    def attr_class(self):
        return WEBPThumbnailImg

    def pre_save(self, model_instance, add):
        file = super().pre_save(model_instance, add)
        if file and not file._committed:
            file.max_width = self.max_width
            file.max_height = self.max_height
        return file

Теперь вы можете использовать эти параметры в class WEBPThumbnailImg(ImageFieldFile) если ваши условия True то сохраняете изображения с новыми парметрами.

Используйте в вашей модели.
class Category(models.Model)

Comments

[mmailcompany]

Exception Type: TypeError
Exception Value:
WEBPThumbnail.attr_class() takes 1 positional argument but 4 were given

P.s

Если вы хотите использовать класс в других проектах то первое, это не надо жестко кодировать max_width и max_height

просто тестировал уменьшение и конвертацию изо

class полностью:

class WEBPThumbnailImg(ImageFieldFile):
    def save(self, name, content, save=True):
        content.file.seek(0)
        image = Image.open(content.file)
        # max_width  = 200
        # max_height = 200
        (w, h) = image.size
        if float(w > max_width):
            img_ratio = (float(image.size[0]) / max_width)
            new_heigh = int(float(image.size[1]) / img_ratio)
            image = image.resize((max_width, new_heigh), Image.LANCZOS)
        elif h > max_height:
            img_ratio = (float(image.size[1]) / max_height)
            new_width = int(float(image.size[0]) / img_ratio)
            image = image.resize((new_width, max_height), Image.LANCZOS)
        else:
            image = image.resize(image.size, Image.LANCZOS)

        image_bytes = io.BytesIO()
        image.save(fp=image_bytes, format="WEBP")
        image_content_file = ContentFile(content=image_bytes.getvalue())
        super().save(name, image_content_file, save)

[Dev-op]

mmailcompany, Да машинально как метод написал. Тут это атрибут класса, сорян.

class WEBPThumbnail(models.ImageField):
    attr_class = WEBPThumbnailImg  # атрибут

    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 200)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)

    def pre_save(self, model_instance, add):
        file = super().pre_save(model_instance, add)
        if file and not file._committed:
            file.max_width = self.max_width
            file.max_height = self.max_height
        return file

[mmailcompany]

Dev-op, да это бывает. я методом тыка подправил. теперь получается передать параметры, ошибок нету

thumbnail    = WEBPThumbnail("Миниатюра", upload_to=upload_cats_thumbnail, blank=True, max_width=400, max_height=200 )

остался вопрос как их в классе WEBPThumbnailImg принять

class WEBPThumbnailImg(ImageFieldFile):
        
    def save(self, name, content, save=True):
        content.file.seek(0)
        image = Image.open(content.file)
        # max_width  = self.max_width
        # max_height = self.max_height
        (w, h) = image.size
        if float(w > max_width):

при загрузке ругается, что нет параметров max_width (name 'max_width' is not defined)

[Dev-op]

mmailcompany, class WEBPThumbnailImg(ImageFieldFile):

1.нужно получать max_width max_height через self
2.if float(w > max_width) будет ошибка исправьте if float(w) > max_width:

[Dev-op]

mmailcompany, зачем закомментировали
max_width = self.max_width
max_height = self.max_height

[mmailcompany]

Dev-op, я убирал комментарий, он все-равно не видит почему-то

[Dev-op]

mmailcompany, напишите ошибки и текущий код
WEBPThumbnail(models.ImageField)
WEBPThumbnailImg(ImageFieldFile)

[mmailcompany]

Dev-op,

class WEBPThumbnailImg(ImageFieldFile):
        
    def save(self, name, content, save=True):
        content.file.seek(0)
        image = Image.open(content.file)
        max_width  = self.max_width
        max_height = self.max_height
        (w, h) = image.size
        if  w > max_width:
            img_ratio = (float(image.size[0]) / max_width)
            new_heigh = int(float(image.size[1]) / img_ratio)
            image = image.resize((max_width, new_heigh), Image.LANCZOS)
        elif h > max_height:
            img_ratio = (float(image.size[1]) / max_height)
            new_width = int(float(image.size[0]) / img_ratio)
            image = image.resize((new_width, max_height), Image.LANCZOS)
        else:
            image = image.resize(image.size, Image.LANCZOS)

        image_bytes = io.BytesIO()
        image.save(fp=image_bytes, format="WEBP")
        image_content_file = ContentFile(content=image_bytes.getvalue())
        super().save(name, image_content_file, save)


class WEBPThumbnail(models.ImageField):
    attr_class = WEBPThumbnailImg

    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 200)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)

    def pre_save(self, model_instance, add):
        file = super().pre_save(model_instance, add)
        if file and not file._committed:
            file.max_width = self.max_width
            file.max_height = self.max_height
        return file

class Category(models.Model):
    name         = models.CharField("Категория", max_length=100, db_index=True)
    slug         = models.SlugField("URL", max_length=44, unique=True, db_index=True)
    # thumbnail    = models.ImageField("Миниатюра", upload_to=upload_cats_thumbnail, blank=True)
    thumbnail    = WEBPThumbnail("Миниатюра", upload_to=upload_cats_thumbnail, blank=True, max_width=400, max_height=200 )

Ругается на строку max_width = self.max_width:

class WEBPThumbnailImg(ImageFieldFile):
        
    def save(self, name, content, save=True):
        content.file.seek(0)
        image = Image.open(content.file)
        max_width  = self.max_width

если вручную прописываю, то на след строку ругается max_height

[mmailcompany]

Dev-op, может я их не так добавляю в class Category?!!

[Dev-op]

mmailcompany,
Старайтесь логи скидывать.
1.

new_heigh = int(float(image.size[1]) / img_ratio)
            image = image.resize((max_width, new_heigh), Image.LANCZOS)

тут ошибка в слове height

2. if h > max_height вместо elif используйте if
3. pre_save не подойдет в текущем коде def pre_save(self, model_instance, add): pre_save используется перед сохранением. Если он вам не нужен, то попробуйте без него написать логику.

[mmailcompany]

Dev-op,
1. не принципиально. главное что в обеих строках одинаково.
2. учту, спасибо
3. да, уже коментил. без него тоже все работает, но если руками прописать величины.

суть в том, что до обработки изо дело не доходит.
обработчик ругается на эти строки:

max_width  = self.max_width
max_height = self.max_height

self.max_width и self.max_height он не видит
если руками прописать им величины, то код полностью отрабатывает как надо...

[Dev-op]

mmailcompany, зачем вам это Django?))

Попробуйте так:

class WEBPThumbnailImg(ImageFieldFile):
    def save(self, name, content, save=True, max_width=None, max_height=None):
        content.file.seek(0)
        image = Image.open(content.file)

        if max_width and max_height:
            (w, h) = image.size
            if w > max_width:
                img_ratio = float(w) / max_width
                new_height = int(float(h) / img_ratio)
                image = image.resize((max_width, new_height), Image.LANCZOS)
            if h > max_height:
                img_ratio = float(h) / max_height
                new_width = int(float(w) / img_ratio)
                image = image.resize((new_width, max_height), Image.LANCZOS)

        # Сохраняем изображение
        image_bytes = io.BytesIO()
        image.save(fp=image_bytes, format="WEBP")
        image_content_file = ContentFile(content=image_bytes.getvalue())
        super().save(name, image_content_file, save)

class WEBPThumbnail(models.ImageField):
    attr_class = WEBPThumbnailImg

    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 200)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)

    def save_form_data(self, instance, data):
        # Передаем параметры max_width и max_height в метод save класса WEBPThumbnailImg
        if data:
            data.max_width = self.max_width
            data.max_height = self.max_height
        super().save_form_data(instance, data)

class Category(models.Model):
    name = models.CharField("Категория", max_length=100, db_index=True)
    slug = models.SlugField("URL", max_length=44, unique=True, db_index=True)
    thumbnail = WEBPThumbnail(
        "Миниатюра", 
        upload_to=upload_cats_thumbnail, 
        blank=True, 
        max_width=400, 
        max_height=200
    )

[mmailcompany]

Dev-op,

зачем вам это Django?))

хочется)))

Попробуйте так:

загружает, но не обрабатывает
условие if max_width and max_height: файлит))

я кст до форума так пробовал

def save(self, name, content, save=True, max_width=None, max_height=None):

не помогает... у самого голова уже пухнет....
лог на код последний, что я выкладывал:

spoiler

консоль тоже самое говорит

сорян, что мозг Вам взорвал... пора отдыхать. спасибо за наводи. может завтра, что додумаю)))

[mmailcompany]

Dev-op,
если так пишу

class WEBPThumbnailImg(ImageFieldFile):

    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 300)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)
        
    def save(self, name, content, save=True, max_width=None, max_height=None):
        content.file.seek(0)
        image = Image.open(content.file)
        max_width  = self.max_width
        max_height = self.max_height
        (w, h) = image.size
        if    w > max_width:
              img_ratio = int(float(image.size[0]) / max_width)
              new_heigh = int(float(image.size[1]) / img_ratio)
              image = image.resize((max_width, new_heigh), Image.LANCZOS)
        elif  h > max_height:
              img_ratio = int(float(image.size[1]) / max_height)
              new_width = int(float(image.size[0]) / img_ratio)
              image = image.resize((new_width, max_height), Image.LANCZOS)
        else:
              image = image.resize(image.size, Image.LANCZOS)

        image_bytes = io.BytesIO()
        image.save(fp=image_bytes, format="WEBP")
        image_content_file = ContentFile(content=image_bytes.getvalue())
        super().save(name, image_content_file, save)


class WEBPThumbnail(models.ImageField):
    attr_class = WEBPThumbnailImg  # атрибут

    def __init__(self, *args, **kwargs):
        self.max_width = kwargs.pop('max_width', 200)
        self.max_height = kwargs.pop('max_height', 200)
        super().__init__(*args, **kwargs)

то обрабатывает, но значения берет сверху (300 и 200). переданные значения не воспринимает и нижние не учитывает

[Dev-op]

mmailcompany,
Попробуйте добавить в

class WEBPThumbnailImg(ImageFieldFile):
    def save(self, name, content, save=True, max_width=None, max_height=None):
self.max_width = max_width
self.max_height = max_height
......

если не получится попробуйте напрямую передать , более просто й вариант:

def save_form_data(self, instance, data):
        if data:
            data.save(data.name, data.file, save=True, max_width=self.max_width, max_height=self.max_height)
        else:
            super().save_form_data(instance, data)

[mmailcompany]

Dev-op,
спасибо за идеи. ответ отмечу решением.
но оформил пока так (может потом придет озарение или, кто-нибудь еще идеи подкинет):

def converter_compressor(content, max_width, max_height):
    content.file.seek(0)
    image = Image.open(content.file)
    (w, h) = image.size
    if    w > max_width:
          img_ratio = int(float(image.size[0]) / max_width)
          new_height = int(float(image.size[1]) / img_ratio)
          image = image.resize((max_width, new_height), Image.LANCZOS)
    elif  h > max_height:
          img_ratio = int(float(image.size[1]) / max_height)
          new_width = int(float(image.size[0]) / img_ratio)
          image = image.resize((new_width, max_height), Image.LANCZOS)
    else:
          image = image.resize(image.size, Image.LANCZOS)

    image_bytes = io.BytesIO()
    image.save(fp=image_bytes, format="WEBP")
    image_content_file = ContentFile(content=image_bytes.getvalue())
    return image_content_file

class WEBPThumbnailImg(ImageFieldFile):
        
    def save(self, name, content, save=True):
        image_content_file = converter_compressor(content, 300, 300)
        super().save(name, image_content_file, save)

class WEBPThumbnail(models.ImageField):
    attr_class = WEBPThumbnailImg