Produits Entr'ouvert » U-Auth

import os

from django.conf import settings

from .forms import *

class ThemeView(OrganizationMixin, TemplateView):

    template_name = 'organization/theme.html'

    def get_success_url(self):

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

    def get_context_data(self, **kwargs):

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

        organization = ctx['organization']

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

                                organization.slug, 'templates')

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

                                organization.slug, 'static')

        ctx['templates'] = []

        ctx['statics'] = []

        if os.path.exists(templates_dir):

            ctx['templates'] = os.listdir(templates_dir)

        if os.path.exists(statics_dir):

            ctx['statics'] = os.listdir(statics_dir)

        ctx['templates_dir'] = templates_dir

        ctx['statics_dir'] = statics_dir

        return ctx

theme = ThemeView.as_view()

class UploadMixin(object):

    template_name = "organization/upload.html"

    def get_context_data(self, **kwargs):

        ctx = super(UploadMixin, 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

        organization = context['organization']

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

                                       organization.slug, self.upload_dir)

        if form.is_valid():

            data = form.cleaned_data[self.filename_param]

            if not os.path.exists(destination_dir):

                os.makedirs(destination_dir)

            try:

                with open(os.path.join(destination_dir, data.name), 'w') as template:

                    template.write(data.read())

                messages.info(request, _('File "%s" successfully uploaded') % data.name)

            except OSError:

                messages.error(request, _('An error occured while uploading file "%s"') % data.name)

            return redirect(self.get_success_url())

        else:

            return self.render_to_response(context)

class TemplateUpload(UploadMixin, ThemeView):

    form_class = TemplateForm

    filename_param = 'template_file'

    upload_dir = 'templates'

template_upload = TemplateUpload.as_view()

class TemplateDelete(ThemeView):

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

        ctx = self.get_context_data(**kwargs)

        template = request.GET.get('template')

        if os.path.exists(os.path.join(ctx['templates_dir'], template)):

            try:

                os.remove(os.path.join(ctx['templates_dir'], template))

                messages.info(request, _('Template %s successfully removed') % template)

            except IOError:

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

        else:

            messages.error(request, _('Unknown template %s') % template)

        return redirect(self.get_success_url())

template_delete = TemplateDelete.as_view()

class StaticUpload(UploadMixin, ThemeView):

    form_class = StaticForm

    filename_param = 'static_file'

    upload_dir = 'static'

static_upload = StaticUpload.as_view()

class StaticDelete(ThemeView):

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

        ctx = self.get_context_data(**kwargs)

        static = request.GET.get('static')

        if os.path.exists(os.path.join(ctx['statics_dir'], static)):

            try:

                os.remove(os.path.join(ctx['statics_dir'], static))

                messages.info(request, _('Static file %s successfully removed') % static)

            except IOError:

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

        else:

            messages.error(request, _('Unknown static %s') % static)

        return redirect(self.get_success_url())

static_delete = StaticDelete.as_view()