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 )