Locking after contest, oops

Author: singerng

ctfapp/decorators.py

@@ -2,7 +2,7 @@
 from django.core.exceptions import PermissionDenied
 from django.shortcuts import render
 
-from ctfapp.utils.time import before_start
+from ctfapp.utils.time import before_start, after_end
 
 from functools import wraps
 
@@ -36,3 +36,13 @@ def wrap(request, *args, **kwargs):
             return view(request, *args, **kwargs)
 
     return wrap
+
+def lock_after_contest(view):
+    @wraps(view, assigned=available_attrs(view))
+    def wrap(request, *args, **kwargs):
+        if after_end() and not request.user.is_staff:
+            return render(request, 'denied.html', {})
+        else:
+            return view(request, *args, **kwargs)
+
+    return wrap

ctfapp/views/submit_problem.py

@@ -8,7 +8,7 @@
 from django.utils import timezone
 
 from ctfapp.models import Problem, CorrectSubmission, IncorrectSubmission
-from ctfapp.decorators import team_required, lock_before_contest
+from ctfapp.decorators import team_required, lock_before_contest, lock_after_contest
 
 # This file handles problem grading and the display for grading.
 
@@ -17,6 +17,7 @@
 @login_required
 @team_required
 @lock_before_contest
+@lock_after_contest
 def submit_problem(request: HttpRequest):
     """
     View for submitting a problem through AJAX.  Login is required.