问题描述
我正在尝试创建一个端点,用于将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)
我要导入的文件:
1楼
主要问题是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')