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.
- Should be able to create new projects.
+ {% if request.user.is_superuser %}
+ - Django Admin
+ {% endif %}
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