Produits Entr'ouvert » Combo

from django.utils.translation import ugettext_lazy as _

from .models import (WcsFormCell, WcsCategoryCell, WcsFormsOfCategoryCell,

                     WcsCurrentFormsCell)

from .utils import get_wcs_options, get_wcs_services

class CategoriesSelectMultiple(forms.SelectMultiple):

    def format_value(self, value):

        # this converts data dictionary to list, for django >=1.11

        return super(CategoriesSelectMultiple, self).format_value(

                value.get('data') or [])

    def render_options(self, choices, value):

        # this converts data dictionary to list, for django <1.11

        value = value.get('data') or []

        return super(CategoriesSelectMultiple, self).render_options(choices, value)

    def value_from_datadict(self, data, files, name):

        if isinstance(data, MultiValueDict):

            return {'data': data.getlist(name)}

        return data.get(name, None)

class WcsCurrentFormsCellForm(forms.ModelForm):

    class Meta:

        model = WcsCurrentFormsCell

        fields = ['wcs_site', 'categories', 'current_forms', 'done_forms']

    def __init__(self, *args, **kwargs):

        super(WcsCurrentFormsCellForm, self).__init__(*args, **kwargs)

        if len(get_wcs_services()) == 1:

            self.fields['wcs_site'].widget = forms.HiddenInput()

        else:

            combo_wcs_sites = [('', _('All'))]

            wcs_sites = [(x, y.get('title')) for x, y in get_wcs_services().items()]

            wcs_sites.sort(key=lambda x: x[1])

            combo_wcs_sites.extend(wcs_sites)

            self.fields['wcs_site'].widget = forms.Select(choices=combo_wcs_sites,

                    attrs={'class': 'wcs-site-select'})

        categories = get_wcs_options('/api/categories/')

        self.fields['categories'].help_text = _('By default forms from all categories are displayed.')

        self.fields['categories'].widget = CategoriesSelectMultiple(choices=categories,

                attrs={'class': 'categories-select'})

    categories = JSONField(blank=True)

    def get_default_form_class(self):

        from .forms import WcsCurrentFormsCellForm

        return WcsCurrentFormsCellForm

        categories_filter = {}

        if self.categories:

            for category in self.categories.get('data', []):

                categories_filter[tuple(category.split(':'))] = True

                forms = context['user_forms'][wcs_site]['data']

                if categories_filter:

                    forms = [x for x in forms if (wcs_site, x.get('category_slug')) in categories_filter]

                    forms = [x for x in forms if not x.get('form_status_is_endpoint')]

                    forms = [x for x in forms if x.get('form_status_is_endpoint')]

                context['user_forms'][wcs_site]['data'] = forms  # put it back

    # limit to a category

    cell.categories = {'data': ['default:test-3']}

    extra_context = cell.get_cell_extra_context(context)

    assert len(extra_context['forms']) == 0

    cell.categories = {'data': ['default:test-9']}

    extra_context = cell.get_cell_extra_context(context)

    assert len(extra_context['forms']) == 5

@wcsctl_present

def test_manager_current_forms(app, admin_user):

    Page.objects.all().delete()

    page = Page(title='One', slug='one', template_name='standard')

    page.save()

    app = login(app)

    resp = app.get('/manage/pages/%s/' % page.id)

    resp = app.get(resp.html.find('option',

                   **{'data-add-url': re.compile('wcscurrentformscell')})['data-add-url'])

    cells = page.get_cells()

    assert len(cells) == 1

    assert isinstance(cells[0], WcsCurrentFormsCell)

    resp = app.get('/manage/pages/%s/' % page.id)

    assert ('data-cell-reference="%s"' % cells[0].get_reference()) in resp.body

    assert len(resp.form['c%s-categories' % cells[0].get_reference()].options) == 4

    resp.form['c%s-categories' % cells[0].get_reference()].value = ['default:test-3', 'default:test-9']

    resp = resp.form.submit().follow()

    assert resp.form['c%s-categories' % cells[0].get_reference()].value == ['default:test-3', 'default:test-9']

    # check wcs_site field is a select box

    assert resp.form['c%s-wcs_site' % cells[0].get_reference()].tag == 'select'

    try:

        # check the wcs_site field is hidden if there's a single one defined in

        # the configuration

        temp_settings = settings.KNOWN_SERVICES.copy()

        default = settings.KNOWN_SERVICES['wcs']['default']

        settings.KNOWN_SERVICES = {'wcs': {'default': default}}

        resp = app.get('/manage/pages/%s/' % page.id)

        assert resp.form['c%s-wcs_site' % cells[0].get_reference()].attrs['type'] == 'hidden'

    finally:

        # restore original settings

        settings.KNOWN_SERVICES = temp_settings