126 lines
3.9 KiB
Python
126 lines
3.9 KiB
Python
from __future__ import unicode_literals
|
|
|
|
import logging
|
|
|
|
from kombu import Exchange, Queue
|
|
|
|
from django.apps import apps
|
|
from django.db.models.signals import post_save
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from acls import ModelPermission
|
|
from common import (
|
|
MayanAppConfig, menu_facet, menu_multi_item, menu_object, menu_secondary,
|
|
menu_tools
|
|
)
|
|
from common.settings import settings_db_sync_task_delay
|
|
from documents.search import document_search, document_page_search
|
|
from documents.signals import post_version_upload
|
|
from documents.widgets import document_link
|
|
from mayan.celery import app
|
|
from navigation import SourceColumn
|
|
from rest_api.classes import APIEndPoint
|
|
|
|
from .handlers import handler_parse_document_version
|
|
from .links import (
|
|
link_document_content, link_entry_list, link_document_content_errors_list,
|
|
link_document_content_download
|
|
)
|
|
from .permissions import permission_content_view
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class DocumentParsingApp(MayanAppConfig):
|
|
has_tests = True
|
|
name = 'document_parsing'
|
|
verbose_name = _('Document parsing')
|
|
|
|
def ready(self):
|
|
super(DocumentParsingApp, self).ready()
|
|
|
|
APIEndPoint(app=self, version_string='1')
|
|
|
|
Document = apps.get_model(
|
|
app_label='documents', model_name='Document'
|
|
)
|
|
|
|
DocumentType = apps.get_model(
|
|
app_label='documents', model_name='DocumentType'
|
|
)
|
|
|
|
DocumentVersion = apps.get_model(
|
|
app_label='documents', model_name='DocumentVersion'
|
|
)
|
|
|
|
DocumentVersionParseError = self.get_model('DocumentVersionParseError')
|
|
|
|
ModelPermission.register(
|
|
model=Document, permissions=(permission_content_view,)
|
|
)
|
|
|
|
SourceColumn(
|
|
source=DocumentVersionParseError, label=_('Document'),
|
|
func=lambda context: document_link(context['object'].document_version.document)
|
|
)
|
|
SourceColumn(
|
|
source=DocumentVersionParseError, label=_('Added'),
|
|
attribute='datetime_submitted'
|
|
)
|
|
SourceColumn(
|
|
source=DocumentVersionParseError, label=_('Result'),
|
|
attribute='result'
|
|
)
|
|
|
|
document_search.add_model_field(
|
|
field='versions__pages__content__content', label=_('Content')
|
|
)
|
|
|
|
document_page_search.add_model_field(
|
|
field='content__content', label=_('Content')
|
|
)
|
|
|
|
menu_facet.bind_links(
|
|
links=(link_document_content,), sources=(Document,)
|
|
)
|
|
menu_multi_item.bind_links(
|
|
links=(link_document_submit_multiple,), sources=(Document,)
|
|
)
|
|
menu_object.bind_links(
|
|
links=(link_document_submit,), sources=(Document,)
|
|
)
|
|
menu_object.bind_links(
|
|
links=(link_document_type_ocr_settings,), sources=(DocumentType,)
|
|
)
|
|
menu_secondary.bind_links(
|
|
links=(
|
|
link_document_content, link_document_ocr_erros_list,
|
|
link_document_ocr_download
|
|
),
|
|
sources=(
|
|
'document_parsing:document_content',
|
|
'document_parsing:document_ocr_error_list',
|
|
'document_parsing:document_ocr_download',
|
|
)
|
|
)
|
|
menu_secondary.bind_links(
|
|
links=(link_entry_list,),
|
|
sources=(
|
|
'document_parsing:entry_list',
|
|
'document_parsing:entry_delete_multiple',
|
|
'document_parsing:entry_re_queue_multiple',
|
|
DocumentVersionParseError
|
|
)
|
|
)
|
|
menu_tools.bind_links(
|
|
links=(
|
|
link_entry_list
|
|
)
|
|
)
|
|
|
|
post_version_upload.connect(
|
|
dispatch_uid='document_parsing_handler_parse_document_version',
|
|
receiver=handler_parse_document_version,
|
|
sender=DocumentVersion
|
|
)
|