当前位置: 代码迷 >> python >> Django导入Excel端点
  详细解决方案

Django导入Excel端点

热度:74   发布时间:2023-07-16 11:22:04.0

我正在尝试创建一个端点,用于将excel文件加载到服务器中并在其中导入数据。 为此,我正在使用软件包。 这是视图代码:

    def create(self, request, *args, **kwargs):
    file_serializer = self.get_serializer(data=request.data)
    file_serializer.is_valid(raise_exception=True)
    if file_serializer.is_valid():
        from tablib import Dataset
        from workflows.submittals.admin import ItemImportResource
        item_model_resource = ItemImportResource()
        file_serializer.save()
        dataset = Dataset()
        file_obj = request.FILES['file']

        imported_data = Dataset().load(open(file_obj).read())

        result = item_model_resource.import_data(dataset, dry_run=True) 
        if not result.has_errors():
            item_model_resource.import_data(dataset, dry_run=False)  

        return Response(file_serializer.data, status=status.HTTP_201_CREATED)
    else:
        return Response(file_serializer.errors, status=status.HTTP_400_BAD_REQUEST)

将文件加载到(tablib)数据集中时,出现此错误:

'invalid file: <InMemoryUploadedFile: Import_Sample_test1.xls (application/vnd.ms-excel)>'

我尝试重命名文件的名称,所以替换了

imported_data = Dataset().load(open(file_obj).read())

imported_data = Dataset().load(open(file_obj.name).read())

然后好像数据集确实加载了文件,因为在响应中我看到了文件的某些字节表示形式,但是我也得到了以下错误消息:

'ascii' codec can't decode byte 0xd0 in position 0: ordinal not in range(128)

任何想法?

更新:我的ItemImportModel模型:

class ItemImportModel(models.Model):

spec_section_identifier = models.CharField('Spec Section #', max_length=15)
spec_section_name = models.CharField('Spec Section Name',max_length=200)
sub_spec_section = models.CharField('Sub Spec Section', max_length=200, null=True, blank=True)
title = models.CharField(max_length=200)
description = models.TextField(null=True, blank=True, default="")
type = models.CharField(max_length=50, choices=Item.ITEM__TYPES)

我要导入的文件:

主要问题是dataset.load部分。 显然,对不同类型的文件应该有不同的处理方式。
这终于对我有用

对于csv-

imported_data = dataset.load(open(file_obj.name).read())

对于xls-

imported_data = dataset.load(open(file_obj.name, 'rb').read(), 'xls')

对于xlsx-

imported_data = dataset.load(open(file_obj.name, 'rb').read(), 'xlsx')
  相关解决方案