Fixed superuser access
This commit is contained in:
parent
d1698a958c
commit
e8e0f1a8ee
@ -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/>
|
||||
|
||||
@ -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
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user