← Back to branch summary

~azzar1/unity/add-show-desktop-key

« back to all changes in this revision

Viewing changes to ivle/database.py

Fix ProjectView's total assigned count by dealing with ResultSets only.

Show diffs side-by-side

added added

removed removed

Lines of Context:
ivle/database.py
379
379
 
380
380
        return enrolment
381
381
 
 
382
    def get_members_by_role(self, role):
 
383
        return Store.of(self).find(User,
 
384
                Enrolment.user_id == User.id,
 
385
                Enrolment.offering_id == self.id,
 
386
                Enrolment.role == role
 
387
                )
 
388
 
382
389
    def get_students(self):
383
 
        enrolments = self.enrolments.find(role=u'student')
384
 
        return [enrolment.user for enrolment in enrolments]
 
390
        return self.get_members_by_role(u'student')
385
391
 
386
392
class Enrolment(Storm):
387
393
    """An enrolment of a user in an offering.