Logo Questions Linux Laravel Mysql Ubuntu Git Menu
 

How to import django-taggit tag in django-import-export

I can't import Django-taggit tags using Django-import-export.

This error is when the value is entered.

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

Also, this error is when the value is blank.

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

I also posted question in This issue.

xlsx table there is an id column too.

enter image description here

models.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

admin.py

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
like image 458
spacekoki Avatar asked Nov 19 '25 12:11

spacekoki


1 Answers

My solution:

Custom widget:

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
        ]

Then we have to override field as well:

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)

And then to use in the resource like that:

    tags = cure_widgets.TagFieldImport(
        attribute="tags",
        column_name="tags",
        widget=cure_widgets.TagWidget(Tag, separator=", ")
    )
like image 72
Aleksandra Rievva Avatar answered Nov 22 '25 00:11

Aleksandra Rievva