Django中处理PIL图片上传与保存的完整流程

核心处理流程

在Django项目中处理图片时，通常面临一个类型转换问题：前端上传的文件在后台接收时是 django.core.files.uploadedfile.InMemoryUploadedFile 类型，而PIL库的 Image.open() 需要处理的是 PIL.JpegImagePlugin.JpegImageFile 类型。因此，核心步骤包括将上传文件转为PIL对象、执行图像处理（如裁剪）、再将处理后的结果转回Django可保存的文件对象。

具体实现步骤

1. 获取并转换图片为PIL对象

首先从请求中获取上传的文件，并使用PIL的 Image.open() 方法将其转换为PIL图像对象。这一步是后续所有图像处理操作的基础。

from PIL import Image

# 获取上传的文件对象，类型为 InMemoryUploadedFile
pic = request.FILES["picture"]

# 转换为 PIL 图像对象
im_pic = Image.open(pic)

2. 执行图片处理（以正方形裁剪为例）

假设我们需要将图片裁剪为正方形，并采用特定的比例进行居中裁剪。根据图片的宽高比，计算裁剪框的位置和大小。

w, h = im_pic.size
if w >= h:
  # 宽大于高，按高度裁剪
  w_start = (w-h)*0.618
  box = (w_start, 0, w_start+h, h)
else:
  # 高大于宽，按宽度裁剪
  h_start = (h-w)*0.618
  box = (0, h_start, w, h_start+w)

# 执行裁剪，得到处理后的 PIL 图像对象
region = im_pic.crop(box)

3. 将处理后的PIL对象转回Django文件对象

为了将处理后的图片保存到数据库，需要将其转换回 InMemoryUploadedFile 类型。这可以通过 io.BytesIO 作为中间缓冲区来实现。

from io import BytesIO
from django.core.files.uploadedfile import InMemoryUploadedFile

# 创建内存缓冲区
pic_io = BytesIO()

# 将PIL图像保存到缓冲区，保持原始格式
region.save(pic_io, im_pic.format)

# 将缓冲区内容转换为 InMemoryUploadedFile 对象
pic_file = InMemoryUploadedFile(
  file=pic_io,
  field_name=None,
  name=pic.name,
  content_type=pic.content_type,
  size=pic.size,
  charset=None
)

4. 保存至数据库

最后，将转换后的 pic_file 赋值给模型字段并保存。

from .models import Picture

# 创建模型实例并保存
p = Picture(picture=pic_file)
p.save()

总结

通过上述步骤，我们完成了从Django上传文件到PIL处理再回传保存的完整闭环。关键在于理解 InMemoryUploadedFile 与 PIL.Image 之间的转换机制，以及利用 BytesIO 作为内存缓冲区的技巧。以上就是Django结合PIL处理上传图片并保存入库的详细内容，更多关于Django图片处理的技术资料请关注本站其它相关文章！