Files
mayan-edms/mayan/apps/documents/serializers.py
Roberto Rosario 74dfa53787 Update documents app
Rename the DeletedDocument proxy model to a TrashedDocument.

Rename the deleted_document views to trashed_document.

Rename the document and deleted_document URL parameters to
trashed_document.

Update URL parameters to the '_id' form.

Add keyword arguments.

Update use of .filter_by_access().

Enclose trashed document restore method in a transaction.

Sort arguments.

Update app for compliance with MERCs 5 and 6.

Add document page view tests.

Add favorite document view tests.

Movernize tests.

Replace use of urlencode with furl.

Update views to use ExternalObjectMixin.

Refactor the document and version download views.

Rename the DocumentDocumentTypeEditView to DocumentChangeTypeView.

Move the trashed document views to their own module.

Signed-off-by: Roberto Rosario <Roberto.Rosario@mayan-edms.com>
2019-01-28 05:25:48 -04:00

471 lines
17 KiB
Python

from __future__ import unicode_literals
from django.utils.encoding import force_text
from rest_framework import serializers
from rest_framework.reverse import reverse
from mayan.apps.common.models import SharedUploadedFile
from mayan.apps.rest_api.relations import MultiKwargHyperlinkedIdentityField
from .models import (
Document, DocumentPage, DocumentType, DocumentTypeFilename,
DocumentVersion, RecentDocument
)
from .settings import setting_language
from .tasks import task_upload_new_version
class DocumentPageSerializer(serializers.HyperlinkedModelSerializer):
#document_versions_url = serializers.SerializerMethodField()
image_url = MultiKwargHyperlinkedIdentityField(
view_kwargs=(
{
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_page_id',
},
{
'lookup_field': 'document.pk', 'lookup_url_kwarg': 'document_id',
},
{
'lookup_field': 'document_version_id', 'lookup_url_kwarg': 'document_version_id',
},
),
view_name='rest_api:document_page-image'
)
#url = serializers.SerializerMethodField()
document_version_url = MultiKwargHyperlinkedIdentityField(
view_kwargs=(
{
'lookup_field': 'document_version_id', 'lookup_url_kwarg': 'document_version_id',
},
{
'lookup_field': 'document.pk', 'lookup_url_kwarg': 'document_id',
}
),
view_name='rest_api:document_version-detail'
)
url = MultiKwargHyperlinkedIdentityField(
view_kwargs=(
{
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_page_id',
},
{
'lookup_field': 'document.pk', 'lookup_url_kwarg': 'document_id',
},
{
'lookup_field': 'document_version_id', 'lookup_url_kwarg': 'document_version_id',
},
),
view_name='rest_api:document_page-detail'
)
class Meta:
fields = ('document_version_url', 'image_url', 'page_number', 'url')
#fields = ('document_version_url', 'page_number', 'url')
model = DocumentPage
"""
def get_document_versions_url(self, instance):
return reverse(
viewname='rest_api:documentversion-detail', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.document_version.pk
}, request=self.context['request'], format=self.context['format']
)
def get_image_url(self, instance):
return reverse(
viewname='rest_api:documentpage-image', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.document_version.pk,
'document_page_id': instance.pk,
}, request=self.context['request'], format=self.context['format']
)
def get_url(self, instance):
return reverse(
viewname='rest_api:documentpage-detail', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.document_version.pk,
'document_page_id': instance.pk,
}, request=self.context['request'], format=self.context['format']
)
"""
class DocumentTypeFilenameSerializer(serializers.ModelSerializer):
class Meta:
model = DocumentTypeFilename
fields = ('filename',)
class DocumentTypeSerializer(serializers.HyperlinkedModelSerializer):
documents_url = serializers.HyperlinkedIdentityField(
lookup_url_kwarg='document_type_id',
view_name='rest_api:document_type-document-list'
)
#documents_count = serializers.SerializerMethodField()
#filenames = DocumentTypeFilenameSerializer(many=True, read_only=True)
class Meta:
extra_kwargs = {
'url': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_type_id',
'view_name': 'rest_api:document_type-detail'
}
}
fields = (
'delete_time_period', 'delete_time_unit', 'documents_url',
#'delete_time_period', 'delete_time_unit',
#'documents_count', 'id', 'label', 'filenames', 'trash_time_period',
'id', 'label', 'trash_time_period',
'trash_time_unit', 'url'
)
model = DocumentType
#def get_documents_count(self, obj):
# return obj.documents.count()
"""
class WritableDocumentTypeSerializer(serializers.ModelSerializer):
documents_url = serializers.HyperlinkedIdentityField(
lookup_field='pk', lookup_url_kwarg='document_type_id',
view_name='rest_api:documenttype-document-list'
)
documents_count = serializers.SerializerMethodField()
class Meta:
extra_kwargs = {
'url': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_type_id',
'view_name': 'rest_api:documenttype-detail'
}
}
fields = (
'delete_time_period', 'delete_time_unit', 'documents_url',
'documents_count', 'id', 'label', 'trash_time_period',
'trash_time_unit', 'url'
)
model = DocumentType
def get_documents_count(self, obj):
return obj.documents.count()
"""
class DocumentVersionSerializer(serializers.HyperlinkedModelSerializer):
#document_url = serializers.SerializerMethodField()
#download_url = serializers.SerializerMethodField()
document_url = serializers.HyperlinkedIdentityField(
lookup_field='document_id', lookup_url_kwarg='document_id',
view_name='rest_api:document-detail'
)
#pages_url = serializers.SerializerMethodField()
pages_url = MultiKwargHyperlinkedIdentityField(
view_kwargs=(
{
'lookup_field': 'document_id', 'lookup_url_kwarg': 'document_id',
},
{
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_version_id',
}
),
view_name='rest_api:document_page-list'
)
size = serializers.SerializerMethodField()
#url = serializers.SerializerMethodField()
url = MultiKwargHyperlinkedIdentityField(
view_kwargs=(
{
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_version_id',
},
{
'lookup_field': 'document_id', 'lookup_url_kwarg': 'document_id',
},
),
view_name='rest_api:document_version-detail'
)
class Meta:
extra_kwargs = {
#'document': {
# 'lookup_field': 'pk', 'lookup_url_kwarg': 'document_id',
# 'view_name': 'rest_api:document-detail'
#},
'file': {'use_url': False},
#'url': {
# 'view_kwargs': (
# {
# 'lookup_field': 'pk', 'lookup_url_kwarg': 'document_version_id',
# },
# {
# 'lookup_field': 'document__pk', 'lookup_url_kwarg': 'document_id',
# },
# ),
# #'lookup_field': 'pk', 'lookup_url_kwarg': 'document_version_id',
# 'view_name': 'rest_api:document_version-detail'
#},
}
fields = (
#'checksum', 'comment', 'document_url', 'download_url', 'encoding',
'checksum', 'comment', 'document_url', 'encoding',
#'checksum', 'comment', 'encoding',
'file', 'mimetype', 'pages_url', 'size', 'timestamp', 'url'
#'file', 'mimetype', 'size', 'timestamp', 'url'
)
model = DocumentVersion
read_only_fields = ('document', 'file', 'size')
def get_size(self, instance):
return instance.size
def build_url_field(self, field_name, model_class):
"""
Create a field representing the object's own URL.
"""
field_class = self.serializer_url_field
field_kwargs = {'kwargs': 1}
return field_class, field_kwargs
"""
def get_document_url(self, instance):
return reverse(
viewname='rest_api:document-detail', kwargs={
'document_id': instance.document.pk
}, request=self.context['request']#, format=self.context['format']
)
def get_download_url(self, instance):
return reverse(
viewname='rest_api:document_version-download', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request']#, format=self.context['format']
)
def get_pages_url(self, instance):
return reverse(
viewname='rest_api:document_version-page-list', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request']#, format=self.context['format']
)
def get_url(self, obj, view_name, request, format):
print ' obj, view_name, request, format', obj, view_name, request, format
# Unsaved objects will not yet have a valid URL.
if hasattr(obj, 'pk') and obj.pk in (None, ''):
return None
lookup_value = getattr(obj, self.lookup_field)
kwargs = {self.lookup_url_kwarg: lookup_value}
return self.reverse(view_name, kwargs=kwargs, request=request, format=format)
"""
"""
def get_url(self, instance, *args, **kwargs):
print ', *args, **kwargs', args, kwargs
return reverse(
viewname='rest_api:document_version-detail', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request']#, format=self.context['format']
)
"""
"""
class WritableDocumentVersionSerializer(serializers.ModelSerializer):
document_url = serializers.SerializerMethodField()
download_url = serializers.SerializerMethodField()
pages_url = serializers.SerializerMethodField()
url = serializers.SerializerMethodField()
class Meta:
extra_kwargs = {
'file': {'use_url': False},
}
fields = (
'checksum', 'comment', 'document_url', 'download_url', 'encoding',
'file', 'mimetype', 'pages_url', 'timestamp', 'url'
)
model = DocumentVersion
read_only_fields = ('document', 'file')
def get_document_url(self, instance):
return reverse(
viewname='rest_api:document-detail', kwargs={
'document_id': instance.document.pk
}, request=self.context['request'], format=self.context['format']
)
def get_download_url(self, instance):
return reverse(
viewname='rest_api:documentversion-download', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request'], format=self.context['format']
)
def get_pages_url(self, instance):
return reverse(
viewname='rest_api:documentversion-page-list', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request'], format=self.context['format']
)
def get_url(self, instance):
return reverse(
viewname='rest_api:documentversion-detail', kwargs={
'document_id': instance.document.pk,
'document_version_id': instance.pk
}, request=self.context['request'], format=self.context['format']
)
class NewDocumentVersionSerializer(serializers.Serializer):
comment = serializers.CharField(allow_blank=True)
file = serializers.FileField(use_url=False)
def save(self, document, _user):
shared_uploaded_file = SharedUploadedFile.objects.create(
file=self.validated_data['file']
)
task_upload_new_version.delay(
comment=self.validated_data.get('comment', ''),
document_id=document.pk,
shared_uploaded_file_id=shared_uploaded_file.pk, user_id=_user.pk
)
"""
class DeletedDocumentSerializer(serializers.HyperlinkedModelSerializer):
document_type_label = serializers.SerializerMethodField()
restore = serializers.HyperlinkedIdentityField(
lookup_field='pk', lookup_url_kwarg='document_id',
view_name='rest_api:trasheddocument-restore'
)
class Meta:
extra_kwargs = {
'document_type': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_type_id',
'view_name': 'rest_api:documenttype-detail'
},
'url': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_id',
'view_name': 'rest_api:trasheddocument-detail'
}
}
fields = (
'date_added', 'deleted_date_time', 'description', 'document_type',
'document_type_label', 'id', 'label', 'language', 'restore',
'url', 'uuid',
)
model = Document
read_only_fields = (
'deleted_date_time', 'description', 'document_type', 'label',
'language'
)
def get_document_type_label(self, instance):
return instance.document_type.label
class DocumentSerializer(serializers.HyperlinkedModelSerializer):
document_type = DocumentTypeSerializer(read_only=True)
#document_type_url = serializers.HyperlinkedIdentityField(
# lookup_field='document_type_id', lookup_url_kwarg='document_type_id',
# view_name='rest_api:document_type-detail'
#)
latest_version = DocumentVersionSerializer(many=False, read_only=True)
versions_url = serializers.HyperlinkedIdentityField(
lookup_field='pk', lookup_url_kwarg='document_id',
view_name='rest_api:document_version-list'
)
#view_name='rest_api:document_type-document-list'
class Meta:
extra_kwargs = {
'document_type': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_type_id',
'view_name': 'rest_api:document_type-detail'
},
'url': {
'lookup_field': 'pk', 'lookup_url_kwarg': 'document_id',
'view_name': 'rest_api:document-detail'
}
}
fields = (
'date_added', 'description', 'document_type', 'id', 'label',
#'date_added', 'description', 'document_type_url', 'id', 'label',
'language', 'latest_version', 'url', 'uuid', 'versions_url',
#'language', 'url', 'uuid', 'versions_url'
)
model = Document
#read_only_fields = ('document_type', 'label')
"""
class WritableDocumentSerializer(serializers.ModelSerializer):
document_type = DocumentTypeSerializer(read_only=True)
latest_version = DocumentVersionSerializer(many=False, read_only=True)
versions = serializers.HyperlinkedIdentityField(
lookup_field='pk', lookup_url_kwarg='document_id',
view_name='rest_api:document-version-list'
)
url = serializers.HyperlinkedIdentityField(
lookup_field='pk', lookup_url_kwarg='document_id',
view_name='rest_api:document-detail',
)
class Meta:
fields = (
'date_added', 'description', 'document_type', 'id', 'label',
'language', 'latest_version', 'url', 'uuid', 'versions',
)
model = Document
read_only_fields = ('document_type',)
class NewDocumentSerializer(serializers.ModelSerializer):
file = serializers.FileField(write_only=True)
def save(self, _user):
document = Document.objects.create(
description=self.validated_data.get('description', ''),
document_type=self.validated_data['document_type'],
label=self.validated_data.get(
'label', force_text(self.validated_data['file'])
),
language=self.validated_data.get(
'language', setting_language.value
)
)
document.save(_user=_user)
shared_uploaded_file = SharedUploadedFile.objects.create(
file=self.validated_data['file']
)
task_upload_new_version.delay(
document_id=document.pk,
shared_uploaded_file_id=shared_uploaded_file.pk, user_id=_user.pk
)
self.instance = document
return document
class Meta:
fields = (
'description', 'document_type', 'id', 'file', 'label', 'language',
)
model = Document
"""
class RecentDocumentSerializer(serializers.ModelSerializer):
class Meta:
fields = ('document', 'datetime_accessed')
model = RecentDocument