将数据从Django FileUploadHandler保存到模型中

Question

我创建了一个Django的FileUploadHandler子类，用于对上传文件中的行进行计数。 我想将此计数保存在模型字段中。

我有什么选择？

我知道我有权访问FileUploadHandler中的request - 我是否应该将行数粘贴到会话中？

Answer 1

我找到解决的办法是坚持聚合信息到返回的类文件对象：

class TemporaryFileUploadHandler(DjangoTemporaryFileUploadHandler): 
    def new_file(self, *args, **kwargs): 
     super(TemporaryFileUploadHandler, self).new_file(*args, **kwargs) 
     self.line_count = 1 

    def receive_data_chunk(self, raw_data, start): 
     self.file.write(raw_data) 
     self.line_count += raw_data.count(b'\n') 

    def file_complete(self, file_size): 
     fobj = super(TemporaryFileUploadHandler, self).file_complete(file_size) 
     setattr(fobj, 'line_count', self.line_count) 
     return fobj