Produits Entr'ouvert » w.c.s.

import itertools

from wcs.formdef import FormDef

from quixote import get_publisher

publisher = get_publisher()

def is_upload(obj):

    # we can't use isinstance() because obj can be a

    # wcs.qommon.form.PicklableUpload or a qommon.form.PicklableUpload

    return obj.__class__.__name__ == 'PicklableUpload'

def is_attachment(obj):

    return obj.__class__.__name__ == 'AttachmentEvolutionPart'

def accumulate_filenames():

    from wcs.carddef import CardDef

    from wcs.wf.form import WorkflowFormEvolutionPart

    for formdef in FormDef.select(ignore_migration=True) + CardDef.select(ignore_migration=True):

        for option_data in (formdef.workflow_options or {}).values():

            if is_upload(option_data):

                yield 'option_data: %s' % option_data.get_fs_filename()

        for formdata in formdef.data_class().select_iterator(ignore_errors=True, itersize=200):

            for field_data in itertools.chain(

                (formdata.data or {}).values(), (formdata.workflow_data or {}).values()

            ):

                if is_upload(field_data):

                    yield 'formdata field_data: %s' % field_data.qfilename

                elif isinstance(field_data, dict) and isinstance(field_data.get('data'), list):

                    for subfield_rowdata in field_data.get('data'):

                        if isinstance(subfield_rowdata, dict):

                            for block_field_data in subfield_rowdata.values():

                                if is_upload(block_field_data):

                                    yield 'block_field_data: %s' % block_field_data.qfilename

            for part in formdata.iter_evolution_parts():

                if is_attachment(part):

                    # yield 'part.filename: %s' % part.filename

                    yield part.filename

                elif isinstance(part, WorkflowFormEvolutionPart):

                    for field_data in (part.data or {}).values():

                        if is_upload(field_data):

                            yield 'WorkflowFormEvolutionPart field_data: %s' % field_data.qfilename

    for user in publisher.user_class.select():

        for field_data in (user.form_data or {}).values():

            if is_upload(field_data):

                yield 'user.form_data field_data: %s' % field_data.qfilename

for filename in accumulate_filenames():

    print(filename)