Django上传图片重命名及PDF嵌入图片问题排查与解决

一、修复图片上传重命名不生效问题 #

你的models.py中已定义重命名函数，但两处错误导致逻辑未生效：

1. ImageField未关联重命名函数
   原代码中image = models.ImageField(upload_to='post_images')用字符串作为upload_to参数，需直接传入定义好的post_images函数。

2. 重命名函数逻辑错误

   • instance.post.num_post调用错误，instance本身就是Post对象，直接用instance.num_post即可。
   • 文件名格式不符合需求，需调整为postimg{num_post}.{ext}。

修改后的models.py代码：

import os
import uuid
from django.db import models

def post_images(instance, filename):
    ext = filename.split('.')[-1]
    # 生成postimg{num_post}格式的文件名
    filename = f"postimg{instance.num_post}.{ext}"
    # 指定上传到media/post_images目录
    return os.path.join('post_images', filename)

class Post(models.Model):
    id = models.UUIDField(primary_key=True, default=uuid.uuid4)
    num_post = models.IntegerField(default=0)
    # 关联重命名函数
    image = models.ImageField(upload_to=post_images)
    caption = models.TextField(max_length=300)
    created_at = models.DateTimeField(auto_now_add=True)
    number_of_likes = models.IntegerField(default=0)
    number_of_dislikes = models.IntegerField(default=0)
    
    def __str__(self):
        return self.caption

二、完善PDF生成逻辑，正确嵌入重命名后的图片 #

views.py中PDF生成存在路径错误、图片读取失效问题，修复如下：

1. 修正图片路径获取
   利用Django模型的image.path属性直接获取图片本地绝对路径，避免手动拼接出错。

2. 添加图片存在性校验
   确保图片文件存在后再嵌入，避免触发报错。

3. 优化PDF保存路径
   将PDF统一保存到media/pdf目录（需提前创建，代码中已自动处理目录不存在的情况）。

修改后的views.py代码：

import io
import os
from django.shortcuts import redirect
from reportlab.pdfgen import canvas
from .models import Post

def upload(request):
    if request.method == 'POST':
        image = request.FILES.get('image_upload')
        caption = request.POST.get('caption', '')
        
        # 获取当前最大帖子编号，避免并发场景下编号重复
        max_num = Post.objects.aggregate(max_num=models.Max('num_post'))['max_num'] or 0
        num_post = max_num + 1
        
        # 创建新帖子
        new_post = Post.objects.create(
            image=image,
            caption=caption,
            num_post=num_post
        )
        
        # 生成PDF
        pdf_dir = os.path.join('media', 'pdf')
        os.makedirs(pdf_dir, exist_ok=True)  # 自动创建目录
        pdf_path = os.path.join(pdf_dir, f"post{num_post}.pdf")
        
        p = canvas.Canvas(pdf_path)
        
        # 嵌入图片（仅当图片存在时执行）
        if new_post.image and os.path.exists(new_post.image.path):
            img_file = new_post.image.path
            p.drawImage(img_file, 0, 0, width=120, preserveAspectRatio=True, mask='auto')
        
        # 添加文本内容
        p.drawString(200, 300, new_post.caption)
        p.drawString(200, 100, str(new_post.created_at))
        p.drawString(200, 600, str(new_post.id))
        
        p.showPage()
        p.save()
        
        return redirect('/')
    else:
        return redirect('/')

关键说明 #

• 重命名生效原理：Django的ImageField的upload_to参数支持传入函数，上传文件时会调用该函数，传入当前模型实例和原始文件名，返回新的保存路径与文件名。
• 路径安全保障：使用new_post.image.path获取绝对路径，避免项目根目录变化导致路径错误；通过os.path.exists校验文件存在性，防止上传失败导致PDF生成报错。
• 并发场景优化：采用“最大编号+1”的方式生成num_post，比count()+1更安全（避免删除帖子后编号重复），高并发场景建议搭配数据库事务或F表达式自增。

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