matthias/mayan-edms
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Perform validation at the model

Browse Source
This commit is contained in:
Roberto Rosario
2014-11-04 17:36:34 -04:00
parent 6a19622e38
commit e6dbb4cd70
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
mayan/apps/checkouts/models.py
View File

@@ -11,6 +11,7 @@ from documents.models import Document
from history.api import create_history from history.api import create_history
from .events import HISTORY_DOCUMENT_CHECKED_OUT from .events import HISTORY_DOCUMENT_CHECKED_OUT
from .exceptions import DocumentAlreadyCheckedOut
from .managers import DocumentCheckoutManager from .managers import DocumentCheckoutManager
logger = logging.getLogger(__name__) logger = logging.getLogger(__name__)
@@ -23,6 +24,8 @@ class DocumentCheckout(models.Model):
document = models.ForeignKey(Document, verbose_name=_(u'Document'), unique=True) document = models.ForeignKey(Document, verbose_name=_(u'Document'), unique=True)
checkout_datetime = models.DateTimeField(verbose_name=_(u'Check out date and time'), auto_now_add=True) checkout_datetime = models.DateTimeField(verbose_name=_(u'Check out date and time'), auto_now_add=True)
expiration_datetime = models.DateTimeField(verbose_name=_(u'Check out expiration date and time'), help_text=_(u'Amount of time to hold the document checked out in minutes.')) expiration_datetime = models.DateTimeField(verbose_name=_(u'Check out expiration date and time'), help_text=_(u'Amount of time to hold the document checked out in minutes.'))
# TODO: simplify user_object to an instance of User
user_content_type = models.ForeignKey(ContentType, null=True, blank=True) # blank and null added for ease of db migration user_content_type = models.ForeignKey(ContentType, null=True, blank=True) # blank and null added for ease of db migration
user_object_id = models.PositiveIntegerField(null=True, blank=True) user_object_id = models.PositiveIntegerField(null=True, blank=True)
user_object = generic.GenericForeignKey(ct_field='user_content_type', fk_field='user_object_id') user_object = generic.GenericForeignKey(ct_field='user_content_type', fk_field='user_object_id')
@@ -40,6 +43,9 @@ class DocumentCheckout(models.Model):
def save(self, *args, **kwargs): def save(self, *args, **kwargs):
new_checkout = not self.pk new_checkout = not self.pk
if not new_checkout or self.document.is_checked_out():
raise DocumentAlreadyCheckedOut
result = super(DocumentCheckout, self).save(*args, **kwargs) result = super(DocumentCheckout, self).save(*args, **kwargs)
if new_checkout: if new_checkout:
create_history(HISTORY_DOCUMENT_CHECKED_OUT, source_object=self.document, data={'user': self.user_object, 'document': self.document}) create_history(HISTORY_DOCUMENT_CHECKED_OUT, source_object=self.document, data={'user': self.user_object, 'document': self.document})