api/course_register_views.py

@@ -6,7 +6,7 @@ from django_query import get_query_value

                 from django_response import response
              

                 from paginator import pagination
              

                 
              

                -from registration.models import BranchCampusAdministratorInfo, CourseRegisterInfo, UserCourseRegisterInfo
              

                +from registration.models import BranchCampusAdministratorInfo, CourseRegisterInfo, UserCourseRegisterInfo, BranchCampusInfo
              

                 
              

                 
              

                 def course_register(request):
              
@@ -55,7 +55,7 @@ def course_register(request):

                     #     'course_name': course_name,
              

                     #     'fields': registers,
              

                     # })
              

                -
              

                +    
              

                     user_course_register, _ = UserCourseRegisterInfo.objects.get_or_create(user_id=user_id, course_id=course_id)
              

                     if not user_course_register.campus_id:
              

                         user_course_register.campus_id = admin.campus_id if admin else ''
              
@@ -73,11 +73,21 @@ def course_register(request):

                 
              

                 
              

                 def course_register_list(request):
              

                +    admin_id = request.POST.get('admin_id', '')
              

                     course_id = request.POST.get('course_id', '')
              

                     page = request.POST.get('page', 1)
              

                     num = request.POST.get('num', 20)
              

                 
              

                -    registers = UserCourseRegisterInfo.objects.filter(course_id=course_id, status=True).order_by('-pk')
              

                +    registers = UserCourseRegisterInfo.objects.filter(status=True).order_by('-pk')
              

                +
              

                +    admin = BranchCampusAdministratorInfo.objects.get(admin_id=admin_id)
              

                +    campus = BranchCampusInfo.objects.get(campus_id=admin.campus_id)
              

                +    if campus.campus_type == BranchCampusInfo.BRANCH:
              

                +        registers = UserCourseRegisterInfo.objects.filter(campus_id=admin.campus_id)
              

                +    
              

                +    if course_id:
              

                +        registers = UserCourseRegisterInfo.objects.filter(course_id=course_id)
              

                +
              

                     count = registers.count()
              

                     registers, left = pagination(registers, page, num)
              

                     registers = [register.data for register in registers]