Produits Entr'ouvert » U-Auth

from django.utils.translation import ugettext as _

from django.core.urlresolvers import reverse_lazy

from django.http import HttpResponseRedirect

from django.views.generic.base import TemplateView

from django.views.generic.list import ListView

from django.views.generic.edit import FormView, UpdateView

from django.views.generic import DetailView

from django.contrib import messages

from django_tables2 import RequestConfig

from .utils import create_user

from .models import LocalAccount, Organization

from .forms import LocalAccountCreateForm, LocalAccountForm

from .tables import AccountTable

class OrganizationMixin(object):

    def get_queryset(self):

        qs = super(OrganizationMixin, self).get_queryset()

        return qs.filter(organization__slug=self.kwargs['organization_slug'])

    def get_success_url(self):

        return reverse_lazy('manage-users', kwargs={'organization_slug': self.kwargs['organization_slug']})

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

        ctx = super(OrganizationMixin, self).get_context_data(*args, **kwargs)

        ctx['organization'] = Organization.objects.get(slug=self.kwargs['organization_slug'])

        return ctx

class ManageView(TemplateView):

    template_name = 'organization/manage.html'

manage = ManageView.as_view()

class UsersPageView(OrganizationMixin, ListView):

    template_name = 'organization/users.html'

    model = LocalAccount

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

        context = super(UsersPageView, self).get_context_data(*args, **kwargs)

        table = AccountTable(context['object_list'])

        RequestConfig(self.request).configure(table)

        context['table'] = table

        return context

users = UsersPageView.as_view()

class UsersCreateView(OrganizationMixin, FormView):

    template_name = 'organization/create-users.html'

    form_class = LocalAccountCreateForm

    def form_valid(self, form):

        data = form.cleaned_data

        data['organization'] = Organization.objects.get(slug=self.kwargs['organization_slug'])

        accounts_number = data.pop('accounts_number')

        accounts_number_start = data.pop('accounts_number_start')

        if accounts_number_start < 1:

            accounts_number_start = 0

        username = data['username']

        if accounts_number > 1:

            for index in xrange(accounts_number_start, accounts_number + accounts_number_start):

                data['username'] = '%s-%s' % (username, index)

                if not create_user(data):

                    messages.error(self, request, _('Error while creating user %s' % data['username']))

                    break

            messages.info(self.request, _('%s users added successfully' % accounts_number))

        else:

            if create_user(data):

                messages.info(self.request, _('User "%s" successfully created' % data['username']))

            else:

                messages.error(self.request, _('Error occured while creating user "%s"' % data['username']))

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

create_users = UsersCreateView.as_view()

class ShowUserView(OrganizationMixin, DetailView):

    model = LocalAccount

    template_name = 'organization/view_user.html'

view_user = ShowUserView.as_view()

class UserEditView(OrganizationMixin, UpdateView):

    template_name = 'organization/edit_user.html'

    model = LocalAccount

    form_class = LocalAccountForm

    def form_valid(self, form):

        username = self.object.username

        if 'delete' in self.request.POST:

            self.object.delete()

            messages.info(self.request, _('Account "%s" successfully deleted' % username))

            return HttpResponseRedirect(self.get_success_url())

        else:

            messages.info(self.request, _('Account "%s" successfully updated' % username))

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

edit_user = UserEditView.as_view()