django-orchestra/orchestra/apps/users/roles/admin.py

from django.contrib import messages
from django.contrib.admin.util import unquote, get_deleted_objects
from django.contrib.admin.templatetags.admin_urls import add_preserved_filters
from django.core.urlresolvers import reverse
from django.db import router
from django.http import Http404, HttpResponseRedirect
from django.template.response import TemplateResponse
from django.shortcuts import redirect
from django.utils.encoding import force_text
from django.utils.html import escape
from django.utils.translation import ugettext, ugettext_lazy as _

from orchestra.admin.utils import get_modeladmin

from .forms import role_form_factory
from ..models import User


class RoleAdmin(object):
    model = None
    name = ''
    url_name = ''
    form = None
    
    def __init__(self, user=None):
        self.user = user
    
    @property
    def exists(self):
        try:
            return getattr(self.user, self.name)
        except self.model.DoesNotExist:
            return False
    
    def get_user(self, request, object_id):
        try:
            user = User.objects.get(pk=unquote(object_id))
        except User.DoesNotExist:
            opts = self.model._meta
            raise Http404(
                _('%(name)s object with primary key %(key)r does not exist.') %
                {'name': force_text(opts.verbose_name), 'key': escape(object_id)}
            )
        return user
    
    def change_view(self, request, object_id):
        modeladmin = get_modeladmin(User)
        user = self.get_user(request, object_id)
        self.user = user
        obj = None
        exists = self.exists
        if exists:
            obj = getattr(user, self.name)
        form_class = self.form if self.form else role_form_factory(self)
        form = form_class(instance=obj)
        opts = User._meta
        app_label = opts.app_label
        title = _("Add %s for user %s" % (self.name, user))
        action = _("Create")
        # User has submitted the form
        if request.method == 'POST':
            form = form_class(request.POST, instance=obj)
            form.user = user
            if form.is_valid():
                obj = form.save()
                context = {
                    'name': obj._meta.verbose_name,
                    'obj': obj,
                    'action': _("saved" if exists else "created")
                }
                modeladmin.log_change(request, request.user, "%s saved" % self.name.capitalize())
                msg = _('The role %(name)s for user "%(obj)s" was %(action)s successfully.') % context
                modeladmin.message_user(request, msg, messages.SUCCESS)
                url = 'admin:%s_%s_change' % (opts.app_label, opts.module_name)
                if not "_continue" in request.POST:
                    return redirect(url, object_id)
                exists = True
        
        if exists:
            title = _("Change %s %s settings" % (user, self.name))
            action = _("Save")
            form = form_class(instance=obj)
        
        context = {
            'title': title,
            'opts': opts,
            'app_label': app_label,
            'form': form,
            'action': action,
            'role': self,
            'roles': [ role(user=user) for role in modeladmin.roles ],
            'media': modeladmin.media
        }
        
        template = 'admin/users/user/role.html'
        app = modeladmin.admin_site.name
        return TemplateResponse(request, template, context, current_app=app)
    
    def delete_view(self, request, object_id):
        "The 'delete' admin view for this model."
        opts = self.model._meta
        app_label = opts.app_label
        modeladmin = get_modeladmin(User)
        user = self.get_user(request, object_id)
        obj = getattr(user, self.name)
        
        using = router.db_for_write(self.model)
        
        # Populate deleted_objects, a data structure of all related objects that
        # will also be deleted.
        (deleted_objects, perms_needed, protected) = get_deleted_objects(
            [obj], opts, request.user, modeladmin.admin_site, using)
        
        if request.POST:  # The user has already confirmed the deletion.
            if perms_needed:
                raise PermissionDenied
            obj_display = force_text(obj)
            modeladmin.log_deletion(request, obj, obj_display)
            modeladmin.delete_model(request, obj)
            post_url = reverse('admin:users_user_change', args=(user.pk,))
            preserved_filters = modeladmin.get_preserved_filters(request)
            post_url = add_preserved_filters(
                {'preserved_filters': preserved_filters, 'opts': opts}, post_url
            )
            return HttpResponseRedirect(post_url)
        
        object_name = force_text(opts.verbose_name)
        
        if perms_needed or protected:
            title = _("Cannot delete %(name)s") % {"name": object_name}
        else:
            title = _("Are you sure?")
        
        context = {
            "title": title,
            "object_name": object_name,
            "object": obj,
            "deleted_objects": deleted_objects,
            "perms_lacking": perms_needed,
            "protected": protected,
            "opts": opts,
            "app_label": app_label,
            'preserved_filters': modeladmin.get_preserved_filters(request),
            'role': self,
        }
        return TemplateResponse(request, 'admin/users/user/delete_role.html',
                context, current_app=modeladmin.admin_site.name)