forked from 42CTF/website
49 lines
2.4 KiB
Python
49 lines
2.4 KiB
Python
|
from django.shortcuts import render, get_object_or_404
|
||
|
|
from django.contrib.auth.models import User
|
||
|
|
from django.contrib.auth.models import timezone
|
||
|
|
from .models import Category, CTF, CTF_flags
|
||
|
|
from .forms import submit_flag
|
||
|
|
from accounts.models import UserProfileInfo
|
||
|
|
|
||
|
|
def category(request, cat_slug):
|
||
|
|
cat = get_object_or_404(Category, slug=cat_slug)
|
||
|
|
ctfs = CTF.objects.filter(category=cat).order_by('-points')
|
||
|
|
for ex in ctfs:
|
||
|
|
ex.solved_num = CTF_flags.objects.filter(ctf=ex).count()
|
||
|
|
ex.solved = ex.solved_by(request.user)
|
||
|
|
return render(request, 'ctfs/ctfs_list.html', {'ctfs' : ctfs, 'cat' : cat})
|
||
|
|
|
||
|
|
def ctf(request, cat_slug, ctf_slug):
|
||
|
|
ctf_info = get_object_or_404(CTF, slug=ctf_slug)
|
||
|
|
flagged = False
|
||
|
|
solved_list = CTF_flags.objects.filter(ctf=ctf_info).order_by('flag_date')
|
||
|
|
if request.user.is_authenticated:
|
||
|
|
if CTF_flags.objects.filter(user=request.user, ctf=ctf_info):
|
||
|
|
flagged = True
|
||
|
|
if request.method == 'POST':
|
||
|
|
if request.user.is_authenticated:
|
||
|
|
form = submit_flag(data=request.POST)
|
||
|
|
if flagged == False and form.is_valid():
|
||
|
|
if CTF.objects.filter(flag=request.POST.get('flag'), slug=ctf_slug):
|
||
|
|
new = CTF_flags(user = request.user, ctf = ctf_info, flag_date = timezone.now())
|
||
|
|
new.save()
|
||
|
|
profil = UserProfileInfo.objects.get(user=request.user)
|
||
|
|
profil.last_submission_date = timezone.now()
|
||
|
|
profil.score += ctf_info.points
|
||
|
|
profil.save()
|
||
|
|
return render(request, 'ctfs/ctf_info.html', {'form' : form, 'ctf' : ctf_info, 'solved_list': solved_list, 'valitated': True})
|
||
|
|
else:
|
||
|
|
return render(request, 'ctfs/ctf_info.html', {'form' : form, 'ctf' : ctf_info, 'solved_list': solved_list, 'failed': True})
|
||
|
|
else:
|
||
|
|
form = submit_flag()
|
||
|
|
return render(request, 'ctfs/ctf_info.html', {'form' : form, 'ctf' : ctf_info, 'solved_list': solved_list, 'alvalitated': True})
|
||
|
|
else:
|
||
|
|
form = submit_flag()
|
||
|
|
return render(request, 'ctfs/ctf_info.html', {'form' : form, 'ctf' : ctf_info, 'solved_list': solved_list})
|
||
|
|
else:
|
||
|
|
form = submit_flag()
|
||
|
|
return render(request, 'ctfs/ctf_info.html', {'form' : form, 'ctf' : ctf_info, 'solved_list': solved_list, 'alvalitated': flagged})
|
||
|
|
|
||
|
|
|
||
|
|
# Create your views here.
|