71 lines
3.8 KiB
Python
71 lines
3.8 KiB
Python
from multiprocessing import Queue
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.utils.translation import ugettext
|
|
from django.db.utils import DatabaseError
|
|
from django.db.models.signals import post_save
|
|
|
|
from navigation.api import register_links, register_menu, register_multi_item_links
|
|
from permissions.api import register_permissions
|
|
from documents.models import Document
|
|
|
|
from ocr.conf.settings import AUTOMATIC_OCR
|
|
from models import DocumentQueue, QueueDocument
|
|
from literals import QUEUEDOCUMENT_STATE_PROCESSING, \
|
|
QUEUEDOCUMENT_STATE_PENDING, DOCUMENTQUEUE_STATE_STOPPED, \
|
|
DOCUMENTQUEUE_STATE_ACTIVE
|
|
|
|
#Permissions
|
|
PERMISSION_OCR_DOCUMENT = 'ocr_document'
|
|
PERMISSION_OCR_DOCUMENT_DELETE = 'ocr_document_delete'
|
|
PERMISSION_OCR_QUEUE_ENABLE_DISABLE = 'ocr_queue_enable_disable'
|
|
PERMISSION_OCR_CLEAN_ALL_PAGES = 'ocr_clean_all_pages'
|
|
|
|
register_permissions('ocr', [
|
|
{'name': PERMISSION_OCR_DOCUMENT, 'label': _(u'Submit document for OCR')},
|
|
{'name': PERMISSION_OCR_DOCUMENT_DELETE, 'label': _(u'Delete document for OCR queue')},
|
|
{'name': PERMISSION_OCR_QUEUE_ENABLE_DISABLE, 'label': _(u'Can enable/disable an OCR queue')},
|
|
{'name': PERMISSION_OCR_CLEAN_ALL_PAGES, 'label': _(u'Can execute an OCR clean up on all document pages')},
|
|
])
|
|
|
|
#Links
|
|
submit_document = {'text': _('submit to OCR queue'), 'view': 'submit_document', 'args': 'object.id', 'famfam': 'hourglass_add', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_DOCUMENT]}}
|
|
re_queue_document = {'text': _('re-queue'), 'view': 're_queue_document', 'args': 'object.id', 'famfam': 'hourglass_add', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_DOCUMENT]}}
|
|
re_queue_multiple_document = {'text': _('re-queue'), 'view': 're_queue_multiple_document', 'famfam': 'hourglass_add', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_DOCUMENT]}}
|
|
queue_document_delete = {'text': _(u'delete'), 'view': 'queue_document_delete', 'args': 'object.id', 'famfam': 'hourglass_delete', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_DOCUMENT_DELETE]}}
|
|
queue_document_multiple_delete = {'text': _(u'delete'), 'view': 'queue_document_multiple_delete', 'famfam': 'hourglass_delete', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_DOCUMENT_DELETE]}}
|
|
|
|
document_queue_disable = {'text': _(u'stop queue'), 'view': 'document_queue_disable', 'args': 'object.id', 'famfam': 'control_stop_blue', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_QUEUE_ENABLE_DISABLE]}}
|
|
document_queue_enable = {'text': _(u'activate queue'), 'view': 'document_queue_enable', 'args': 'object.id', 'famfam': 'control_play_blue', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_QUEUE_ENABLE_DISABLE]}}
|
|
|
|
all_document_ocr_cleanup = {'text': _(u'clean up pages content'), 'view': 'all_document_ocr_cleanup', 'famfam': 'text_strikethrough', 'permissions': {'namespace': 'ocr', 'permissions': [PERMISSION_OCR_CLEAN_ALL_PAGES]}}
|
|
|
|
register_links(Document, [submit_document], menu_name='sidebar')
|
|
register_links(DocumentQueue, [document_queue_disable, document_queue_enable])
|
|
|
|
register_multi_item_links(['queue_document_list'], [re_queue_multiple_document, queue_document_multiple_delete])
|
|
|
|
#Menus
|
|
register_menu([
|
|
{'text': _('OCR'), 'view': 'queue_document_list', 'links':[
|
|
#ocr_queue
|
|
], 'famfam': 'hourglass', 'position': 4}])
|
|
|
|
|
|
try:
|
|
default_queue, created = DocumentQueue.objects.get_or_create(name='default')
|
|
if created:
|
|
default_queue.label = ugettext(u'Default')
|
|
default_queue.save()
|
|
except DatabaseError:
|
|
#syncdb
|
|
pass
|
|
|
|
|
|
def document_post_save(sender, instance, **kwargs):
|
|
if kwargs.get('created', False):
|
|
if AUTOMATIC_OCR:
|
|
DocumentQueue.objects.queue_document(instance)
|
|
|
|
post_save.connect(document_post_save, sender=Document)
|