Django create_image方法图片缩放异常，寻求排查修改方案

Let's walk through the problems in your create_image method and get it working properly:

1. Fix the StringIO vs BytesIO Error #

The first error you're seeing (TypeError: initial_value must be str or None, not PngImageFile) happens because you're using StringIO (designed for text data) to handle binary image content. You need to use BytesIO instead, which is built for binary data.

Update your import from:

from io import StringIO

to:

from io import BytesIO

And replace all instances of StringIO() with BytesIO() in the method.

2. Define the Resized Image Variable #

Your code calculates the new dimensions but never actually resizes the image. Add this line right after calculating the final width and height values:

resizedImage = image.resize((int(width), int(height)), Image.Resampling.LANCZOS)

(Note: If you're using an older PIL/Pillow version, use Image.LANCZOS instead of Image.Resampling.LANCZOS)

3. Add Missing Imports #

You're using hashlib and os but haven't imported them. Add these imports at the top of your create_image method (or at the top of managers.py):

import hashlib
import os
from django.core.files.base import ContentFile

4. Fix File Handling Logic #

Your current file operations are messy and incorrect. Instead of writing to a temporary file, we can use BytesIO to hold the resized image in memory, then convert it to a Django ContentFile that works natively with ImageField.

Here's the revised, cleaned-up create_image method:

class ImageManager(models.Manager):
    def create_image(self, **kwargs):
        from io import BytesIO
        from PIL import Image
        import hashlib
        import os
        from django.core.files.base import ContentFile

        # Validations (raise error)
        missing_args = []
        if 'image' not in kwargs:
            missing_args.append('image')
        if 'album' not in kwargs:
            missing_args.append('album')
        
        if missing_args:
            error_msg = f"create_image() missing {len(missing_args)} required keyword argument{'s' if len(missing_args)>1 else ''}: {', '.join(missing_args)}"
            raise TypeError(error_msg)

        # Downscale image if necessary
        uploaded_file = kwargs['image']
        image = Image.open(uploaded_file)
        width, height = image.size

        # Set max dimensions based on image orientation
        if width > height:
            max_width, max_height = 960, 720
        elif width < height:
            max_width, max_height = 720, 960
        else:
            max_width, max_height = 720, 720

        # Calculate resize ratio (only resize if image is larger than max dimensions)
        ratio = min(max_width / width, max_height / height)
        if ratio < 1:
            new_width = int(width * ratio)
            new_height = int(height * ratio)
            resized_image = image.resize((new_width, new_height), Image.Resampling.LANCZOS)
        else:
            # No resize needed, use original image
            resized_image = image

        # Handle transparency for PNGs when saving as JPEG
        if resized_image.mode in ('RGBA', 'LA'):
            background = Image.new('RGB', resized_image.size, (255, 255, 255))
            background.paste(resized_image, mask=resized_image.split()[-1])
            resized_image = background

        # Save resized image to in-memory BytesIO
        img_io = BytesIO()
        resized_image.save(img_io, format='JPEG', quality=90)
        img_io.seek(0)

        # Generate unique filename
        filename = hashlib.md5(img_io.getvalue()).hexdigest() + '.jpeg'
        # Create Django-compatible ContentFile
        django_file = ContentFile(img_io.getvalue(), name=filename)

        # Create and return the Image object
        return self.create(image=django_file, album=kwargs['album'])

Key Improvements: #

• Uses BytesIO for proper binary image handling
• Adds high-quality resizing with a Lanczos filter
• Handles transparent images (like PNGs) when converting to JPEG
• Eliminates messy temporary file operations using Django's ContentFile
• Cleans up validation logic for better readability
• Skips resizing if the image is already smaller than the maximum allowed dimensions

After applying these changes, your image upload and resizing should work without the TypeError you encountered.

内容的提问来源于stack exchange，提问作者Matt McCarthy