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

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

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

先决条件

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

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

如果您没有安装 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']

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

“上传的图像必须是矩形（不是正方形）。”

测试验证器

您可以通过尝试上传方形和矩形图像来测试功能。

1. 方形图像（例如 300x300）:
   验证器将拒绝该文件并引发 ValidationError。

2. 矩形图像（例如 400x300）:
   验证器将接受该文件，并且图像将成功上传。

最后的注释

通过使用这种方法，您可以在 Django 应用程序中无缝地强制执行图像尺寸要求。 Pillow 库可以轻松处理图像大小，而 Django 的验证系统可以让您轻松集成自定义逻辑。

要点：

• 使用 Pillow 提取图像尺寸。
• 当上传的图像不符合您的标准时引发验证错误。
• 将验证器集成到 Django 模型中以确保数据完整性。

通过结合 Django 和 Pillow，您可以创建强大且灵活的图像上传规则，从而提高 Web 应用程序的质量。

编码愉快！ ？

以上是如何使用 Python 验证 Django 中的矩形图像的详细内容。更多信息请关注PHP中文网其他相关文章！