From a8c2c3c5b4352b7afeee4253007216a2c3d81bf0 Mon Sep 17 00:00:00 2001 From: Piotr Mitros Date: Sat, 10 Mar 2012 19:38:51 -0500 Subject: [PATCH] More helpers --- student/models.py | 12 ++++++++++++ 1 file changed, 12 insertions(+) diff --git a/student/models.py b/student/models.py index a36658dd30..024da56260 100644 --- a/student/models.py +++ b/student/models.py @@ -29,6 +29,8 @@ class UserProfile(models.Model): meta = models.CharField(blank=True, max_length=255) # JSON dictionary for future expansion courseware = models.CharField(blank=True, max_length=255, default='course.xml') +## TODO: Should be renamed to generic UserGroup, and possibly +# Given an optional field for type of group class UserTestGroup(models.Model): users = models.ManyToManyField(User, db_index=True) name = models.CharField(blank=False, max_length=32, db_index=True) @@ -91,3 +93,13 @@ def user_count(): def active_user_count(): return User.objects.filter(is_active = True).count() +def create_group(name, description): + utg = UserTestGroup() + utg.name = name + utg.description = description + utg.save() + +def add_user_to_group(group, user): + utg = UserTestGroup.objects.get(name = group) + utg.users.add(User.objects.get(username = user)) + utg.save()