Fixed superuser access

This commit is contained in:
Tris 2020-09-14 13:42:52 +10:00
parent d1698a958c
commit e8e0f1a8ee
2 changed files with 14 additions and 3 deletions

View File

@ -11,6 +11,9 @@
Sorry, not much you can do here yet.
<ul>
<li>Should be able to create new projects.</li>
{% if request.user.is_superuser %}
<li><a href="/admin">Django Admin</a></li>
{% endif %}
</ul>
<br/>

View File

@ -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