add X (test) course for admins

diff --git a/teleforma/views/ae.py b/teleforma/views/ae.py
index ee2adcfd1fc72b95bac24dfe57ddfbf9704a2adb..d43eacbf953b483e7be03b188253fa14df7db763 100644 (file)
--- a/teleforma/views/ae.py
+++ b/teleforma/views/ae.py
@@ -36,14 +36,14 @@
 from teleforma.views.core import *
 
 
-def format_ae_courses(courses, course=None, queryset=None, types=None):
+def format_ae_courses(courses, course=None, queryset=None, types=None, admin=False):
     if queryset:
         for c in queryset:
-            if c and c.code != 'X':
+            if c and (c.code != 'X' or admin == True):
                 courses.append({'course': c, 'types': c.types.all(),
                 'date': c.date_modified, 'number': c.number})
     elif course:
-        if course.code != 'X':
+        if course.code != 'X' or admin == True:
             courses.append({'course': course, 'types': course.types.all(),
             'date': course.date_modified, 'number': course.number})
 
@@ -76,7 +76,7 @@ def get_ae_courses(user, date_order=False, num_order=False):
                             queryset=magistrals)
 
     elif user.is_staff or user.is_superuser:
-        courses = format_ae_courses(courses, queryset=Course.objects.all())
+        courses = format_ae_courses(courses, queryset=Course.objects.all(), admin=True)
     else:
         courses = None