Enroll allowed users on account creation

IONISx · Mar 14, 2015 · 6a5b5d4 · 6a5b5d4
1 parent 870a39c
commit 6a5b5d4
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/common/djangoapps/third_party_auth/pipeline.py b/common/djangoapps/third_party_auth/pipeline.py
@@ -84,7 +84,7 @@ def B(*args, **kwargs):
     ItemAlreadyInCartException,
     AlreadyEnrolledInCourseException
 )
-from student.models import CourseEnrollment, CourseEnrollmentException
+from student.models import CourseEnrollment, CourseEnrollmentException, CourseEnrollmentAllowed
 from course_modes.models import CourseMode
 from opaque_keys.edx.keys import CourseKey
 
@@ -474,6 +474,11 @@ def create_user_from_oauth(strategy, details, user, is_new, *args, **kwargs):
             log.error("UserProfile creation failed for user {id}.".format(id=user.id))
             raise
 
+        ceas = CourseEnrollmentAllowed.objects.filter(email=user.email)
+        for cea in ceas:
+            if cea.auto_enroll:
+                CourseEnrollment.enroll(user, cea.course_id)
+
         create_comments_service_user(user)