models.py 5.25 KB
Newer Older
1
from itertools import chain
2 3
from logging import getLogger

4
from django.conf import settings
5
from django.contrib.auth.models import User, Group
6
from django.contrib.auth.signals import user_logged_in
7
from django.db.models import (
8
    Model, ForeignKey, OneToOneField, CharField, IntegerField, TextField
9
)
10 11 12 13 14 15
from django.template.loader import render_to_string
from django.utils.translation import ugettext_lazy as _, override

from model_utils.models import TimeStampedModel
from model_utils.fields import StatusField
from model_utils import Choices
16 17

from vm.models import Instance
18
from acl.models import AclBase
19

20 21
logger = getLogger(__name__)

22

23 24 25
class Favourite(Model):
    instance = ForeignKey(Instance)
    user = ForeignKey(User)
26 27


28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
class Notification(TimeStampedModel):
    STATUS = Choices(('new', _('new')),
                     ('delivered', _('delivered')),
                     ('read', _('read')))

    status = StatusField()
    to = ForeignKey(User)
    subject = CharField(max_length=128)
    message = TextField()

    class Meta:
        ordering = ['-created']

    @classmethod
    def send(cls, user, subject, template, context={}):
        try:
            language = user.profile.preferred_language
        except:
            language = None
        with override(language):
            context['user'] = user
            rendered = render_to_string(template, context)
            subject = unicode(subject)
        return cls.objects.create(to=user, subject=subject, message=rendered)


54 55 56 57 58 59 60 61 62
class Profile(Model):
    user = OneToOneField(User)
    preferred_language = CharField(verbose_name=_('preferred language'),
                                   choices=settings.LANGUAGES,
                                   max_length=32,
                                   default=settings.LANGUAGE_CODE, blank=False)
    org_id = CharField(  # may be populated from eduPersonOrgId field
        unique=True, blank=True, null=True, max_length=64,
        help_text=_('Unique identifier of the person, e.g. a student number.'))
63
    instance_limit = IntegerField(default=5)
64

65 66 67
    def notify(self, subject, template, context={}):
        return Notification.send(self.user, subject, template, context)

68

69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89
class GroupProfile(AclBase):
    ACL_LEVELS = (
        ('operator', _('operator')),
        ('owner', _('owner')),
    )

    group = OneToOneField(Group)
    org_id = CharField(
        unique=True, blank=True, null=True, max_length=64,
        help_text=_('Unique identifier of the group at the organization.'))
    description = TextField()

    @classmethod
    def search(cls, name):
        try:
            return cls.objects.get(org_id=name).group
        except cls.DoesNotExist:
            return Group.objects.get(name=name)


def get_or_create_profile(self):
90
    obj, created = GroupProfile.objects.get_or_create(group_id=self.pk)
91 92 93 94 95
    return obj

Group.profile = property(get_or_create_profile)


96
def create_profile(sender, user, request, **kwargs):
97 98
    if not user.pk:
        return False
99 100 101 102 103
    profile, created = Profile.objects.get_or_create(user=user)
    return created

user_logged_in.connect(create_profile)

104
if hasattr(settings, 'SAML_ORG_ID_ATTRIBUTE'):
105
    logger.debug("Register save_org_id to djangosaml2 pre_user_save")
106 107
    from djangosaml2.signals import pre_user_save

108
    def save_org_id(sender, **kwargs):
109
        logger.debug("save_org_id called by %s", sender.username)
110
        attributes = kwargs.pop('attributes')
111
        atr = settings.SAML_ORG_ID_ATTRIBUTE
112 113 114 115 116 117
        try:
            value = attributes[atr][0]
        except Exception as e:
            value = None
            logger.info("save_org_id couldn't find attribute. %s", unicode(e))

118 119 120 121
        if sender.pk is None:
            sender.save()
            logger.debug("save_org_id saved user %s", unicode(sender))

122 123
        profile, created = Profile.objects.get_or_create(user=sender)
        if created or profile.org_id != value:
124 125
            logger.info("org_id of %s added to user %s's profile",
                        value, sender.username)
126 127
            profile.org_id = value
            profile.save()
128 129 130
        else:
            logger.debug("org_id of %s already added to user %s's profile",
                         value, sender.username)
131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153
        memberatrs = getattr(settings, 'SAML_GROUP_ATTRIBUTES', [])
        for group in chain(*[attributes[i] for i in memberatrs]):
            try:
                g = GroupProfile.search(group)
            except Group.DoesNotExist:
                logger.debug('cant find membergroup %s', group)
            else:
                logger.debug('could find membergroup %s (%s)',
                             group, unicode(g))
                g.user_set.add(sender)

        owneratrs = getattr(settings, 'SAML_GROUP_OWNER_ATTRIBUTES', [])
        for group in chain(*[attributes[i] for i in owneratrs]):
            try:
                g = GroupProfile.search(group)
            except Group.DoesNotExist:
                logger.debug('cant find ownergroup %s', group)
            else:
                logger.debug('could find ownergroup %s (%s)',
                             group, unicode(g))
                g.profile.set_level(sender, 'owner')

        return False  # User did not change
154

155 156
    pre_user_save.connect(save_org_id)

157 158
else:
    logger.debug("Do not register save_org_id to djangosaml2 pre_user_save")