Produits Entr'ouvert » U-Auth

class UsersImportForm(forms.Form):

    users_file = forms.FileField(_('Users file'))

{% extends 'uauth/base.html' %}

{% load i18n %}

{% block content %}

<form method="post" action="{% url "import-users" organization.slug %}" enctype="multipart/form-data">

  {% csrf_token %}

  {{ form.as_p }}

  <h3>{% trans "Expected file format" %}: CSV</h3>

  <ul>

    <li>{% trans "5 columns: login, first name, last name, expiration date, password" %}</li>

    <li>{% trans "date format: YYYY-MM-DD" %}</li>

    <li>{% trans "if empty password, it will be generated" %}</li>

    <li>{% trans "if user exists, it will be updated" %}</li>

    <li>{% trans "separator: comma" %}</li>

    <li>{% trans "encoding: utf-8" %}</li>

    <li>{% trans "first line is ignored" %}</li>

  </ul>

  <p>Example:

    <div class="example">

      "login", "first name", "last name", "expiration", "password"<br />

      "foo", "Foo", "User", "2015-12-31", "secret" <br />

      "bar", "User", "Bar", "2015-04-30", ""<br />

      "test", "", "", "", ""<br />

    </div>

  </p>

  <p><button name="upload">{% trans "Import" %}</button>

</form>

{% endblock %}

<a href="{% url "import-users" organization.slug %}" rel="popup">{% trans "Import users" %}</a>

    url(r'^users/import$', import_users, name='import-users'),

def create_or_update_users(data):

    created = 0

    updated = 0

    for user in data:

        try:

            account = LocalAccount.objects.get(username=user['username'])

            if not user['password']:

                del user['password']

            account.__dict__.update(user)

            account.save()

            updated += 1

        except LocalAccount.DoesNotExist:

            if create_user(user):

                created += 1

    return created, updated

import csv

import datetime

from django.shortcuts import render, redirect

from django.views.generic import DetailView, View

from .utils import create_user, create_or_update_users

from .forms import LocalAccountCreateForm, LocalAccountForm, UsersImportForm

            return redirect(self.get_success_url())

class ImportUsersView(OrganizationMixin, TemplateView):

    form_class = UsersImportForm

    template_name = 'organization/import_users.html'

    def get_context_data(self, **kwargs):

        ctx = super(ImportUsersView, self).get_context_data(**kwargs)

        ctx['form'] = self.form_class()

        return ctx

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

        form = self.form_class(request.POST, request.FILES)

        context = self.get_context_data(**kwargs)

        context['form'] = form

        if form.is_valid():

            data = form.cleaned_data['users_file']

            dialect = csv.Sniffer().sniff(data.read(1024))

            data.seek(0)

            reader = csv.reader(data, dialect)

            reader.next()

            users = []

            for row in reader:

                try:

                    user = {'username': row[0].strip(),

                            'first_name': row[1].strip(),

                            'last_name': row[2].strip(),

                            'password': row[4].strip(),

                            'organization': context['organization']

                        }

                except IndexError:

                    # ignore wrong lines

                    continue

                try:

                    user['expiration_date'] = datetime.datetime.strptime(row[3], '%Y-%m-%d')

                except ValueError:

                    pass

                users.append(user)

            created, updated = create_or_update_users(users)

            if created:

                messages.info(request, _('%s accounts added' % created))

            if updated:

                messages.info(request, _('%s accounts updated' % updated))

            return redirect(self.get_success_url())

        else:

            return self.render_to_response(context)

import_users = ImportUsersView.as_view()

}

div.example {

    background: #eee;

    border: 1px solid #bbb;

    font-family: Monospace;