11import datetime
2+ from operator import truediv
23
34from django .contrib import messages
45from django .http import Http404 , HttpResponseRedirect
@@ -55,13 +56,20 @@ def new_submission(request, event):
5556
5657
5758def edit_submission (request , event , submission_id ):
59+ back = request .GET .get ('r' , '/' )
5860 if request .user .is_authenticated :
5961 player = get_player_info_for_user (request .user )
62+ player_id = player .get ('id' )
63+ if not player_id :
64+ return HttpResponseRedirect (back )
6065 else :
61- return HttpResponseRedirect ('/' )
66+ return HttpResponseRedirect (back )
6267 event = get_object_or_404 (Event , slug = event )
63- submission = get_object_or_404 (Submission , id = submission_id , event = event , hidden = False , players__in = [player ['id' ]])
64-
68+ submission = get_object_or_404 (Submission , id = submission_id , event = event , hidden = False , players__in = [player_id ])
69+ if submission :
70+ submission_found = True
71+ else :
72+ submission_found = False
6573 if request .method == 'POST' :
6674 form = SubmissionForm (request .POST )
6775
@@ -83,7 +91,7 @@ def edit_submission(request, event, submission_id):
8391 else :
8492 request .session ['previous_form' ] = request .POST
8593 messages .add_message (request , messages .ERROR , _ ('Ilmoittautuminen epäonnistui, tarkista lomake.' ))
86- return HttpResponseRedirect (request .GET .get ('next' , '/' ))
94+ return HttpResponseRedirect (request .GET .get ('next' , '/' ) + "?r=" + back )
8795
8896 if request .method == 'GET' :
8997
@@ -99,6 +107,8 @@ def edit_submission(request, event, submission_id):
99107 'require_authentication' : True ,
100108 'form' : form ,
101109 'submission_id' : submission_id ,
110+ 'back' : back ,
111+ 'submission_found' : submission_found
102112 }
103113 return render (request , 'marathon/edit_submission.html' , context )
104114 raise Http404 ('Tämmöstä ei oo' )
0 commit comments