def _get_curated_groups_ids_and_sids(advisor):
sids = []
curated_group_ids = []
for curated_group in CuratedGroup.get_curated_groups_by_owner_id(advisor.id):
curated_group_ids.append(curated_group.id)
sids = sids + CuratedGroup.get_all_sids(curated_group.id)
return curated_group_ids, sids
def get_students_with_alerts(curated_group_id):
offset = get_param(request.args, 'offset', 0)
limit = get_param(request.args, 'limit', 50)
benchmark = get_benchmarker(
f'curated group {curated_group_id} students_with_alerts')
benchmark('begin')
curated_group = CuratedGroup.find_by_id(curated_group_id)
if not curated_group:
raise ResourceNotFoundError(
f'Sorry, no curated group found with id {curated_group_id}.')
if not _can_current_user_view_curated_group(curated_group):
raise ForbiddenRequestError(
f'Current user, {current_user.get_uid()}, cannot view curated group {curated_group.id}'
)
students = Alert.include_alert_counts_for_students(
benchmark=benchmark,
viewer_user_id=current_user.get_id(),
group={'sids': CuratedGroup.get_all_sids(curated_group_id)},
count_only=True,
offset=offset,
limit=limit,
)
alert_count_per_sid = {}
for s in list(filter(lambda s: s.get('alertCount') > 0, students)):
sid = s.get('sid')
alert_count_per_sid[sid] = s.get('alertCount')
sids = list(alert_count_per_sid.keys())
benchmark('begin profile query')
students_with_alerts = get_student_profile_summaries(sids=sids)
benchmark('end profile query')
for student in students_with_alerts:
student['alertCount'] = alert_count_per_sid[student['sid']]
benchmark('end')
return tolerant_jsonify(students_with_alerts)
def test_curated_groups_by_sid(self, client, coe_advisor,
coe_advisor_groups):
"""API delivers accurate set of student SIDs."""
sids = CuratedGroup.get_all_sids(
curated_group_id=coe_advisor_groups[0].id)
sample_sid = sids[0]
assert self._api_my_curated_groups_by_sid(
client, sid=sample_sid) == [coe_advisor_groups[0].id]
def download_csv(curated_group_id):
benchmark = get_benchmarker(
f'curated group {curated_group_id} download_csv')
benchmark('begin')
curated_group = CuratedGroup.find_by_id(curated_group_id)
if not curated_group:
raise ResourceNotFoundError(
f'No curated group found with id: {curated_group_id}')
if curated_group.owner_id != current_user.get_id():
raise ForbiddenRequestError(
f'Current user, {current_user.get_uid()}, does not own curated group {curated_group.id}'
)
return response_with_students_csv_download(
sids=CuratedGroup.get_all_sids(curated_group_id), benchmark=benchmark)
def get_my_curated_groups():
curated_groups = []
user_id = current_user.get_id()
for curated_group in CuratedGroup.get_curated_groups_by_owner_id(user_id):
api_json = curated_group.to_api_json(include_students=False)
students = [{'sid': sid} for sid in CuratedGroup.get_all_sids(curated_group.id)]
students_with_alerts = Alert.include_alert_counts_for_students(
viewer_user_id=user_id,
group={'students': students},
count_only=True,
)
api_json['alertCount'] = sum(s['alertCount'] for s in students_with_alerts)
api_json['totalStudentCount'] = len(students)
curated_groups.append(api_json)
return curated_groups
def download_csv(curated_group_id):
benchmark = get_benchmarker(
f'curated group {curated_group_id} download_csv')
benchmark('begin')
curated_group = CuratedGroup.find_by_id(curated_group_id)
params = request.get_json()
fieldnames = get_param(params, 'csvColumnsSelected', [])
if not curated_group:
raise ResourceNotFoundError(
f'No curated group found with id: {curated_group_id}')
if not _can_current_user_view_curated_group(curated_group):
raise ForbiddenRequestError(
f'Current user, {current_user.get_uid()}, cannot view curated group {curated_group.id}'
)
return response_with_students_csv_download(
sids=CuratedGroup.get_all_sids(curated_group_id),
fieldnames=fieldnames,
benchmark=benchmark)
def get_my_curated_groups():
benchmark = get_benchmarker('my_curated_groups')
curated_groups = []
user_id = current_user.get_id()
for curated_group in CuratedGroup.get_curated_groups(owner_id=user_id):
students = [{
'sid': sid
} for sid in CuratedGroup.get_all_sids(curated_group.id)]
students_with_alerts = Alert.include_alert_counts_for_students(
benchmark=benchmark,
viewer_user_id=user_id,
group={'students': students},
count_only=True,
)
curated_groups.append({
**curated_group.to_api_json(include_students=False),
'alertCount':
sum(s['alertCount'] for s in students_with_alerts),
'sids': [student['sid'] for student in students],
'totalStudentCount':
len(students),
})
return curated_groups
def _get_sids_per_curated_groups(curated_group_ids=None):
sids = set()
for curated_group_id in curated_group_ids or ():
if curated_group_id:
sids = sids.union(CuratedGroup.get_all_sids(curated_group_id))
return sids