Produits Entr'ouvert » Authentic 2

class RoleMembersSearchForm(UserSearchForm):

    all_members = forms.BooleanField(initial=False, label=_('View all members'), required=False)

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

        disable_all_members = kwargs.pop('disable_all_members', False)

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

        if disable_all_members:

            self.fields['all_members'].widget.attrs['disabled'] = True

    search_form_class = forms.RoleMembersSearchForm

    @property

    def table_class(self):

        if self.view_all_members:

            return tables.RoleMembersTable

        return tables.MixedUserRoleTable

    def dispatch(self, request, *args, **kwargs):

        self.children = views.filter_view(self.request, self.get_object().children(include_self=False))

        return super().dispatch(request, *args, **kwargs)

    def get_table_data(self):

        if self.view_all_members:

            return super().get_table_data()

        members = views.filter_view(self.request, self.object.members.all())

        members = self.filter_by_search(members)

        return list(self.children) + list(members)

        via_prefetch = Prefetch('roles', queryset=self.children, to_attr='via')

    @property

    def view_all_members(self):

        return self.search_form.is_valid() and self.search_form.cleaned_data.get('all_members')

    def get_search_form_kwargs(self):

        kwargs = super().get_search_form_kwargs()

        if not self.children:

            kwargs['data']['search-all_members'] = 'on'

            kwargs['disable_all_members'] = True

        return kwargs

Role = get_role_model()

        record = kwargs['record']

        if isinstance(record, User) and not record.is_active:

class UserOrRoleColumn(UserLinkColumn):

    def render(self, **kwargs):

        value = super().render(**kwargs)

        if isinstance(kwargs['record'], Role):

            value = html.format_html(_('Members of role {value}'), value=value)

        return value

class MixedUserRoleTable(tables.Table):

    name = UserOrRoleColumn(

        verbose_name=_('Members'),

        text=lambda x: x.get_full_name() if isinstance(x, User) else x.name,

        orderable=False,

    )

    class Meta:

        attrs = {'class': 'main', 'id': 'user-table'}

 {% if search_form.cleaned_data.all_members %}

 {% endif %}

        return {'request': self.request, 'data': self.request.GET.dict()}

    response.forms['search-form']['search-all_members'] = True

    response = response.forms['search-form'].submit()

    assert not resp.pyquery('a.role-inheritance-%s:contains("Role a")' % relation)

def test_role_members_user_role_mixed_table(app, superuser, settings, simple_role, simple_user):

    simple_user.roles.add(simple_role)

    url = reverse('a2-manager-role-members', kwargs={'pk': simple_role.pk})

    resp = login(app, superuser, url)

    # no children, directly display members details

    assert resp.forms['search-form']['search-all_members'].value == 'on'

    assert 'disabled' in resp.forms['search-form']['search-all_members'].attrs

    assert 'Download list as CSV' in resp.text

    column_names = [text_content(el) for el in resp.pyquery('table#user-table th') if text_content(el)]

    assert column_names == [

        'User',

        'Username',

        'Email Address',

        'First Name',

        'Last Name',

        'Organizational Unit',

        'Direct member',

        'Inherited from',

    ]

    rows = [text_content(el) for el in resp.pyquery('tr td.link')]

    assert rows == ['Jôhn Dôe']

    # add child

    role = Role.objects.create(name='Role a', ou=get_default_ou())

    user = User.objects.create(username='user1')

    user.roles.add(role)

    simple_role.add_child(role)

    resp = app.get(url)

    assert not resp.forms['search-form']['search-all_members'].value

    assert 'disabled' not in resp.forms['search-form']['search-all_members'].attrs

    assert 'Download list as CSV' not in resp.text

    column_names = [text_content(el) for el in resp.pyquery('table#user-table th') if text_content(el)]

    assert column_names == ['Members']

    rows = [text_content(el) for el in resp.pyquery('tr td.name')]

    assert rows == ['Members of role Role a', 'Jôhn Dôe']

    resp.forms['search-form']['search-all_members'] = True

    resp = resp.forms['search-form'].submit()

    assert resp.forms['search-form']['search-all_members'].value == 'on'

    assert 'disabled' not in resp.forms['search-form']['search-all_members'].attrs

    assert 'Download list as CSV' in resp.text

    rows = [text_content(el) for el in resp.pyquery('tr td.link')]

    assert rows == ['user1', 'Jôhn Dôe']