Django保存图片时如何裁剪为正方形并重命名为随机数？

问题说明 #

需要在Django模型保存时实现两个需求：

1. 将上传图片裁剪为正方形，且不拉伸变形
2. 把图片文件名改为随机数

目前存在两个问题：

• 现有代码的尺寸调整逻辑无法实现正方形裁剪，只会按比例缩放
• 使用os.rename重命名时出现[WinError 32]错误（文件被占用）

现有代码 #

模型代码 #

user = models.OneToOneField(User, on_delete=models.CASCADE)
bio = models.TextField(null=True, blank=True)
image = models.ImageField(default='/profile_pics/default.jpg',
                          null=True,
                          blank=True,
                          upload_to='profile_pics')

def __str__(self):
    return f'{self.user.username} Profile'

def save(self, **kwargs):
    super().save()
    new_name = random.randint(1, 236325984)

    img = Image.open(self.image.path)

    if img.height > 300 or img.width > 300:
        output_size = (200, 200)
        img.thumbnail(output_size)

    if img.height < 299 or img.width < 299:
        output_size = (200, 200)
        img.thumbnail(output_size)

    img.save(self.image.path)

尝试的重命名代码 #

new_name = random.randint(1, 236325984)
name = f'{new_name}.{img.format.lower()}'
os.rename(self.image.path, os.path.join(os.path.dirname(self.image.path), name))

解决方案 #

1. 实现正方形无拉伸裁剪 #

thumbnail方法只会按比例缩放图片，无法实现正方形裁剪。需要先计算居中裁剪区域，裁剪后再缩放至目标尺寸：

def save(self, **kwargs):
    super().save(**kwargs)  # 先执行父类保存逻辑，确保文件已上传
    
    if not self.image:
        return  # 跳过默认图或空图的处理
    
    img = Image.open(self.image.path)
    width, height = img.size
    
    # 计算居中正方形裁剪区域
    min_dim = min(width, height)
    left = (width - min_dim) // 2
    top = (height - min_dim) // 2
    right = left + min_dim
    bottom = top + min_dim
    
    # 执行裁剪
    img = img.crop((left, top, right, bottom))
    
    # 缩放至200x200
    img.thumbnail((200, 200))
    
    # 保存修改后的图片
    img.save(self.image.path)

2. 解决重命名WinError32问题 #

WinError32是因为Image.open打开文件后未关闭，文件处于占用状态导致无法重命名。更规范的做法是利用Django的upload_to自定义函数，在文件上传时直接生成随机文件名：

# 在模型文件顶部定义文件名生成函数
def get_random_profile_pic_name(instance, filename):
    # 提取原文件后缀
    ext = filename.split('.')[-1]
    # 生成随机文件名
    random_name = random.randint(1, 236325984)
    # 返回相对于MEDIA_ROOT的保存路径
    return f'profile_pics/{random_name}.{ext}'

# 修改ImageField的upload_to参数
image = models.ImageField(default='/profile_pics/default.jpg',
                          null=True,
                          blank=True,
                          upload_to=get_random_profile_pic_name)

这样文件在上传阶段就会自动使用随机文件名，避免了后续重命名的文件占用问题。

完整修正后的模型代码 #

from django.db import models
from django.contrib.auth.models import User
from PIL import Image
import random

def get_random_profile_pic_name(instance, filename):
    ext = filename.split('.')[-1]
    random_name = random.randint(1, 236325984)
    return f'profile_pics/{random_name}.{ext}'

class Profile(models.Model):
    user = models.OneToOneField(User, on_delete=models.CASCADE)
    bio = models.TextField(null=True, blank=True)
    image = models.ImageField(default='/profile_pics/default.jpg',
                              null=True,
                              blank=True,
                              upload_to=get_random_profile_pic_name)

    def __str__(self):
        return f'{self.user.username} Profile'

    def save(self, **kwargs):
        super().save(**kwargs)
        
        if not self.image:
            return
        
        img = Image.open(self.image.path)
        width, height = img.size
        
        # 裁剪为居中正方形
        min_dim = min(width, height)
        left = (width - min_dim) // 2
        top = (height - min_dim) // 2
        right = left + min_dim
        bottom = top + min_dim
        
        img = img.crop((left, top, right, bottom))
        
        # 缩放至200x200
        img.thumbnail((200, 200))
        
        img.save(self.image.path)

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