如何使用 Python 驗證 Django 中的矩形圖像-Python教學-PHP中文網

如何使用 Python 驗證 Django 中的矩形圖像

Linda Hamilton

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

原創

191 人瀏覽過

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']

登入後複製

當使用者上傳方形圖片時，他們會看到錯誤訊息：