acdc7dca48
to a resource based one. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
54 lines
1.7 KiB
Python
54 lines
1.7 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from rest_framework import serializers
|
|
|
|
from acls.models import AccessControlList
|
|
from documents.models import Document
|
|
from documents.serializers import DocumentSerializer
|
|
|
|
from .models import DocumentCheckout
|
|
from .permissions import permission_document_checkout
|
|
|
|
|
|
class DocumentCheckoutSerializer(serializers.ModelSerializer):
|
|
def __init__(self, *args, **kwargs):
|
|
super(DocumentCheckoutSerializer, self).__init__(*args, **kwargs)
|
|
self.fields['document'] = DocumentSerializer()
|
|
|
|
class Meta:
|
|
model = DocumentCheckout
|
|
|
|
|
|
class NewDocumentCheckoutSerializer(serializers.ModelSerializer):
|
|
block_new_version = serializers.BooleanField()
|
|
document_pk = serializers.IntegerField(
|
|
help_text=_('Primary key of the document to be checked out.'),
|
|
write_only=True
|
|
)
|
|
expiration_datetime = serializers.DateTimeField()
|
|
|
|
class Meta:
|
|
fields = (
|
|
'block_new_version', 'document', 'document_pk',
|
|
'expiration_datetime', 'id'
|
|
)
|
|
model = DocumentCheckout
|
|
read_only_fields = ('document',)
|
|
write_only_fields = ('document_pk',)
|
|
|
|
def create(self, validated_data):
|
|
document = Document.objects.get(pk=validated_data.pop('document_pk'))
|
|
|
|
AccessControlList.objects.check_access(
|
|
permissions=permission_document_checkout,
|
|
user=self.context['request'].user, obj=document
|
|
)
|
|
|
|
validated_data['document'] = document
|
|
validated_data['user'] = self.context['request'].user
|
|
return super(NewDocumentCheckoutSerializer, self).create(
|
|
validated_data
|
|
)
|