gistfile1.txt

from django.core.management.base import BaseCommand
from django.contrib.auth.models import Group
from events.models import Event
from role.models import Role


class Command(BaseCommand):
    """
    This is a one time command that updates an event's general users to groups
    """

    def handle(self, *args, **options):
        # get AEPA2012
        aepa = Event.objects.get(pk=6)
        tf = Event.objects.get(pk=2)

        # get all the users for aepa and add them to group
        aepa_role = Role.objects.get(event=aepa, roletype__name='General')
        aepa_group = Group.objects.get(name='%s: Delegate' % aepa.name)
        for profile in aepa_role.profiles.all():
            profile.user.groups.add(aepa_group)
            print profile.user

        # get all the users for aepa and add them to group
        tf_role = Role.objects.get(event=tf, roletype__name='General')
        tf_group = Group.objects.get(name='%s: Delegate' % tf.name)
        for profile in tf_role.profiles.all():
            profile.user.groups.add(tf_group)
            print profile.user