def put(self, request, filename):
file_obj = request.data['file']
Document.import_file(file_obj.temporary_file_path(),
username=request.user.username,
file_title=filename)
return Response(status=204)
def put(self, request, filename):
file_obj = request.data['file']
doc = Document.import_file(file_obj.temporary_file_path(),
username=request.user.username,
file_title=filename)
if isinstance(doc, Document):
serializer = DocumentSerializer(doc)
return Response(serializer.data)
return Response(status=200)
def test_import_file(self):
src_file_path = os.path.join(
BASE_DIR, "data", "andromeda.pdf"
)
ret = Document.import_file(
filepath=src_file_path,
start_ocr_async=False,
upload=False
)
self.assertTrue(ret, "Storage Quota reached for the user.")
self.assertEqual(
Document.objects.filter(title="andromeda.pdf").count(),
1,
"Document andromeda.pdf was not created."
)
def test_delete_pages(self):
# Create a document with two pages
src_file_path = os.path.join(
BASE_DIR, "data", "berlin.pdf"
)
doc = Document.import_file(
filepath=src_file_path,
start_ocr_async=False,
file_title="berlin.pdf",
upload=False
)
self.assertEqual(
doc.page_count,
2
)
# initial version of any document is 0
self.assertEqual(
doc.version,
0
)
doc.delete_pages(page_numbers=[1])
self.assertEqual(
doc.page_count,
1
)
self.assertEqual(
doc.page_set.count(),
1
)
# version should have been incremented
self.assertEqual(
doc.version,
1
)
