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)