генерировать миниатюру при загрузке изображения django - PullRequest
Новая
       4

генерировать миниатюру при загрузке изображения django

0 голосов
/ 08 января 2020

У меня есть сайт, где несколько больших изображений размера закачанных каждый день после того, как они go живая из-за сильный размером его фиксации времени, поэтому я хочу, чтобы генерировать миниатюру версии, когда изображение загружено: 

class Image(models.Model):
      license_type = (
    ('Royalty-Free','Royalty-Free'),
    ('Rights-Managed','Rights-Managed')
                         )
      image_number = models.CharField(default=random_image_number,max_length=12,unique=True)
      title = models.CharField(default=random_image_number,max_length = 100)
      image = models.ImageField(upload_to = 'image' , default = 'demo/demo.png')
      thumbnail = models.ImageField(upload_to='thumbs', editable=False)
      category = models.ForeignKey('Category', null=True, blank=True, on_delete=models.CASCADE)
      shoot = models.ForeignKey(ImageShoot, on_delete=models.CASCADE, related_name='Image', null=True,blank=True)
      image_keyword = models.TextField(max_length=1000)
      credit = models.CharField(max_length=150, null=True)
      location = models.CharField(max_length=100, null=True)
      license_type = models.CharField(max_length=20,choices=license_type, default='')
      uploaded_at = models.TimeField(auto_now_add=True)

      def __str__(self):
          return self.title

      def save(self, *args, **kwargs):

          super(Image, self).save(*args, **kwargs)
          if not self.make_thumbnail():
             raise Exception('Could not create thumbnail - is the file type valid?')

      def make_thumbnail(self):
          fh = storage.open(self.Image.name)
          try:
             image = PILImage.open(fh)
          except:
             return False
          image.thumbnail((400,400),PILImage.ANTIALIAS)
          fh.close()
          thumb_name, thumb_extension = os.path.splitext(self.Image.name)
          thumb_extension = thumb_extension.lower()

          thumb_filename = thumb_name + '_thumb' + thumb_extension

          if thumb_extension in ['.jpg', '.jpeg']:
             FTYPE = 'JPEG'
          elif thumb_extension == '.gif':
               FTYPE = 'GIF'
          elif thumb_extension == '.png':
               FTYPE = 'PNG'
          else:
             return False    # Unrecognized file type

    # Save thumbnail to in-memory file as StringIO
         temp_thumb = StringIO()
         image.save(temp_thumb, FTYPE)
         temp_thumb.seek(0)

    # Load a ContentFile into the thumbnail field so it gets saved
         self.thumbnail.save(thumb_filename, ContentFile(temp_thumb.read()), save=True)
         temp_thumb.close()

         return True

admin.py : 

@admin.register(Image)
class ImageAdmin(admin.ModelAdmin):
      readonly_fields=['image_number','uploaded_at']
      fields = ['title','image_number','shoot','category',
         'image','image_keyword','credit','license_type','location','uploaded_at']

сейчас это ошибка: 

string argument expected, got 'bytes'

трассировка: 

File "/home/tboss/Desktop/environment/live/backend/venv/lib/python3.7/site-packages/PIL/JpegImagePlugin.py", line 779, in _save
ImageFile._save(im, fp, [("jpeg", (0, 0) + im.size, 0, rawmode)], bufsize)
File "/home/tboss/Desktop/environment/live/backend/venv/lib/python3.7/site-packages/PIL/ImageFile.py", line 513, in _save
fp.write(d)
TypeError: string argument expected, got 'bytes'

когда я пользователь BytesIO:

1014 * я получаю это ошибка: 
maximum recursion depth exceeded in comparison

трассировка: 

 File "/home/tboss/Desktop/environment/live/backend/venv/lib/python3.7/site-packages/PIL/TiffImagePlugin.py", line 319, in __init__
  if isinstance(value, Fraction):
 File "/home/tboss/Desktop/environment/live/backend/venv/lib/python3.7/abc.py", line 139, in __instancecheck__
 return _abc_instancecheck(cls, instance)

RecursionError: максимальная глубина рекурсии превышен по сравнению

1022 * я ток загрузка изображения от администратора, но и собираюсь с помощью ФПИ. в настоящее время не создает миниатюру ............................................. .................................................. ..............

1 Ответ

1 голос
/ 08 января 2020

Как сказано в комментариях, вы можете работать с кодировкой Base64 изображения вместо. Для того, чтобы сделать это, сначала необходимо изменить поле эскиза к следующему: 

thumbnail = models.CharField(max_length=2000, blank=True, null=True)

Затем переопределить модель метод сохранения и добавить следующий код: 

def save(self, force_insert=False, force_update=False, using=None, update_fields=None):
    if not self.image:
        self.thumbnail = None
    else:
        thumbnail_size = 50, 50
        data_img = BytesIO()
        tiny_img = Image.open(self.image)
        tiny_img.thumbnail(thumbnail_size)
        tiny_img.save(data_img, format="BMP")
        tiny_img.close()
        try:
            self.thumbnail = "data:image/jpg;base64,{}".format(
                base64.b64encode(data_img.getvalue()).decode("utf-8")
            )
        except UnicodeDecodeError:
            self.blurred_image = None

    super(Image, self).save(force_insert, force_update, using, update_fields)

Вы можете изменить THUMBNAIL_SIZE переменные в соответствии с вашими потребностями. Наконец, добавьте следующую строку в раздел импорта: 

import base64
from io import BytesIO

Не забудьте запустить makemigrations и перенести команды! Дайте мне знать, если у вас есть какие-либо проблемы.

...