from __future__ import absolute_import from json import dumps, loads from django.contrib import messages from django.contrib.auth.models import User from django.contrib.auth.views import login, password_change from django.contrib.contenttypes.models import ContentType from django.conf import settings from django.core.exceptions import PermissionDenied from django.core.urlresolvers import reverse from django.http import HttpResponseRedirect from django.shortcuts import redirect, render_to_response from django.template import RequestContext from django.utils.http import urlencode from django.utils.translation import ugettext_lazy as _ from django.views.generic import FormView from django.views.generic.edit import CreateView, DeleteView, UpdateView from django.views.generic.list import ListView from acls.models import AccessEntry from permissions.models import Permission from .forms import (ChoiceForm, EmailAuthenticationForm, LicenseForm, UserForm, UserForm_view) from .settings import LOGIN_METHOD def multi_object_action_view(request): """ Proxy view called first when using a multi object action, which then redirects to the appropiate specialized view """ next = request.POST.get('next', request.GET.get('next', request.META.get('HTTP_REFERER', reverse('main:home')))) action = request.GET.get('action', None) id_list = u','.join([key[3:] for key in request.GET.keys() if key.startswith('pk_')]) items_property_list = [loads(key[11:]) for key in request.GET.keys() if key.startswith('properties_')] if not action: messages.error(request, _(u'No action selected.')) return HttpResponseRedirect(request.META.get('HTTP_REFERER', reverse('main:home'))) if not id_list and not items_property_list: messages.error(request, _(u'Must select at least one item.')) return HttpResponseRedirect(request.META.get('HTTP_REFERER', reverse('main:home'))) # Separate redirects to keep backwards compatibility with older # functions that don't expect a properties_list parameter if items_property_list: return HttpResponseRedirect('%s?%s' % ( action, urlencode({'items_property_list': dumps(items_property_list), 'next': next})) ) else: return HttpResponseRedirect('%s?%s' % ( action, urlencode({'id_list': id_list, 'next': next})) ) def get_obj_from_content_type_string(string): model, pk = string.split(u',') ct = ContentType.objects.get(model=model) return ct.get_object_for_this_type(pk=pk) def assign_remove(request, left_list, right_list, add_method, remove_method, left_list_title=None, right_list_title=None, decode_content_type=False, extra_context=None, grouped=False): left_list_name = u'left_list' right_list_name = u'right_list' if request.method == 'POST': if u'%s-submit' % left_list_name in request.POST.keys(): unselected_list = ChoiceForm(request.POST, prefix=left_list_name, choices=left_list()) if unselected_list.is_valid(): for selection in unselected_list.cleaned_data['selection']: if grouped: flat_list = [] for group in left_list(): flat_list.extend(group[1]) else: flat_list = left_list() label = dict(flat_list)[selection] if decode_content_type: selection_obj = get_obj_from_content_type_string(selection) else: selection_obj = selection try: add_method(selection_obj) except: if settings.DEBUG: raise else: messages.error(request, _(u'Unable to remove %(selection)s.') % { 'selection': label, 'right_list_title': right_list_title}) elif u'%s-submit' % right_list_name in request.POST.keys(): selected_list = ChoiceForm(request.POST, prefix=right_list_name, choices=right_list()) if selected_list.is_valid(): for selection in selected_list.cleaned_data['selection']: if grouped: flat_list = [] for group in right_list(): flat_list.extend(group[1]) else: flat_list = right_list() label = dict(flat_list)[selection] if decode_content_type: selection = get_obj_from_content_type_string(selection) try: remove_method(selection) except: if settings.DEBUG: raise else: messages.error(request, _(u'Unable to add %(selection)s.') % { 'selection': label, 'right_list_title': right_list_title}) unselected_list = ChoiceForm(prefix=left_list_name, choices=left_list()) selected_list = ChoiceForm(prefix=right_list_name, choices=right_list()) context = { 'subtemplates_list': [ { 'name': 'main/generic_form_subtemplate.html', 'grid': 6, 'context': { 'form': unselected_list, 'title': left_list_title or ' ', 'submit_label': _(u'Add'), 'submit_icon_famfam': 'add' } }, { 'name': 'main/generic_form_subtemplate.html', 'grid': 6, 'grid_clear': True, 'context': { 'form': selected_list, 'title': right_list_title or ' ', 'submit_label': _(u'Remove'), 'submit_icon_famfam': 'delete' } }, ], } if extra_context: context.update(extra_context) return render_to_response('main/generic_form.html', context, context_instance=RequestContext(request)) def current_user_details(request): """ Display the current user's details """ form = UserForm_view(instance=request.user) return render_to_response( 'main/generic_form.html', { 'form': form, 'title': _(u'Current user details'), 'read_only': True, }, context_instance=RequestContext(request)) def current_user_edit(request): """ Allow an user to edit his own details """ next = request.POST.get('next', request.GET.get('next', request.META.get('HTTP_REFERER', reverse('common:current_user_details')))) if request.method == 'POST': form = UserForm(instance=request.user, data=request.POST) if form.is_valid(): if User.objects.filter(email=form.cleaned_data['email']).exclude(pk=request.user.pk).count(): messages.error(request, _(u'E-mail conflict, another user has that same email.')) else: form.save() messages.success(request, _(u'Current user\'s details updated.')) return HttpResponseRedirect(next) else: form = UserForm(instance=request.user) return render_to_response( 'main/generic_form.html', { 'form': form, 'next': next, 'title': _(u'Edit current user details'), }, context_instance=RequestContext(request)) def login_view(request): """ Control how the use is to be authenticated, options are 'email' and 'username' """ kwargs = {'template_name': 'main/login.html'} if LOGIN_METHOD == 'email': kwargs['authentication_form'] = EmailAuthenticationForm if not request.user.is_authenticated(): context = {'web_theme_view_type': 'plain'} else: context = {} return login(request, extra_context=context, **kwargs) def license_view(request): """ Display the included LICENSE file from the about menu """ form = LicenseForm() return render_to_response( 'main/generic_detail.html', { 'form': form, 'title': _(u'License'), }, context_instance=RequestContext(request)) def password_change_view(request): """ Password change wrapper for better control """ context = {'title': _(u'Current user password change')} return password_change( request, extra_context=context, template_name='main/password_change_form.html', post_change_redirect=reverse('common:password_change_done'), ) def password_change_done(request): """ View called when the new user password has been accepted """ messages.success(request, _(u'Your password has been successfully changed.')) return redirect('common:current_user_details') class MayanPermissionCheckMixin(object): permissions_required = None def dispatch(self, request, *args, **kwargs): if self.permissions_required: Permission.objects.check_permissions(self.request.user, self.permissions_required) return super(MayanPermissionCheckMixin, self).dispatch(request, *args, **kwargs) class ExtraContextMixin(object): extra_context = {} def get_extra_context(self): return self.extra_context def get_context_data(self, **kwargs): context = super(ExtraContextMixin, self).get_context_data(**kwargs) context.update(self.get_extra_context()) return context class MayanViewMixin(object): post_action_redirect = None def dispatch(self, request, *args, **kwargs): self.next_url = self.request.POST.get('next', self.request.GET.get('next', self.post_action_redirect if self.post_action_redirect else self.request.META.get('HTTP_REFERER', reverse('main:home')))) self.previous_url = self.request.POST.get('previous', self.request.GET.get('previous', self.request.META.get('HTTP_REFERER', reverse('main:home')))) return super(MayanViewMixin, self).dispatch(request, *args, **kwargs) def get_context_data(self, **kwargs): context = super(MayanViewMixin, self).get_context_data(**kwargs) context.update( { 'next': self.next_url, 'previous': self.previous_url } ) return context class SingleObjectEditView(MayanPermissionCheckMixin, ExtraContextMixin, MayanViewMixin, UpdateView): template_name = 'main/generic_form.html' def form_invalid(self, form): result = super(SingleObjectEditView, self).form_invalid(form) try: messages.error(self.request, _('Error saving %s details.') % self.extra_context['object_name']) except KeyError: messages.error(self.request, _('Error saving details.')) return result def form_valid(self, form): result = super(SingleObjectEditView, self).form_valid(form) try: messages.success(self.request, _('%s details saved successfully.') % self.extra_context['object_name'].capitalize()) except KeyError: messages.success(self.request, _('Details saved successfully.')) return result class SingleObjectCreateView(MayanPermissionCheckMixin, ExtraContextMixin, MayanViewMixin, CreateView): template_name = 'main/generic_form.html' def form_invalid(self, form): result = super(SingleObjectCreateView, self).form_invalid(form) try: messages.error(self.request, _('Error creating new %s.') % self.extra_context['object_name']) except KeyError: messages.error(self.request, _('Error creating object.')) return result def form_valid(self, form): result = super(SingleObjectCreateView, self).form_valid(form) try: messages.success(self.request, _('%s created successfully.') % self.extra_context['object_name'].capitalize()) except KeyError: messages.success(self.request, _('New object created successfully.')) return result class SingleObjectDeleteView(MayanPermissionCheckMixin, ExtraContextMixin, MayanViewMixin, DeleteView): template_name = 'main/generic_confirm.html' def get_context_data(self, **kwargs): context = super(SingleObjectDeleteView, self).get_context_data(**kwargs) context.update({'delete_view': True}) return context def delete(self, request, *args, **kwargs): try: result = super(SingleObjectDeleteView, self).delete(request, *args, **kwargs) except Exception as exception: try: messages.error(self.request, _('Error deleting %s.') % self.extra_context['object_name']) except KeyError: messages.error(self.request, _('Error deleting object.')) raise exception else: try: messages.success(self.request, _('%s deleted successfully.') % self.extra_context['object_name'].capitalize()) except KeyError: messages.success(self.request, _('Object deleted successfully.')) return result class SingleObjectListView(MayanPermissionCheckMixin, ExtraContextMixin, MayanViewMixin, ListView): template_name = 'main/generic_list.html' object_permission = None def get_queryset(self): queryset = super(SingleObjectListView, self).get_queryset() if self.object_permission: try: # Check to see if the user has the permissions globally Permission.objects.check_permissions(self.request.user, [self.object_permission]) except PermissionDenied: # No global permission, filter ther queryset per object + permission return AccessEntry.objects.filter_objects_by_access(self.object_permission, self.request.user, queryset) else: # Has the permission globally, return all results return queryset else: return queryset class MultiFormView(FormView): prefixes = {} prefix = None def get_form_kwargs(self, form_name): kwargs = {} kwargs.update({'initial': self.get_initial(form_name)}) kwargs.update({'prefix': self.get_prefix(form_name)}) if self.request.method in ('POST', 'PUT'): kwargs.update({ 'data': self.request.POST, 'files': self.request.FILES, }) return kwargs def _create_form(self, form_name, klass): form_kwargs = self.get_form_kwargs(form_name) form_create_method = 'create_%s_form' % form_name if hasattr(self, form_create_method): form = getattr(self, form_create_method)(**form_kwargs) else: form = klass(**form_kwargs) return form def get_forms(self, form_classes): return dict([(key, self._create_form(key, klass)) for key, klass in form_classes.items()]) def get_initial(self, form_name): initial_method = 'get_%s_initial' % form_name if hasattr(self, initial_method): return getattr(self, initial_method)() else: return self.initial.copy() def get_prefix(self, form_name): return self.prefixes.get(form_name, self.prefix) def get(self, request, *args, **kwargs): form_classes = self.get_form_classes() forms = self.get_forms(form_classes) return self.render_to_response(self.get_context_data(forms=forms)) def forms_valid(self, forms): for form_name, form in forms.items(): form_valid_method = '%s_form_valid' % form_name if hasattr(self, form_valid_method): return getattr(self, form_valid_method)(form) self.all_forms_valid(forms) return HttpResponseRedirect(self.get_success_url()) def forms_invalid(self, forms): return self.render_to_response(self.get_context_data(forms=forms)) def post(self, request, *args, **kwargs): form_classes = self.get_form_classes() form_name = request.POST.get('action') forms = self.get_forms(form_classes) if all([form.is_valid() for form in forms.values()]): return self.forms_valid(forms) else: return self.forms_invalid(forms)