Produits Entr'ouvert » U-Auth

class FederationsView(OrganizationMixin, TemplateView):

    template_name = 'organization/federations.html'

    def get_success_url(self):

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

    def get_context_data(self, **kwargs):

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

        org = ctx['organization']

        federations_dir = os.path.join(settings.ORGANIZATIONS_DIR,

                                    org.slug, 'federations')

        ctx['federations_dir'] = federations_dir

        if os.path.exists(federations_dir):

            ctx['federations'] = os.listdir(federations_dir)

        else:

            ctx['federations'] = []

        return ctx

federations = FederationsView.as_view()

class FederationsAdd(UploadMixin, FederationsView):

    form_class = FederationForm

    filename_param = 'federation_file'

    upload_dir = 'federations'

federations_add = FederationsAdd.as_view()

class FederationsDelete(TemplateView):

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

        ctx = self.get_context_data(**kwargs)

        s = request.GET.get('federation')

        if os.path.exists(os.path.join(ctx['federations_dir'], f)):

            try:

                os.remove(os.path.join(ctx['federations_dir'], f))

                messages.info(request, _('Federation file %s successfully removed') % f)

            except IOError:

                messages.error(request, _('An error occured while removing file %s') % f)

        else:

            messages.error(request, _('Unknown federation file %s') % f)

        return redirect(self.get_success_url())

federations_delete = FederationsDelete.as_view()