Produits Entr'ouvert » Combo

    first_data_key = 'json'

        context['concerned_user'] = self.get_concerned_user(context)

        data_urls = [{'key': self.first_data_key, 'url': self.url, 'cache_duration': self.cache_duration,

        self._json_content = extra_context[self.first_data_key]

# combo - content management system

# Copyright (C) 2014-2018  Entr'ouvert

#

# This program is free software: you can redistribute it and/or modify it

# under the terms of the GNU Affero General Public License as published

# by the Free Software Foundation, either version 3 of the License, or

# (at your option) any later version.

#

# This program is distributed in the hope that it will be useful,

# but WITHOUT ANY WARRANTY; without even the implied warranty of

# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

# GNU Affero General Public License for more details.

#

# You should have received a copy of the GNU Affero General Public License

# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import django.apps

from django.utils.translation import ugettext_lazy as _

class AppConfig(django.apps.AppConfig):

    name = 'combo.profile'

    verbose_name = _('Profile')

default_app_config = 'combo.profile.AppConfig'

from collections import OrderedDict

import copy

from django.utils.dateparse import parse_date

from django.utils.translation import ugettext_lazy as _

from combo.data.models import JsonCellBase

from combo.data.library import register_cell_class

@register_cell_class

class ProfileCell(JsonCellBase):

    template_name = 'combo/profile.html'

    first_data_key = 'profile'

    class Meta:

        verbose_name = _('Profile')

    @property

    def url(self):

        idp = settings.KNOWN_SERVICES.get('authentic').values()[0]

        return '{%% load combo %%}%sapi/users/{{ concerned_user|name_id }}' % idp.get('url')

    def get_cell_extra_context(self, context):

        extra_context = super(ProfileCell, self).get_cell_extra_context(context)

        extra_context['profile_fields'] = OrderedDict()

        if extra_context.get('profile') is not None:

            for attribute in settings.USER_PROFILE_CONFIG.get('fields'):

                extra_context['profile_fields'][attribute['name']] = copy.copy(attribute)

                value = extra_context['profile'].get(attribute['name'])

                if value:

                    if attribute['kind'] in ('birthdate', 'date'):

                        value = parse_date(value)

                extra_context['profile_fields'][attribute['name']]['value'] = value

        else:

            extra_context['error'] = 'unknown user'

        return extra_context

{% load i18n %}

{% block cell-content %}

<div class="profile">

<h2>{% trans "Profile" %}</h2>

{% for key, details in profile_fields.items %}

  {% if details.value and details.user_visible %}

    <p><span class="label">{{ details.label }}</span> <span class="value">{{ details.value }}</span></p>

  {% endif %}

{% endfor %}

{% if error == 'unknown user' %}

<p>{% trans 'Unknown User' %}</p>

{% endif %}

{% endblock %}

from django.conf import settings

from django.template import VariableDoesNotExist

if 'mellon' in settings.INSTALLED_APPS:

    from mellon.models import UserSAMLIdentifier

@register.filter

def name_id(user):

    saml_id = UserSAMLIdentifier.objects.filter(user=user).last()

    if saml_id:

        return saml_id.name_id

    # it is important to raise this so get_templated_url is aborted and no call

    # is tried with a missing user argument.

    raise VariableDoesNotExist('name_id')

USER_PROFILE_CONFIG = {

    'fields': [

        {

            'name': 'first_name',

            'kind': 'string',

            'label': 'First Name',

            'user_visible': True,

        },

        {

            'name': 'birthdate',

            'kind': 'birthdate',

            'label': 'Birth Date',

            'user_visible': True,

        }

    ]

}

import datetime

import json

import mock

import pytest

from django.test import override_settings

from combo.data.models import Page

from combo.profile.models import ProfileCell

pytestmark = pytest.mark.django_db

def test_profile_cell(app, admin_user):

    page = Page()

    page.save()

    cell = ProfileCell(page=page, order=0)

    cell.save()

    with override_settings(

                KNOWN_SERVICES={'authentic': {'idp': {

                    'title': 'IdP', 'url': 'http://example.org/'}}}), \

            mock.patch('combo.utils.requests.get') as requests_get, \

            mock.patch('combo.public.templatetags.combo.UserSAMLIdentifier') as user_saml:

        data = {'first_name': 'Foo', 'birthdate': '2018-08-10'}

        requests_get.return_value = mock.Mock(

                content=json.dumps(data),

                json=lambda: data,

                status_code=200)

        def filter_mock(user=None):

            assert user is admin_user

            return mock.Mock(last=lambda: mock.Mock(name_id='123456'))

        mocked_objects = mock.Mock()

        mocked_objects.filter = mock.Mock(side_effect=filter_mock)

        user_saml.objects = mocked_objects

        context = cell.get_cell_extra_context({'synchronous': True, 'selected_user': admin_user})

        assert context['profile_fields']['first_name']['value'] == 'Foo'

        assert context['profile_fields']['birthdate']['value'] == datetime.date(2018, 8, 10)

        assert requests_get.call_args[0][0] == 'http://example.org/api/users/123456'

        def filter_mock_missing(user=None):

            return mock.Mock(last=lambda: None)

        mocked_objects.filter = mock.Mock(side_effect=filter_mock_missing)

        context = cell.get_cell_extra_context({'synchronous': True, 'selected_user': admin_user})

        assert context['error'] == 'unknown user'

        assert requests_get.call_count == 1  # no new call was made