如何使用 Python 验证 Django 中的矩形图像-Python教程-PHP中文网

如何使用 Python 验证 Django 中的矩形图像

Linda Hamilton

发布： 2024-12-19 11:45:13

原创

208 人浏览过

How to Validate Rectangular Images in Django Using Python

在 Django 项目中处理图像上传时，可能会出现需要强制执行特定尺寸的情况，例如确保上传的图像是矩形（而不是正方形）。这对于个人资料标题、横幅或需要非方形格式的媒体特别有用。

在本文中，我们将使用 Django 验证系统和 Pillow 库逐步介绍一个简单的解决方案。

先决条件

在实施解决方案之前，请确保安装了以下依赖项：

Django（用于网络框架功能）
枕头（用于图像处理）

如果您没有安装 Pillow，您可以使用以下方式添加它：

python -m pip install pillow

登录后复制

编写验证器

要验证上传的图像是否是矩形，我们需要检查图像的宽度和高度。如果两个尺寸相等，则意味着图像是正方形，我们将引发验证错误。

这是自定义验证器的代码：

from django.core.exceptions import ValidationError
from PIL import Image

def validate_rectangular_image(image):
    """
    Validator to ensure an uploaded image is rectangular and not square.
    """
    image = Image.open(image)  # Open the uploaded image using Pillow
    width, height = image.size  # Extract dimensions

    if width == height:  # Check if image is square
        raise ValidationError("Uploaded image must be rectangular (not square).")

    return image

登录后复制

将验证器与 Django 模型集成

要在 Django 应用程序中使用此验证器，您可以将其添加到模型字段中。例如，假设您在用户个人资料横幅的模型中有一个 ImageField：

from django.db import models
from .validators import validate_rectangular_image  # Import the custom validator

class Profile(models.Model):
    name = models.CharField(max_length=100)
    banner_image = models.ImageField(
        upload_to='banners/', 
        validators=[validate_rectangular_image],
        help_text="Please upload a rectangular image for the banner."
    )

    def __str__(self):
        return self.name

登录后复制

运作原理：

每当文件上传到banner_image字段时，都会调用validate_rectangle_image函数。
如果图像是正方形，则会引发 ValidationError，从而阻止保存文件。
只有矩形图片才会通过验证并成功上传。

处理表单中的验证错误

如果您使用 Django 表单进行图像上传，当用户提交无效图像时，将会向用户显示错误。

例如，一个简单的表单可能如下所示：

from django import forms
from .models import Profile

class ProfileForm(forms.ModelForm):
    class Meta:
        model = Profile
        fields = ['name', 'banner_image']

登录后复制

当用户上传方形图片时，他们会看到错误消息：