def ready(self):
super(DocumentSignaturesApp, self).ready()
DocumentVersion.register_post_save_hook(
1, document_version_post_save_hook)
DocumentVersion.register_pre_open_hook(1, document_pre_open_hook)
ModelPermission.register(model=Document,
permissions=(
permission_document_verify,
permission_signature_delete,
permission_signature_download,
permission_signature_upload,
))
menu_facet.bind_links(links=(link_document_verify, ),
sources=(Document, ))
menu_sidebar.bind_links(
links=(link_document_signature_upload,
link_document_signature_download,
link_document_signature_delete),
sources=('signatures:document_verify',
'signatures:document_signature_upload',
'signatures:document_signature_download',
'signatures:document_signature_delete'))
def add_version_to_doc(self, _doc, file):
_max_version = len(_doc.document_versions.all()) # DocumentVersion.objects.filter(document=_doc).count()
_version = DocumentVersion()
_version.version_number = _max_version + 1
_version.document = _doc
_version.relatedFile.save(os.path.basename(file.name), File(file))
_version.save()
def version_fields(self, document):
self.fields["version_update"] = forms.ChoiceField(
label=_(u"Version update"), choices=DocumentVersion.get_version_update_choices(document.latest_version)
)
self.fields["release_level"] = forms.ChoiceField(
label=_(u"Release level"), choices=RELEASE_LEVEL_CHOICES, initial=RELEASE_LEVEL_FINAL
)
self.fields["serial"] = forms.IntegerField(
label=_(u"Release level serial"), initial=0, widget=forms.widgets.TextInput(attrs={"style": "width: auto;"})
)
self.fields["comment"] = forms.CharField(
label=_(u"Comment"), required=False, widget=forms.widgets.Textarea(attrs={"rows": 4})
)
def version_fields(self, document):
self.fields['version_update'] = forms.ChoiceField(
label=_(u'Version update'),
choices=DocumentVersion.get_version_update_choices(
document.latest_version))
self.fields['release_level'] = forms.ChoiceField(
label=_(u'Release level'),
choices=RELEASE_LEVEL_CHOICES,
initial=RELEASE_LEVEL_FINAL,
)
self.fields['serial'] = forms.IntegerField(
label=_(u'Release level serial'),
initial=0,
widget=forms.widgets.TextInput(attrs={'style': 'width: auto;'}),
)
self.fields['comment'] = forms.CharField(
label=_(u'Comment'),
required=False,
widget=forms.widgets.Textarea(attrs={'rows': 4}),
)
def document_version_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 pk: %s', instance.pk)
if instance.document.document_type.ocr:
instance.submit_for_ocr()
Document.add_to_class('submit_for_ocr', document_ocr_submit)
DocumentVersion.add_to_class('submit_for_ocr', document_version_ocr_submit)
class_permissions(Document, [PERMISSION_OCR_DOCUMENT])
register_tool(link_entry_list)
APIEndPoint('ocr')
register_model_list_columns(DocumentVersionOCRError, [
{
'name': _('Document'), 'attribute': encapsulate(lambda entry: document_link(entry.document_version.document))
},
{
'name': _('Added'), 'attribute': 'datetime_submitted'
},
{
except GPGDecryptionError:
# At least return the original raw content
descriptor.seek(0)
return descriptor
else:
descriptor.close()
return StringIO(result.data)
else:
return descriptor
def document_post_save_hook(instance):
if not instance.pk:
document_signature, created = DocumentVersionSignature.objects.get_or_create(
document_version=instance.latest_version,
)
register_links(Document, [document_verify], menu_name='form_header')
register_links(['signatures:document_verify', 'signatures:document_signature_upload', 'signatures:document_signature_download', 'signatures:document_signature_delete'], [document_signature_upload, document_signature_download, document_signature_delete], menu_name='sidebar')
DocumentVersion.register_pre_open_hook(1, document_pre_open_hook)
DocumentVersion.register_post_save_hook(1, document_post_save_hook)
class_permissions(Document, [
PERMISSION_DOCUMENT_VERIFY,
PERMISSION_SIGNATURE_DELETE,
PERMISSION_SIGNATURE_DOWNLOAD,
PERMISSION_SIGNATURE_UPLOAD,
])
def ready(self):
super(OCRApp, self).ready()
APIEndPoint(app=self, version_string='1')
Document.add_to_class('submit_for_ocr', document_ocr_submit)
DocumentVersion.add_to_class(
'submit_for_ocr', document_version_ocr_submit
)
ModelPermission.register(
model=Document, permissions=(
permission_ocr_document, permission_ocr_content_view
)
)
SourceColumn(
source=DocumentVersionOCRError, label=_('Document'),
func=lambda context: document_link(context['object'].document_version.document)
)
SourceColumn(
source=DocumentVersionOCRError, label=_('Added'),
attribute='datetime_submitted'
)
SourceColumn(
source=DocumentVersionOCRError, label=_('Result'),
attribute='result'
)
app.conf.CELERY_QUEUES.append(
Queue('ocr', Exchange('ocr'), routing_key='ocr'),
)
app.conf.CELERY_ROUTES.update(
{
'ocr.tasks.task_do_ocr': {
'queue': 'ocr'
},
}
)
document_search.add_model_field(
field='versions__pages__ocr_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_entry_list,),
sources=(
'ocr:entry_list', 'ocr:entry_delete_multiple',
'ocr:entry_re_queue_multiple', DocumentVersionOCRError
)
)
menu_tools.bind_links(
links=(
link_document_submit_all, link_document_type_submit,
link_entry_list
)
)
post_save.connect(
initialize_new_ocr_settings,
dispatch_uid='initialize_new_ocr_settings', sender=DocumentType
)
post_version_upload.connect(
post_version_upload_ocr, dispatch_uid='post_version_upload_ocr',
sender=DocumentVersion
)
namespace = PropertyNamespace('ocr', _('OCR'))
try:
pdftotext = sh.Command(setting_pdftotext_path.value)
except sh.CommandNotFound:
namespace.add_property(
'pdftotext', _('pdftotext version'), _('not found'),
report=True
)
except Exception:
namespace.add_property(
'pdftotext', _('pdftotext version'),
_('error getting version'), report=True
)
else:
namespace.add_property(
'pdftotext', _('pdftotext version'), pdftotext('-v').stderr,
report=True
)
try:
tesseract = sh.Command(setting_tesseract_path.value)
except sh.CommandNotFound:
namespace.add_property(
'tesseract', _('tesseract version'), _('not found'),
report=True
)
except Exception:
namespace.add_property(
'tesseract', _('tesseract version'),
_('error getting version'), report=True
)
else:
namespace.add_property(
'tesseract', _('tesseract version'), tesseract('-v').stderr,
report=True
)
else:
return descriptor
def document_post_save_hook(instance):
if not instance.pk:
document_signature, created = DocumentVersionSignature.objects.get_or_create(
document_version=instance.latest_version, )
register_links(Document, [document_verify], menu_name='form_header')
register_links([
'signatures:document_verify', 'signatures:document_signature_upload',
'signatures:document_signature_download',
'signatures:document_signature_delete'
], [
document_signature_upload, document_signature_download,
document_signature_delete
],
menu_name='sidebar')
DocumentVersion.register_pre_open_hook(1, document_pre_open_hook)
DocumentVersion.register_post_save_hook(1, document_post_save_hook)
class_permissions(Document, [
PERMISSION_DOCUMENT_VERIFY,
PERMISSION_SIGNATURE_DELETE,
PERMISSION_SIGNATURE_DOWNLOAD,
PERMISSION_SIGNATURE_UPLOAD,
])
def ready(self):
super(OCRApp, self).ready()
APIEndPoint(app=self, version_string='1')
Document.add_to_class('submit_for_ocr', document_ocr_submit)
DocumentVersion.add_to_class('submit_for_ocr',
document_version_ocr_submit)
ModelPermission.register(model=Document,
permissions=(permission_ocr_document,
permission_ocr_content_view))
SourceColumn(source=DocumentVersionOCRError,
label=_('Document'),
func=lambda context: document_link(context[
'object'].document_version.document))
SourceColumn(source=DocumentVersionOCRError,
label=_('Added'),
attribute='datetime_submitted')
SourceColumn(source=DocumentVersionOCRError,
label=_('Result'),
attribute='result')
app.conf.CELERY_QUEUES.append(
Queue('ocr', Exchange('ocr'), routing_key='ocr'), )
app.conf.CELERY_ROUTES.update({
'ocr.tasks.task_do_ocr': {
'queue': 'ocr'
},
})
document_search.add_model_field(
field='versions__pages__ocr_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_entry_list, ),
sources=('ocr:entry_list', 'ocr:entry_delete_multiple',
'ocr:entry_re_queue_multiple', DocumentVersionOCRError))
menu_tools.bind_links(links=(link_document_submit_all,
link_document_type_submit,
link_entry_list))
post_save.connect(initialize_new_ocr_settings,
dispatch_uid='initialize_new_ocr_settings',
sender=DocumentType)
post_version_upload.connect(post_version_upload_ocr,
dispatch_uid='post_version_upload_ocr',
sender=DocumentVersion)
namespace = PropertyNamespace('ocr', _('OCR'))
try:
pdftotext = sh.Command(setting_pdftotext_path.value)
except sh.CommandNotFound:
namespace.add_property('pdftotext',
_('pdftotext version'),
_('not found'),
report=True)
except Exception:
namespace.add_property('pdftotext',
_('pdftotext version'),
_('error getting version'),
report=True)
else:
namespace.add_property('pdftotext',
_('pdftotext version'),
pdftotext('-v').stderr,
report=True)
try:
tesseract = sh.Command(setting_tesseract_path.value)
except sh.CommandNotFound:
namespace.add_property('tesseract',
_('tesseract version'),
_('not found'),
report=True)
except Exception:
namespace.add_property('tesseract',
_('tesseract version'),
_('error getting version'),
report=True)
else:
namespace.add_property('tesseract',
_('tesseract version'),
tesseract('-v').stderr,
report=True)
def document_version_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 pk: %s', instance.pk)
if instance.document.document_type.ocr:
instance.submit_for_ocr()
Document.add_to_class('submit_for_ocr', document_ocr_submit)
DocumentVersion.add_to_class('submit_for_ocr', document_version_ocr_submit)
class_permissions(Document, [PERMISSION_OCR_DOCUMENT])
register_tool(link_entry_list)
APIEndPoint('ocr')
register_model_list_columns(DocumentVersionOCRError, [
{
'name':
_('Document'),
'attribute':
encapsulate(
lambda entry: document_link(entry.document_version.document))
},
