如何在 django-import-export 中导入 django-taggit 标签

Question

我无法使用 Django-import-export 导入 Django-taggit 标签。

这个错误是在输入值时。

Line number: 1 - invalid literal for int() with base 10: 'def'

此外，此错误是当值为空时。

Line number: 2 - Cannot add <QuerySet []> (<class 'django.db.models.query.QuerySet'>). Expected <class 'django.db.models.base.ModelBase'> or str.

我也在这个问题中发布了问题。

xlsx 表也有一个 id 列。

模型.py

from django.db import models
from django.urls import reverse
from taggit.managers import TaggableManager

class KnowHow(models.Model):    

    author = models.ForeignKey('auth.User',on_delete=models.CASCADE)
    title = models.CharField(max_length=200)
    text = models.TextField(blank=True)
    file = models.FileField(blank=True,upload_to='explicit_knowhows')
    free_tags = TaggableManager(blank=True)

    def __str__(self):
        return self.title

管理文件

from django.contrib import admin
from import_export import resources
from import_export import fields
from import_export.admin import ImportExportModelAdmin

from .models import KnowHow
# Register your models here.

class KnowHowResource(resources.ModelResource):

    class Meta:
        model = KnowHow
        import_id_fields = ['id']

@admin.register(KnowHow)
class knowHowAdmin(ImportExportModelAdmin):
    resource_class = KnowHowResource

Answer 1

我的解决方案：

自定义小部件：

from import_export import fields
from import_export import widgets
from taggit.forms import TagField
from taggit.models import Tag

class TagWidget(widgets.ManyToManyWidget):
    def render(self, value, obj=None):
        return self.separator.join(
            [obj.name for obj in value.all()]
        )

    def clean(self, value, row=None, *args, **kwargs):
        values = TagField().clean(value)
        return [
            Tag.objects.get_or_create(name=tag)[0]
            for tag in values
        ]

然后我们还必须覆盖字段：

class TagFieldImport(fields.Field):

    def save(self, obj, data, is_m2m=False):
        # This method is overridden because originally code
        # getattr(obj, attrs[-1]).set(cleaned, clean=True) doesn't unpack cleaned value
        if not self.readonly:
            attrs = self.attribute.split('__')
            for attr in attrs[:-1]:
                obj = getattr(obj, attr, None)
            cleaned = self.clean(data)
            if cleaned is not None or self.saves_null_values:
                if not is_m2m:
                    setattr(obj, attrs[-1], cleaned)
                else:
                    # Change only here
                    getattr(obj, attrs[-1]).set(*cleaned, clean=True)

然后像这样在资源中使用：

    tags = cure_widgets.TagFieldImport(
        attribute="tags",
        column_name="tags",
        widget=cure_widgets.TagWidget(Tag, separator=", ")
    )

Answer 2

我使用的另一种可能的解决方案/解决方法。

定义一个辅助模型文本字段来存储逗号分隔的标签列表。 我们称之为“tags_char”。 然后覆盖模型的 save 方法，将逗号分隔的列表转换为 Tag 对象。

# models.py
from django.db import models
from taggit.managers import TaggableManager

class Book(models.Model):
    
    tags_char = models.TextField(blank=True)
    tags = TaggableManager(blank=True)
    
    def save(self, *args, **kwargs):
        if ',' in self.tags_char and self.pk:
            for tag in self.tags_char.split(','):
                self.tags.add(tag)
        super(Book, self).save(*args, **kwargs)

最后一步是调整ModelResource。 我们从导入中排除标签，而是从文件中导入到 tags_char。 知道一旦保存对象，转换就会发生。

#admin.py
class BookResource(resources.ModelResource):
    class Meta:
        model = Book
        exclude = ('id', 'tags') 

不幸的是，这种变通方法存在一个问题。 由于 django 导入导出的工作方式，在为对象分配 id (pk) 之前调用 save() 方法。 如果 id 不存在，则无法添加标签。 这就是检查save方法中是否存在self.pk的原因。 if ',' in self.tags_char and self.pk

有三种替代方法可以解决这个问题。

1. 如果您要补充数据或在将其发布到网站之前对其进行审查。 您需要手动重新保存它们 => 问题已解决（我就是这种情况）
2. 只需导入文件两次。 第二次围绕 id 的遗嘱存在。
3. 在要导入的文件中提供一个 ID。