如何修改Django的ImageField使其支持SVG格式文件上传

Django 内置的ImageField依赖 Pillow 库执行图片格式合法性校验，而 Pillow 本身不支持 SVG 矢量图格式识别，因此上传 SVG 时会触发格式校验失败的报错。

方案1：自定义校验规则（推荐，兼顾安全性和兼容性） #

保留原有图片校验逻辑的同时，新增 SVG 格式的合法性校验，步骤如下：

1. 定义自定义校验函数，同时支持普通位图和SVG格式校验：

import os
import xml.etree.ElementTree as ET
from django.core.exceptions import ValidationError
from PIL import Image

def validate_image_and_svg(file):
    file_ext = os.path.splitext(file.name)[1].lower()
    # 针对SVG单独校验
    if file_ext == '.svg':
        try:
            tree = ET.parse(file)
            root = tree.getroot()
            # 校验根标签是否为标准SVG标签
            if root.tag != '{http://www.w3.org/2000/svg}svg':
                raise ValidationError("上传的文件不是合法SVG格式")
        except Exception as e:
            raise ValidationError("SVG文件损坏或格式非法") from e
    # 其他格式走原生Pillow校验逻辑
    else:
        try:
            with Image.open(file) as img:
                img.verify()
        except Exception as e:
            raise ValidationError("上传的文件不是合法图片格式") from e

2. 替换模型中ImageField为FileField，绑定自定义校验器：

import uuid
from django.db import models
from django.contrib.auth import get_user_model

class Recipe(models.Model):
    id = models.UUIDField( 
        primary_key=True,
        default=uuid.uuid4,
        editable=False,
    )
    author = models.ForeignKey(get_user_model() , on_delete=models.CASCADE, null=True, related_name='author') 
    # 替换原有ImageField
    photo_main = models.FileField(
        upload_to='media/', 
        blank=True,
        validators=[validate_image_and_svg]
    )
    title = models.CharField(max_length=150)
    description = models.TextField(blank=True)

3. 调整DRF序列化器配置，避免序列化器层默认的ImageField校验拦截：

from rest_framework import serializers
from .models import Recipe

class RecipeSerializer(serializers.ModelSerializer):
    class Meta:
        model = Recipe
        fields = '__all__'

如果序列化器单独指定了photo_main字段为serializers.ImageField，需要改为serializers.FileField。

方案2：简化后缀校验（适合低安全要求场景） #

如果不需要严格校验文件内容，只校验文件后缀即可，直接修改模型字段如下：

from django.core.validators import FileExtensionValidator

# 其他字段不变
photo_main = models.FileField(
    upload_to='media/', 
    blank=True,
    validators=[FileExtensionValidator(allowed_extensions=['png', 'jpg', 'jpeg', 'svg'])]
)

该方案实现简单，但无法防范篡改后缀的恶意文件上传。

• SVG属于可执行XML格式，存在XSS注入风险，生产环境建议额外添加SVG内容过滤规则，移除文件内的script标签、on*事件属性等恶意内容。
• 部署时需要确保静态资源服务器配置了正确的SVG MIME类型（image/svg+xml），避免前端无法正常渲染SVG资源。

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