Use copyfileobj to save documents to files

Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
This commit is contained in:
Roberto Rosario
2019-04-29 04:28:25 -04:00
parent efff750f06
commit 825af3152f
4 changed files with 11 additions and 18 deletions

View File

@@ -3,6 +3,7 @@ from __future__ import absolute_import, unicode_literals
import hashlib
import logging
import os
import shutil
import uuid
from django.apps import apps
@@ -310,23 +311,13 @@ class DocumentVersion(models.Model):
sender=Document, instance=self.document
)
def save_to_file(self, filepath, buffer_size=1024 * 1024):
def save_to_file(self, file_object):
"""
Save a copy of the document from the document storage backend
to the local filesystem
"""
input_descriptor = self.open()
output_descriptor = open(filepath, 'wb')
while True:
copy_buffer = input_descriptor.read(buffer_size)
if copy_buffer:
output_descriptor.write(copy_buffer)
else:
break
output_descriptor.close()
input_descriptor.close()
return filepath
input_file_object = self.open()
shutil.copyfileobj(fsrc=input_file_object, fdst=file_object)
@property
def size(self):