62 lines
2.1 KiB
Python
62 lines
2.1 KiB
Python
from __future__ import absolute_import
|
|
|
|
import logging
|
|
|
|
from django.dispatch import receiver
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from south.signals import post_migrate
|
|
|
|
from acls.api import class_permissions
|
|
from documents.models import Document, DocumentVersion
|
|
from documents.signals import post_version_upload
|
|
from main.api import register_maintenance_links
|
|
from navigation.api import register_links
|
|
from navigation.links import link_spacer
|
|
from project_tools.api import register_tool
|
|
from rest_api.classes import APIEndPoint
|
|
|
|
from .links import (all_document_ocr_cleanup, ocr_tool_link,
|
|
queue_document_list, queue_document_multiple_delete,
|
|
re_queue_multiple_document, submit_document,
|
|
submit_document_multiple)
|
|
from .models import DocumentQueue
|
|
from .permissions import PERMISSION_OCR_DOCUMENT
|
|
from .tasks import task_do_ocr
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
register_links(Document, [submit_document])
|
|
register_links([Document], [link_spacer, submit_document_multiple], menu_name='multi_item_links')
|
|
register_links(['ocr:queue_document_list'], [re_queue_multiple_document, queue_document_multiple_delete])
|
|
register_links(['ocr:queue_document_list'], [queue_document_list], menu_name='secondary_menu')
|
|
|
|
register_maintenance_links([all_document_ocr_cleanup], namespace='ocr', title=_(u'OCR'))
|
|
|
|
|
|
def document_ocr_submit(self):
|
|
task_do_ocr.apply_async(args=[self.pk], queue='ocr')
|
|
|
|
|
|
@receiver(post_version_upload, dispatch_uid='post_version_upload_ocr', sender=DocumentVersion)
|
|
def post_version_upload_ocr(sender, instance, **kwargs):
|
|
logger.debug('received post_version_upload')
|
|
logger.debug('instance: %s', instance)
|
|
if instance.document.document_type.ocr:
|
|
instance.document.submit_for_ocr()
|
|
|
|
|
|
@receiver(post_migrate, dispatch_uid='create_default_queue')
|
|
def create_default_queue_signal_handler(sender, **kwargs):
|
|
if kwargs['app'] == 'ocr':
|
|
DocumentQueue.objects.get_or_create(name='default')
|
|
|
|
|
|
Document.add_to_class('submit_for_ocr', document_ocr_submit)
|
|
|
|
class_permissions(Document, [PERMISSION_OCR_DOCUMENT])
|
|
|
|
register_tool(ocr_tool_link)
|
|
|
|
APIEndPoint('ocr')
|