54 lines
2.0 KiB
Python
54 lines
2.0 KiB
Python
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
from django.utils.translation import ugettext
|
|
from django.core.exceptions import ObjectDoesNotExist
|
|
|
|
from documents.models import Document
|
|
|
|
from ocr.literals import DOCUMENTQUEUE_STATE_STOPPED, \
|
|
DOCUMENTQUEUE_STATE_CHOICES, QUEUEDOCUMENT_STATE_PENDING, \
|
|
QUEUEDOCUMENT_STATE_CHOICES
|
|
from ocr.manager import DocumentQueueManager
|
|
|
|
|
|
class DocumentQueue(models.Model):
|
|
name = models.CharField(max_length=64, unique=True, verbose_name=_(u'name'))
|
|
label = models.CharField(max_length=64, verbose_name=_(u'label'))
|
|
state = models.CharField(max_length=4,
|
|
choices=DOCUMENTQUEUE_STATE_CHOICES,
|
|
default=DOCUMENTQUEUE_STATE_STOPPED,
|
|
verbose_name=_(u'state'))
|
|
|
|
objects = DocumentQueueManager()
|
|
|
|
class Meta:
|
|
verbose_name = _(u'document queue')
|
|
verbose_name_plural = _(u'document queues')
|
|
|
|
def __unicode__(self):
|
|
return self.label
|
|
|
|
|
|
class QueueDocument(models.Model):
|
|
document_queue = models.ForeignKey(DocumentQueue, verbose_name=_(u'document queue'))
|
|
document = models.ForeignKey(Document, verbose_name=_(u'document'))
|
|
datetime_submitted = models.DateTimeField(verbose_name=_(u'date time submitted'), auto_now_add=True, db_index=True)
|
|
delay = models.BooleanField(verbose_name=_(u'delay ocr'), default=False)
|
|
state = models.CharField(max_length=4,
|
|
choices=QUEUEDOCUMENT_STATE_CHOICES,
|
|
default=QUEUEDOCUMENT_STATE_PENDING,
|
|
verbose_name=_(u'state'))
|
|
result = models.TextField(blank=True, null=True, verbose_name=_(u'result'))
|
|
node_name = models.CharField(max_length=32, verbose_name=_(u'node name'), blank=True, null=True)
|
|
|
|
class Meta:
|
|
ordering = ('datetime_submitted',)
|
|
verbose_name = _(u'queue document')
|
|
verbose_name_plural = _(u'queue documents')
|
|
|
|
def __unicode__(self):
|
|
try:
|
|
return unicode(self.document)
|
|
except ObjectDoesNotExist:
|
|
return ugettext(u'Missing document.')
|