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 link_document_ocr_download, link_document_submit
), ),
sources=( sources=(
'ocr:document_submit', 'ocr:document_ocr_content', 'ocr:document_ocr_content_delete',
'ocr:document_ocr_download', 'ocr:document_ocr_error_list' 'ocr:document_ocr_content', 'ocr:document_ocr_download',
'ocr:document_ocr_error_list', 'ocr:document_submit',
) )
) )
menu_secondary.bind_links( menu_secondary.bind_links(

View File

@@ -22,7 +22,7 @@ logger = logging.getLogger(__name__)
class DocumentPageOCRContentManager(models.Manager): class DocumentPageOCRContentManager(models.Manager):
def delete_ocr_content_for(self, document, user=None): def delete_content_for(self, document, user=None):
with transaction.atomic(): with transaction.atomic():
for document_page in document.pages.all(): for document_page in document.pages.all():
self.filter(document_page=document_page).delete() 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): class OCRViewsTestCase(OCRViewTestMixin, GenericDocumentViewTestCase):
# PyOCR's leak descriptor in get_available_languages and image_to_string # 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): def test_document_type_ocr_settings_view_no_permission(self):
response = self._request_document_type_ocr_settings_view() response = self._request_document_type_ocr_settings_view()
self.assertEqual(response.status_code, 404) self.assertEqual(response.status_code, 404)
def test_document_type_ocr_settings_view_with_access(self): def test_document_type_ocr_settings_view_with_access(self):
self.grant_access( self.grant_access(
obj=self.test_document.document_type, obj=self.test_document_type,
permission=permission_document_type_ocr_setup permission=permission_document_type_ocr_setup
) )

View File

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