Produits Entr'ouvert » Authentic 2

    A2_USERS_CSV_IMPORT_APP_ID_LABEL=Setting(default='app_id', definition='Label of the applicative identifier used when importing users from csv'),

import csv

from django.contrib.auth import get_user_model

from django.db import transaction

from authentic2 import app_settings

from authentic2.models import UserExternalId

def import_users_from_csv(csv_file, ou):

    # TODO prevent illegal attribute-setting

    # TODO do not built the complete list from csv, may be memory-consuming

    # TODO detect the dialect

    # TODO CLI

    reader = csv.reader(csv_file) # XXX dialect

    rows = list(reader)

    app_id_label = app_settings.A2_USERS_CSV_IMPORT_APP_ID_LABEL

    try:

        rows[0].index(app_id_label)

    except ValueError:

        return -1, 'The csv file must contain an app_id column.'

    with transaction.atomic():

        User = get_user_model()

        for i in range(1,len(rows)):

            user_data = dict(zip(rows[0], rows[i]))

            app_id = user_data.pop(app_id_label)

            user = User.objects.create(**user_data)

            user.ou = ou

            user_external_id = UserExternalId.objects.create(

                user=user, external_id=app_id, source='import')

    return 0, 'Created %s users.' % (len(rows)-1)

class UsersCsvImportForm(forms.Form):

    csv_file = forms.FileField(

        label=_('CSV file'),

        help_text=_('CSV file for the import.')

    )

    ou = forms.ChoiceField(

        label=_('Organizational unit'),

        help_text=_('The newly created users will be registered to the chosen OU.'),

        choices=utils.get_ou_choices(),

        initial=get_default_ou().uuid

    )

{% extends "authentic2/manager/form.html" %}

{% load i18n %}

{% block page-title %}

  {{ block.super }} - {{ view.title }}

{% endblock %}

{% block breadcrumb %}

  {{ block.super }}

  <a href="..">{% trans "Users" %}</a>

  <a href="">{{ view.title }}</a>

{% endblock %}

{% block content %}

  {% trans "Comma-separated value file with named columns:" %}

  <ul>

    <li>{{ app_id_label }}</li>

  {% for attr in attributes %}

    <li>{{ attr.name }}</li>

  {% endfor %}

  </ul>

  <form enctype="multipart/form-data" method="post">

    {% csrf_token %}

    {{ form.as_p }}

    {% if form.instance and form.instance.id %}

      <button class="submit-button">{% trans "Submit" %}</button>

      <button class="cancel-button" name="cancel" formnovalidate>{% trans "Cancel" %}</button>

    {% else %}

      <button class="submit-button">{% trans "Create" %}</button>

    {% endif %}

  </form>

{% endblock %}

        url(r'^users/import/$',

            user_views.users_csv_import, name='a2-manager-users-import'),

from django.views.generic import View, FormView

    UserChangePasswordForm, ChooseUserRoleForm, UserRoleSearchForm, UserChangeEmailForm,

    UsersCsvImportForm)

class UsersCsvImportView(FormView):

    form_class = UsersCsvImportForm

    template_name = 'authentic2/manager/users_csv_import.html'

    title = _('CSV import')

    success_url = '..'

    permissions = ['custom_user.add_user']

    def get_context_data(self, **kwargs):

        from authentic2.models import Attribute

        ctx = super(UsersCsvImportView, self).get_context_data()

        ctx.update({

            'attributes': Attribute.objects.all(),

            'app_id_label': a2_app_settings.A2_USERS_CSV_IMPORT_APP_ID_LABEL

        })

        return ctx

    def form_valid(self, form):

        from authentic2.data_transfer import import_users_from_csv

        ou_uuid = form.cleaned_data.get('ou')

        ou = get_ou_model().objects.get(uuid=ou_uuid) or get_default_ou()

        csv_file = form.cleaned_data.get('csv_file')

        if csv_file:

            status, message = import_users_from_csv(csv_file, ou)

        else:

            status, message = -1, 'Please provide a csv file.'

        # XXX django message

        return super(UsersCsvImportView, self).form_valid(form)

users_csv_import = UsersCsvImportView.as_view()

@GlobalCache(timeout=10)

def get_ou_choices():

    Ou = get_ou_model()

    return [(ou.uuid, ou.name) for ou in Ou.objects.all()]

def example_csv():

    # todo

    pass

app_id,first_name,last_name

123,Hervé,Dupont

456,René,Durant