diff --git a/interface/templates/interface/manage.html b/interface/templates/interface/manage.html index 011a4ea..43db88f 100644 --- a/interface/templates/interface/manage.html +++ b/interface/templates/interface/manage.html @@ -11,6 +11,9 @@ Sorry, not much you can do here yet.
diff --git a/interface/views.py b/interface/views.py index 2fc6237..8624e32 100644 --- a/interface/views.py +++ b/interface/views.py @@ -122,6 +122,10 @@ def register(request): request.session['ensemble'] = registered[code] return redirect('ensemble_detail') + if request.user.is_superuser and code: + request.session['ensemble'] = models.Ensemble.objects.get(code=code).pk + return redirect('ensemble_detail') + if request.method == "POST": form = forms.CodeForm(request.POST) @@ -141,16 +145,20 @@ def register(request): else: form = forms.CodeForm(initial=request.GET) - - current = models.Ensemble.objects.filter(pk__in=registered.values()) + + if request.user.is_superuser: + current = models.Ensemble.objects.all() + else: + current = models.Ensemble.objects.filter(pk__in=registered.values()) return render(request, 'interface/register.html', {'form': form, 'current': current}) def on_login(sender, **kwargs): + user = kwargs['user'] request = kwargs['request'] registered = request.session.get('registered', {}) - for e in kwargs['user'].ensembles.all(): + for e in user.ensembles.all(): if not e.code in registered: registered[e.code] = e.pk request.session['registered'] = registered