Cleanup OCR app code and link registration

Signed-off-by: Roberto Rosario <roberto.rosario@mayan-edms.com>
This commit is contained in:
Roberto Rosario
2019-09-24 15:37:18 -04:00
parent d7818b64b0
commit b03758ed8b
4 changed files with 20 additions and 11 deletions

View File

@@ -157,8 +157,9 @@ class OCRApp(MayanAppConfig):
link_document_ocr_download, link_document_submit
),
sources=(
'ocr:document_submit', 'ocr:document_ocr_content',
'ocr:document_ocr_download', 'ocr:document_ocr_error_list'
'ocr:document_ocr_content_delete',
'ocr:document_ocr_content', 'ocr:document_ocr_download',
'ocr:document_ocr_error_list', 'ocr:document_submit',
)
)
menu_secondary.bind_links(

View File

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
class DocumentPageOCRContentManager(models.Manager):
def delete_ocr_content_for(self, document, user=None):
def delete_content_for(self, document, user=None):
with transaction.atomic():
for document_page in document.pages.all():
self.filter(document_page=document_page).delete()

View File

@@ -56,12 +56,6 @@ class OCRViewTestMixin(object):
}
)
def _request_document_type_ocr_settings_view(self):
return self.get(
viewname='ocr:document_type_ocr_settings',
kwargs={'pk': self.test_document.document_type.pk}
)
class OCRViewsTestCase(OCRViewTestMixin, GenericDocumentViewTestCase):
# PyOCR's leak descriptor in get_available_languages and image_to_string
@@ -194,13 +188,27 @@ class OCRViewsTestCase(OCRViewTestMixin, GenericDocumentViewTestCase):
),
)
class DocumentTypeOCRViewTestMixin(object):
def _request_document_type_ocr_settings_view(self):
return self.get(
viewname='ocr:document_type_ocr_settings',
kwargs={'pk': self.test_document_type.pk}
)
class DocumentTypeOCRViewsTestCase(
DocumentTypeOCRViewTestMixin, GenericDocumentViewTestCase
):
auto_upload_document = False
def test_document_type_ocr_settings_view_no_permission(self):
response = self._request_document_type_ocr_settings_view()
self.assertEqual(response.status_code, 404)
def test_document_type_ocr_settings_view_with_access(self):
self.grant_access(
obj=self.test_document.document_type,
obj=self.test_document_type,
permission=permission_document_type_ocr_setup
)

View File

@@ -46,7 +46,7 @@ class DocumentOCRContentDeleteView(MultipleObjectConfirmActionView):
return result
def object_action(self, form, instance):
DocumentPageOCRContent.objects.delete_ocr_content_for(
DocumentPageOCRContent.objects.delete_content_for(
document=instance, user=self.request.user
)