Django中如何检测用户上传的是ImageField还是FileField？

初始实现（存在问题） #

我原本写了下面的模型和函数，想给图片和音频文件设置不同的上传路径：

def upload_to(instance, filename):
    # 尝试判断文件类型并放入对应目录
    if isinstance(instance, models.ImageField):
        print("detected image file")
        return "x/img/%s" % (filename)
    else:
        print("detected non-image file")
        return "x/aud/%s" % (filename)

class Tile(models.Model):
    image = models.ImageField(upload_to=upload_to, default="../static/x/img/default.svg")
    sound = models.FileField(upload_to=upload_to, null=True, blank=True)

问题点 #

很快发现upload_to里的判断逻辑完全错误：这里的instance是包含所有字段的Tile对象实例，不是单个字段的实例，根本没法判断当前上传的是ImageField还是FileField的文件。

改进后的实现 #

参考Warren的建议，我重新实现了方案，简化后代码如下：

def upload_image_to(instance, filename):    
    filename = filenamer_helper(instance, filename)
    return f"x/img/{instance.user.id}/{instance.grid.id}/{filename}"


def upload_audio_to(instance, filename):
    filename = filenamer_helper(instance, filename)
    return f"x/aud/{instance.user.id}/{instance.grid.id}/{filename}"


# 给文件名添加规范化的三位数字前缀
def filenamer_helper(instance, filename):
    needs_tile_number = False
    # 检查文件名开头是否已经包含tile编号
    if f"{instance.tile_number}_" not in filename[0:4]:
        needs_tile_number = True
    # 如果需要，给文件名添加三位数字格式的tile编号前缀
    if needs_tile_number:
        if instance.tile_number < 10:
            filename = f"00{instance.tile_number}_{filename}"
        elif instance.tile_number < 100:
            filename = f"0{instance.tile_number}_{filename}"
        else:
            filename = f"{instance.tile_number}_{filename}"
    return filename
    

class Tile(models.Model):
    user = models.ForeignKey(User, on_delete=models.CASCADE, related_name="user_tiles")
    grid = models.ForeignKey(Grid, on_delete=models.CASCADE, related_name="grid_tiles")
    tile_number = models.IntegerField(default=0)
    image = models.ImageField(upload_to=upload_image_to, default="../static/x/img/default.svg")
    audio = models.FileField(upload_to=upload_audio_to, null=True, blank=True)
    text = models.CharField(max_length=100, null=True, blank=True)

方案说明 #

• 为image和audio字段分别指定独立的upload_to函数，直接区分上传路径，避免了原方案的判断逻辑错误。
• 通过filenamer_helper统一处理文件名：自动检查并添加三位数字格式的tile_number前缀，确保文件名规范统一。
• 上传路径加入了user.id和grid.id，能更好地按用户和网格组织文件，避免文件冲突。

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