web/ldt_utils/user/models.py
author ymh <ymh.work@gmail.com>
Tue, 08 Jun 2010 15:31:42 +0200
changeset 4 7c994c98d1df
parent 0 web/ldt/user/models.py@ecdfc63274bf
permissions -rw-r--r--
change ldt.ldt to ldt.ldt_utils

from django.db import models
from django.contrib.auth.models import User, Group, UserManager
from django.utils.translation import gettext as _
from django.contrib import admin
import datetime


class IriGroup(Group):
    description = models.TextField(null=True, blank=True)
    
    def __unicode__(self):
        return self.name

            
class ldt(User):
    irigroups = models.ManyToManyField(IriGroup, blank=True)
    
    class Meta:
        verbose_name = 'iri user'
        verbose_name_plural = 'iri users'
    
    def __unicode__(self):
        return self.username
        

class ldtManager(UserManager):
    def create_user(self, username, email, password=None):
        "Creates and saves a User with the given username, e-mail and password."
        now = datetime.datetime.now()
        user = ldt(None, username, '', '', email.strip().lower(), 'placeholder', False, True, False, now, now)
        if password:
            user.set_password(password)
        else:
            user.set_unusable_password()
        user.save()
        return user   

User.objects = ldtManager()
User.objects.contribute_to_class(User, "objects")

admin.site.register(ldt)
admin.site.register(IriGroup)