Moved DocumentQueueManager to a separate manager.py file

This commit is contained in:
Roberto Rosario
2011-05-24 23:20:02 -04:00
parent d3ebc6577f
commit 3fd7a3181e
2 changed files with 14 additions and 13 deletions

13
apps/ocr/manager.py Normal file
View File

@@ -0,0 +1,13 @@
from django.db import models
from ocr.exceptions import AlreadyQueued
class DocumentQueueManager(models.Manager):
def queue_document(self, document, queue_name='default'):
document_queue = self.model.objects.get(name=queue_name)
if document_queue.queuedocument_set.filter(document=document):
raise AlreadyQueued
document_queue.queuedocument_set.create(document=document, delay=True)
return document_queue