74c97314d7
Add keyword arguments. Sort arguments and models. Move literals to their own module. Prepend handler_ to signal handlers. Signed-off-by: Roberto Rosario <roberto.rosario.gonzalez@gmail.com>
71 lines
2.2 KiB
Python
71 lines
2.2 KiB
Python
from __future__ import unicode_literals
|
|
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from mayan.apps.metadata.api import (
|
|
decode_metadata_from_querystring, save_metadata_list
|
|
)
|
|
from mayan.apps.metadata.forms import DocumentMetadataFormSet
|
|
|
|
from mayan.apps.sources.wizards import WizardStep, WizardStepDocumentType
|
|
|
|
|
|
class WizardStepMetadata(WizardStep):
|
|
form_class = DocumentMetadataFormSet
|
|
label = _('Enter document metadata')
|
|
name = 'metadata_entry'
|
|
number = 1
|
|
|
|
@classmethod
|
|
def condition(cls, wizard):
|
|
"""
|
|
Skip step if document type has no associated metadata
|
|
"""
|
|
cleaned_data = wizard.get_cleaned_data_for_step(WizardStepDocumentType.name) or {}
|
|
|
|
document_type = cleaned_data.get('document_type')
|
|
|
|
if document_type:
|
|
return document_type.metadata.exists()
|
|
|
|
@classmethod
|
|
def get_form_initial(cls, wizard):
|
|
initial = []
|
|
|
|
step_data = wizard.get_cleaned_data_for_step(WizardStepDocumentType.name)
|
|
if step_data:
|
|
document_type = step_data['document_type']
|
|
for document_type_metadata_type in document_type.metadata.all():
|
|
initial.append(
|
|
{
|
|
'document_type': document_type,
|
|
'metadata_type': document_type_metadata_type.metadata_type,
|
|
}
|
|
)
|
|
|
|
return initial
|
|
|
|
@classmethod
|
|
def done(cls, wizard):
|
|
result = {}
|
|
cleaned_data = wizard.get_cleaned_data_for_step(cls.name)
|
|
if cleaned_data:
|
|
for identifier, metadata in enumerate(wizard.get_cleaned_data_for_step(cls.name)):
|
|
if metadata.get('update'):
|
|
result['metadata%s_id' % identifier] = metadata['id']
|
|
result['metadata%s_value' % identifier] = metadata['value']
|
|
|
|
return result
|
|
|
|
@classmethod
|
|
def step_post_upload_process(cls, document, querystring=None):
|
|
metadata_dict_list = decode_metadata_from_querystring(querystring=querystring)
|
|
if metadata_dict_list:
|
|
save_metadata_list(
|
|
metadata_list=metadata_dict_list, document=document,
|
|
create=True
|
|
)
|
|
|
|
|
|
WizardStep.register(step=WizardStepMetadata)
|