diff --git a/rowers/plannedsessions.py b/rowers/plannedsessions.py index 04f6c194..1155cf53 100644 --- a/rowers/plannedsessions.py +++ b/rowers/plannedsessions.py @@ -798,7 +798,12 @@ def race_can_adddiscipline(r,race): def race_can_withdraw(r,race): - records = VirtualRaceResult.objects.filter( + if race.sessiontype == 'race': + recordobj = VirtualRaceResult + else: + recordobj = IndoorVirtualRaceResult + + records = recordobj.objects.filter( userid=r.id, race=race ) diff --git a/rowers/views.py b/rowers/views.py index 307ba0de..96687396 100644 --- a/rowers/views.py +++ b/rowers/views.py @@ -16124,6 +16124,7 @@ def virtualevent_view(request,id=0): @login_required() def virtualevent_withdraw_view(request,id=0,recordid=None): r = getrower(request.user) + try: race = VirtualRace.objects.get(id=id) except VirtualRace.DoesNotExist: