Django Импорт-экспорт

Я использую Django import-export для добавления большой электронной таблицы в модель. Сам импорт работает отлично, но я хочу добавить два дополнительных столбца в набор данных, которые определяют имя файла excel и дату его загрузки.

Вот моя модель

from django.db import models


class DataImport(models.Model):
    Column1 = models.CharField(max_length=15, null=True)
    Column2 = models.CharField(max_length=50, null=True)
    Column3 = models.CharField(max_length=25, null=True)
    Column4 = models.CharField(max_length=150, null=True)
    FileName = models.CharField(max_length=150, null=True)
    UploadDate = models.CharField(max_length=150, null=True)



    def __str__(self):
        return str(self.Column1)

Мой класс ресурсов

from dashboard.models import DataImport
from import_export import resources, fields
from import_export.fields import Field


class DataImportResource(resources.ModelResource):
    Column1 = Field(attribute='Column1', column_name="Column_1")
    Column2 = Field(attribute="Column2", column_name="Column_2")
    Column3 = Field(attribute='Column3', column_name="Column_3")
    Column4 = Field(attribute='Column4', column_name="Column_4")


    def skip_row(self, instance, original):
        if instance.Column1.strip() != 'value':
            return True
        if instance.Column2.strip() not in {'value1, 'value2', 'value3', 'value4'}:
            return True
        else:
            return False

    class Meta:
        model = DataImport

и мой класс администраторов

class DataImportAdmin(ImportExportModelAdmin):
    resource_class = DataImportResource

admin.site.register(DataImport, DataImportAdmin)

в идеале я хотел бы, чтобы пользователь выбрал файл excel, а затем выбрал месяц из списка выбора, который затем попадет в поле "UploadDate" в базе данных. из-за дублирующего (но необходимого) характера данных мне нужно сделать это для каждой строки в файле excel. любые предложения о том, как добавить пользовательские поля к каждой строке в выгрузке?

Вернуться на верх