from __future__ import absolute_import, unicode_literals from django.contrib import messages from django.http import HttpResponseRedirect from django.urls import reverse_lazy from django.utils.translation import ugettext_lazy as _ from django.utils.translation import ungettext from mayan.apps.common.generics import ( FormView, MultipleObjectConfirmActionView, SingleObjectEditView, SingleObjectListView ) from mayan.apps.common.mixins import ExternalObjectMixin from mayan.apps.documents.forms import DocumentTypeFilteredSelectForm from mayan.apps.documents.models import Document, DocumentType from .icons import icon_file_metadata from .models import DocumentVersionDriverEntry from .permissions import ( permission_document_type_file_metadata_setup, permission_file_metadata_submit, permission_file_metadata_view ) class DocumentDriverListView(ExternalObjectMixin, SingleObjectListView): external_object_class = Document external_object_permission = permission_file_metadata_view external_object_pk_url_kwarg = 'document_id' def get_extra_context(self): return { 'hide_object': True, 'no_results_icon': icon_file_metadata, 'no_results_text': _( 'File metadata are the attributes of the document\'s file. ' 'They can range from camera information used to take a photo ' 'to the author that created a file. File metadata are set ' 'when the document\'s file was first created. File metadata ' 'attributes reside in the file itself. They are not the ' 'same as the document metadata, which are user defined and ' 'reside in the database.' ), 'no_results_title': _('No file metadata available.'), 'object': self.external_object, 'title': _( 'File metadata drivers for: %s' ) % self.external_object, } def get_source_queryset(self): return self.external_object.latest_version.file_metadata_drivers.all() class DocumentVersionDriverEntryFileMetadataListView(ExternalObjectMixin, SingleObjectListView): external_object_class = DocumentVersionDriverEntry external_object_permission = permission_file_metadata_view external_object_pk_url_kwarg = 'document_version_driver_id' def get_extra_context(self): return { 'hide_object': True, 'no_results_title': _('No file metadata available.'), 'object': self.external_object.document_version.document, 'title': _( 'File metadata attribures for: %(document)s, for driver: %(driver)s' ) % { 'document': self.external_object.document_version.document, 'driver': self.external_object.driver }, } def get_source_queryset(self): return self.external_object.entries.all() class DocumentSubmitView(MultipleObjectConfirmActionView): model = Document object_permission = permission_file_metadata_submit pk_url_kwarg = 'document_id' success_message_singular = '%(count)d document submitted to the file metadata queue.' success_message_plural = '%(count)d documents submitted to the file metadata queue.' def get_extra_context(self): queryset = self.object_list result = { 'title': ungettext( singular='Submit the selected document to the file metadata queue?', plural='Submit the selected documents to the file metadata queue?', number=queryset.count() ) } if queryset.count() == 1: result['object'] = queryset.first() return result def object_action(self, form, instance): instance.submit_for_file_metadata_processing() class DocumentTypeSettingsEditView(ExternalObjectMixin, SingleObjectEditView): external_object_class = DocumentType external_object_permission = permission_document_type_file_metadata_setup external_object_pk_url_kwarg = 'document_type_id' fields = ('auto_process',) post_action_redirect = reverse_lazy(viewname='documents:document_type_list') def get_extra_context(self): return { 'object': self.external_object, 'title': _( 'Edit file metadata settings for document type: %s' ) % self.external_object } def get_object(self, queryset=None): return self.external_object.file_metadata_settings class DocumentTypeSubmitView(FormView): extra_context = { 'title': _( 'Submit all documents of a type for file metadata processing.' ) } form_class = DocumentTypeFilteredSelectForm post_action_redirect = reverse_lazy(viewname='common:tools_list') def get_form_extra_kwargs(self): return { 'allow_multiple': True, 'permission': permission_file_metadata_submit, 'user': self.request.user } def form_valid(self, form): count = 0 for document_type in form.cleaned_data['document_type']: for document in document_type.documents.all(): document.submit_for_file_metadata_processing() count += 1 messages.success( message=_( '%(count)d documents added to the file metadata processing ' 'queue.' ) % { 'count': count, }, request=self.request ) return HttpResponseRedirect(redirect_to=self.get_success_url())