Private
Public Access
1
0

adding time team import

This commit is contained in:
Sander Roosendaal
2022-07-05 07:16:21 +02:00
parent 65f00b6ec3
commit 273b9a9bb6
8 changed files with 96 additions and 3 deletions

View File

@@ -10,7 +10,7 @@ import collections
from django.views.generic.edit import CreateView
from boatmovers.models import Athlete, Crew, Race, Result
import boatmovers.tasks as tasks
from boatmovers.forms import CsvForm
from boatmovers.forms import CsvForm, TimeTeamForm
from boatmovers.scrapers import *
#from rowers.rows import handle_uploaded_file
import django_rq
@@ -138,6 +138,39 @@ def race_verify(request, id=0):
return HttpResponseRedirect(reverse('race_view',kwargs={'id':race.id}))
def race_handle_timeteam(request,id=0):
race = get_object_or_404(Race, pk=id)
if race.verified or race.processed:
messages.error(request,"Cannot upload CSV file for processed or verified race")
url = reverse("race_view",kwargs={'id':id})
return HttpResponseRedirect(url)
form = TimeTeamForm(initial={'resulturl':race.resulturl})
if request.method == 'POST':
form = TimeTeamForm(request.POST)
if form.is_valid():
startorder = form.cleaned_data['startorder']
url = form.cleaned_data['resulturl']
race.resulturl = url
race.save()
job = queue.enqueue(
tasks.handle_timeteam,
url,race.id,race.gender,startorder)
messages.info(request,'URL has been submitted for processing')
url = reverse('race_view',kwargs={'id':race.id})
return HttpResponseRedirect(url)
return render(request,
'timeteamform.html',
{
'race':race,
'form':form,
})
def handle_uploaded_file(f):
with open('media/results.csv', 'wb+') as destination:
for chunk in f.chunks():