OpenSlides/openslides/participant/models.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
    openslides.participant.models
    ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

    Models for the participant app.

    :copyright: 2011 by the OpenSlides team, see AUTHORS.
    :license: GNU GPL, see LICENSE for more details.
"""

from django.db import models
from django.db.models import Q
from django.contrib.auth.models import User
from django.utils.translation import ugettext as _

from participant.api import gen_password

class Profile(models.Model):
    GENDER_CHOICES = (
        ('none', _('Not specified')),
        ('male', _('Male')),
        ('female', _('Female')),
    )
    TYPE_CHOICE = (
        ('delegate', _('Delegate')),
        ('guest', _('Guest')),
        ('observer', _('Observer')),
        ('staff', _('Staff')),
    )

    user = models.OneToOneField(User, unique=True, editable=False)
    gender = models.CharField(max_length=50, choices=GENDER_CHOICES, default='none', verbose_name = _("Gender"))
    group = models.CharField(max_length=100, null=True, blank=True, verbose_name = _("Group"))
    type = models.CharField(max_length=100, choices=TYPE_CHOICE, default='delegate', verbose_name = _("Typ"))
    committee = models.CharField(max_length=100, null=True, blank=True, verbose_name = _("Committee"))
    firstpassword = models.CharField(max_length=100, null=True, blank=True, verbose_name = _("First Password"))


    def reset_password(self):
        self.user.set_password(self.firstpassword)

    def __unicode__(self):
        if self.group:
            return "%s (%s)" % (self.user.get_full_name(), self.group)
        return "%s" % self.user.get_full_name()


    class Meta:
        permissions = (
            ('can_see_participant', "Can see participant"),
            ('can_manage_participant', "Can manage participant"),
        )

def set_first_user_passwords():
    count = 0
    for user in Profile.objects.filter(Q(firstpassword='') | Q(firstpassword__isnull=True)):
        count = count + 1
        user.firstpassword = gen_password()
        user.user.set_password(user.firstpassword)
        user.user.save()
        user.save()
    return count