通过DRF或Django Admin清除用户头像时重置ImageField为默认值

修正模型的save方法 #

你之前重写的save()方法写法有误，不能直接修改avatar.name属性，正确做法是直接给self.avatar赋值默认路径。修改后的模型代码如下：

class Profile(models.Model):
    AVATAR_UPLOAD_TO = "account_avatar/"
    DEFAULT_AVATAR = "default.jpg"

    user = models.OneToOneField(CustomUser, on_delete=models.CASCADE)
    avatar = sorl.thumbnail.ImageField(
        upload_to=AVATAR_UPLOAD_TO,
        default=f"{AVATAR_UPLOAD_TO}{DEFAULT_AVATAR}",
        blank=True,
        null=True
    )

    def save(self, *args, **kwargs):
        # 当avatar为空时，重置为默认值
        if not self.avatar:
            self.avatar = self.avatar.field.default
        super().save(*args, **kwargs)

当用户清除头像后，self.avatar会变为None或空实例，此时直接将其赋值为字段定义的默认值即可。

配合DRF序列化器处理前端请求（若使用DRF） #

如果通过DRF接口处理头像清除操作，需要在序列化器中处理空值，避免前端传空时直接存为None。示例代码：

from rest_framework import serializers
from .models import Profile

class ProfileSerializer(serializers.ModelSerializer):
    class Meta:
        model = Profile
        fields = ['user', 'avatar']

    def update(self, instance, validated_data):
        # 检测到avatar为空时，替换为默认值
        if 'avatar' in validated_data and validated_data['avatar'] is None:
            validated_data['avatar'] = instance.avatar.field.default
        return super().update(instance, validated_data)

额外注意事项 #

• 确保default.jpg文件存在于MEDIA_ROOT/account_avatar/目录下，否则会出现图片加载失败的情况。
• 若使用Django Admin后台，清除头像后点击保存，模型的save()方法会自动触发，将头像路径重置为默认值。

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