如何在 django-import-export 中进行字段验证

Question

以下是我的模型：

class Product(models.Model):
    product_title = models.CharField(max_length=100, null=False, 
verbose_name='Product title')
    product_description = models.TextField(max_length=250, 
verbose_name='Product description')
    product_qty = models.IntegerField(verbose_name='Quantity')
    product_mrp = models.FloatField(verbose_name='Maximum retail price')
    product_offer_price = models.FloatField(verbose_name='Selling price')

我想在保存之前对 product_offer_price 字段进行验证，我已为其发布并用工作解决方案回答了这个问题。

需要验证的是：

 if product_offer_price > product_mrp:
    raise ValidationError

现在上述问题的解决方案非常适用于管理表单。

但是，我已经实现了django-import-export，我在admin中批量导入产品数据，在批量导入时我需要类似的验证。

如何实现？

Answer 1

嗯，这是一个小小的研究过程。

终于明白了。

问题是避免在导入导出库中使用 ProductForm。内部库导入调用实例的方法 save()，但是如果我们在模型（而不是表单）中引发 ValidationError = 500 且 DEBUG = False，并在回溯页面中引发 DEBUG = True。所以我们应该在 import_export 资源中使用 "before_import" 方法，在 django.forms 表单中使用 "clean" 方法。

admin.py

from forms import ProductForm
from models import Product
from import_export import resources
from import_export.admin import ImportExportActionModelAdmin
from django.forms import ValidationError

class ProductResource(resources.ModelResource):

    class Meta:
        model = Product

    def before_import(self, dataset, using_transactions, dry_run, **kwargs):
        for row in dataset:
            if int(row[4]) < int(row[5]):
                raise ValidationError('Product offer price cannot be greater than Product MRP. '
                                      'Error in row with id = %s' % row[0])


class ProductAdmin(ImportExportActionModelAdmin):
    list_display = ('product_title', 'product_description', 'product_qty', 'product_mrp', 'product_offer_price')
    form = ProductForm
    resource_class = ProductResource


admin.site.register(Product, ProductAdmin)

forms.py

from django import forms
from models import Product


class ProductForm(forms.ModelForm):
    class Meta:
        model = Product
        exclude = [id, ]

    def clean(self):
        product_offer_price = self.cleaned_data.get('product_offer_price')
        product_mrp = self.cleaned_data.get('product_mrp')
        if product_offer_price > product_mrp:
            raise forms.ValidationError("Product offer price cannot be greater than Product MRP.")
        return self.cleaned_data

models.py

class Product(models.Model):
    product_title = models.CharField(max_length=100, null=False, verbose_name='Product title')
    product_description = models.TextField(max_length=250, verbose_name='Product description')
    product_qty = models.IntegerField(verbose_name='Quantity')
    product_mrp = models.FloatField(verbose_name='Maximum retail price')
    product_offer_price = models.FloatField(verbose_name='Selling price')

Answer 2

这是 Django Rest Framework 的另一种简单方法

def importcsv(request, company):
        for row in dataset['company']:
            if(row != company):
            raise PermissionDenied("You do not have permission to Enter Clients in Other Company, Be Careful")

Answer 3

一种更简单的方法可能是通过向资源 class:

添加自定义 before_import_row 方法来挂接到 import/export 工作流程

class ProductResource(resources.ModelResource):

    class Meta:
        model = Product

    def before_import_row(self, row, **kwargs):
        if int(row[4]) < int(row[5]):
            raise ValidationError('Product offer price cannot be greater than Product MRP. '
                                  'Error in row with id = %s' % row[0])

如何在 django-import-export 中进行字段验证

How to do field validation in django-import-export

django

django-admin

django-import-export

admin.py

forms.py

models.py