Commit 8dfcbde6 by Őry Máté

dashboard: make acl form work (no error handling)

parent 984e705c
import re
from django.contrib.auth.models import User, Group
from django.core import signing from django.core import signing
from django.http import HttpResponse from django.shortcuts import redirect
from django.utils.translation import ugettext_lazy as _ from django.views.generic import TemplateView, DetailView, UpdateView
from django.views.generic import TemplateView, DetailView
from django_tables2 import SingleTableView from django_tables2 import SingleTableView
from guardian.shortcuts import (get_users_with_perms, get_groups_with_perms, from guardian.shortcuts import (get_users_with_perms, get_groups_with_perms,
get_perms) get_perms, remove_perm, assign_perm)
from tables import VmListTable
from vm.models import Instance from vm.models import Instance
...@@ -29,6 +30,20 @@ class IndexView(TemplateView): ...@@ -29,6 +30,20 @@ class IndexView(TemplateView):
return context return context
def split(t, at):
Split collection at first occurance of given element.
>>> split("FooBar", "B")
('Foo', 'Bar')
>>> split(range(5), 2)
([0, 1], [2, 3, 4])
pos = t.index(at)
return t[:pos], t[pos:]
def first_common_element(a, b): def first_common_element(a, b):
for i in a: for i in a:
if i in b: if i in b:
...@@ -51,12 +66,22 @@ def get_acl_data(obj): ...@@ -51,12 +66,22 @@ def get_acl_data(obj):
'url': obj.get_absolute_url()} 'url': obj.get_absolute_url()}
class VmDetailView(DetailView): def set_acl_level(obj, whom, level):
levels = obj._meta.permissions
levelids = [id for (id, name) in levels]
to_remove, to_add = split(levelids, level)
for p in to_remove:
remove_perm(p, whom, obj)
for p in to_add:
assign_perm(p, whom, obj)
class VmDetailView(UpdateView):
template_name = "dashboard/vm-detail.html" template_name = "dashboard/vm-detail.html"
queryset = Instance.objects.all() queryset = Instance.objects.all()
def get_context_data(self, **kwargs): def get_context_data(self, **kwargs):
context = super(DetailView, self).get_context_data(**kwargs) context = super(VmDetailView, self).get_context_data(**kwargs)
instance = context['instance'] instance = context['instance']
if instance.node: if instance.node:
port = instance.vnc_port port = instance.vnc_port
...@@ -69,6 +94,27 @@ class VmDetailView(DetailView): ...@@ -69,6 +94,27 @@ class VmDetailView(DetailView):
context['acl'] = get_acl_data(instance) context['acl'] = get_acl_data(instance)
return context return context
def post(self, request, *args, **kwargs):
super(VmDetailView, self).post(request, *args, **kwargs)
context = self.get_context_data(**kwargs)
instance = context['instance']
for key, value in request.POST.items():
m = re.match('perm-([ug])-(\d+)', key)
if m:
type, id = m.groups()
entity = {'u': User, 'g': Group}[type].objects.get(id=id)
set_acl_level(instance, entity, value)
name = request.POST['perm-new-name']
value = request.POST['perm-new']
if name:
entity = User.objects.get(username=name)
except User.DoesNotExist:
entity = Group.objects.get(name=name)
set_acl_level(instance, entity, value)
return redirect(instance)
class VmList(SingleTableView): class VmList(SingleTableView):
template_name = "dashboard/vm-list.html" template_name = "dashboard/vm-list.html"
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or sign in to comment