adding time team import
This commit is contained in:
@@ -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():
|
||||
|
||||
Reference in New Issue
Block a user