Merge document type optional and required metadata type add views

This commit is contained in:
Roberto Rosario
2014-12-01 16:14:14 -04:00
parent 32d2608bdd
commit d257d6db85
3 changed files with 49 additions and 27 deletions

View File

@@ -24,7 +24,8 @@ from .permissions import (PERMISSION_METADATA_DOCUMENT_ADD,
PERMISSION_METADATA_TYPE_DELETE,
PERMISSION_METADATA_TYPE_EDIT,
PERMISSION_METADATA_TYPE_VIEW)
from .serializers import DocumentMetadataSerializer, MetadataTypeSerializer
from .serializers import (DocumentMetadataSerializer,
DocumentTypeNewMetadataType, MetadataTypeSerializer)
class APIMetadataTypeListView(generics.ListCreateAPIView):
@@ -136,12 +137,13 @@ class APIDocumentMetadataView(generics.RetrieveUpdateDestroyAPIView):
return super(APIDocumentMetadataView, self).put(*args, **kwargs)
class APIDocumentTypeMetadataTypeListView(generics.ListAPIView):
serializer_class = MetadataTypeSerializer
class APIDocumentTypeMetadataTypeOptionalListView(generics.ListCreateAPIView):
permission_classes = (MayanPermission,)
mayan_view_permissions = {'POST': [PERMISSION_DOCUMENT_TYPE_EDIT]}
required_metadata = False
def get_queryset(self):
document_type = get_object_or_404(DocumentType, pk=self.kwargs['document_type_pk'])
try:
@@ -149,14 +151,42 @@ class APIDocumentTypeMetadataTypeListView(generics.ListAPIView):
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_DOCUMENT_TYPE_VIEW, self.request.user, document_type)
return document_type.metadata.all()
return document_type.metadata_type.filter(required=self.required_metadata)
def get(self, *args, **kwargs):
"""Returns a list of selected document type allowed metadata types."""
return super(APIDocumentTypeMetadataTypeListView, self).get(*args, **kwargs)
return super(APIDocumentTypeMetadataTypeOptionalListView, self).get(*args, **kwargs)
def get_serializer_class(self):
if self.request.method == 'GET':
return MetadataTypeSerializer
elif self.request.method == 'POST':
return DocumentTypeNewMetadataType
def post(self, request, *args, **kwargs):
"""
Add a metadata type to a document type.
"""
document_type = get_object_or_404(DocumentType, pk=self.kwargs['document_type_pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_DOCUMENT_TYPE_EDIT])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_DOCUMENT_TYPE_EDIT, self.request.user, document_type)
serializer = self.get_serializer(data=self.request.POST)
if serializer.is_valid():
metadata_type = get_object_or_404(MetadataType, pk=serializer.data['metadata_type_pk'])
document_type.metadata_type.add(metadata_type, required=self.required_metadata)
return Response(status=status.HTTP_201_CREATED)
class APIDocumentTypeMetadataTypeView(views.APIView):
class APIDocumentTypeMetadataTypeRequiredListView(APIDocumentTypeMetadataTypeOptionalListView):
required_metadata = True
class APIDocumentTypeMetadataTypeRequiredView(views.APIView):
def delete(self, request, *args, **kwargs):
"""
Remove a metadata type from a document type.
@@ -169,21 +199,5 @@ class APIDocumentTypeMetadataTypeView(views.APIView):
AccessEntry.objects.check_access(PERMISSION_DOCUMENT_TYPE_EDIT, self.request.user, document_type)
metadata_type = get_object_or_404(MetadataType, pk=self.kwargs['metadata_type_pk'])
document_type.metadata.remove(metadata_type)
document_type.metadata_type.remove(metadata_type)
return Response(status=status.HTTP_204_NO_CONTENT)
def post(self, request, *args, **kwargs):
"""
Add a metadata type to a document type.
"""
# TODO: Change to be a POST method of APIDocumentTypeMetadataTypeListView
document_type = get_object_or_404(DocumentType, pk=self.kwargs['document_type_pk'])
try:
Permission.objects.check_permissions(self.request.user, [PERMISSION_DOCUMENT_TYPE_EDIT])
except PermissionDenied:
AccessEntry.objects.check_access(PERMISSION_DOCUMENT_TYPE_EDIT, self.request.user, document_type)
metadata_type = get_object_or_404(MetadataType, pk=self.kwargs['metadata_type_pk'])
# TODO: set required flag
document_type.metadata.add(metadata_type)
return Response(status=status.HTTP_201_CREATED)