from django.views.generic import TemplateView from django.shortcuts import render, redirect from django.contrib import messages from django.utils.translation import ugettext as _ from django.conf import settings from .forms import EmailSettingForm, LDAPSettingForm, BasicSettingForm, \ TerminalSettingForm from .models import Setting from .mixins import AdminUserRequiredMixin from .signals import ldap_auth_enable class BasicSettingView(AdminUserRequiredMixin, TemplateView): form_class = BasicSettingForm template_name = "common/basic_setting.html" def get_context_data(self, **kwargs): context = { 'app': _('Settings'), 'action': _('Basic setting'), 'form': self.form_class(), } kwargs.update(context) return super().get_context_data(**kwargs) def post(self, request): form = self.form_class(request.POST) if form.is_valid(): form.save() msg = _("Update setting successfully, please restart program") messages.success(request, msg) return redirect('settings:basic-setting') else: context = self.get_context_data() context.update({"form": form}) return render(request, self.template_name, context) class EmailSettingView(AdminUserRequiredMixin, TemplateView): form_class = EmailSettingForm template_name = "common/email_setting.html" def get_context_data(self, **kwargs): context = { 'app': _('Settings'), 'action': _('Email setting'), 'form': self.form_class(), } kwargs.update(context) return super().get_context_data(**kwargs) def post(self, request): form = self.form_class(request.POST) if form.is_valid(): form.save() msg = _("Update setting successfully, please restart program") messages.success(request, msg) return redirect('settings:email-setting') else: context = self.get_context_data() context.update({"form": form}) return render(request, self.template_name, context) class LDAPSettingView(AdminUserRequiredMixin, TemplateView): form_class = LDAPSettingForm template_name = "common/ldap_setting.html" def get_context_data(self, **kwargs): context = { 'app': _('Settings'), 'action': _('LDAP setting'), 'form': self.form_class(), } kwargs.update(context) return super().get_context_data(**kwargs) def post(self, request): form = self.form_class(request.POST) if form.is_valid(): form.save() if "AUTH_LDAP" in form.cleaned_data: ldap_auth_enable.send(form.cleaned_data["AUTH_LDAP"]) msg = _("Update setting successfully, please restart program") messages.success(request, msg) return redirect('settings:ldap-setting') else: context = self.get_context_data() context.update({"form": form}) return render(request, self.template_name, context) class TerminalSettingView(AdminUserRequiredMixin, TemplateView): form_class = TerminalSettingForm template_name = "common/terminal_setting.html" def get_context_data(self, **kwargs): command_storage = settings.TERMINAL_COMMAND_STORAGE context = { 'app': _('Settings'), 'action': _('Terminal setting'), 'form': self.form_class(), 'command_storage': command_storage, } kwargs.update(context) return super().get_context_data(**kwargs) def post(self, request): form = self.form_class(request.POST) if form.is_valid(): form.save() msg = _("Update setting successfully, please restart program") messages.success(request, msg) return redirect('settings:terminal-setting') else: context = self.get_context_data() context.update({"form": form}) return render(request, self.template_name, context)