Andal.LND/edx-platform
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add a get_course_cohorts function and test

Browse Source
This commit is contained in:
Victor Shnayder
2013-01-19 14:58:46 -05:00
committed by Victor Shnayder
parent 90a5703419
commit 3488083e6b
2 changed files with 53 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
common/djangoapps/course_groups/models.py
View File

@@ -40,15 +40,28 @@ def get_cohort(user, course_id):
Returns:
A CourseUserGroup object if the User has a cohort, or None.
"""
group_type = CourseUserGroup.COHORT
try:
group = CourseUserGroup.objects.get(course_id=course_id, group_type=group_type,
users__id=user.id)
group = CourseUserGroup.objects.get(course_id=course_id,
group_type=CourseUserGroup.COHORT,
users__id=user.id)
except CourseUserGroup.DoesNotExist:
group = None
if group:
return group
# TODO: add auto-cohorting logic here
return None
def get_course_cohorts(course_id):
"""
Get a list of all the cohorts in the given course.
Arguments:
course_id: string in the format 'org/course/run'
Returns:
A list of CourseUserGroup objects. Empty if there are no cohorts.
"""
return list(CourseUserGroup.objects.filter(course_id=course_id,
group_type=CourseUserGroup.COHORT))