summaryrefslogtreecommitdiff
path: root/src/django_co_connector/models.py
blob: bb225e94c337e35cedca102e5bb8aeeadd86bf55 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
'''
Created on Apr 5, 2011

@author: leifj
'''

from django.db import models
from django.db.models.fields import CharField, URLField, DateTimeField, IntegerField
from django.contrib.auth.models import Group
from django.db.models.fields.related import OneToOneField
from django.dispatch.dispatcher import Signal
from django_co_connector.settings import CO_ATTRIBUTES
import logging

class GroupConnector(models.Model):
    attribute = CharField(max_length=1024)
    value = CharField(max_length=1024)
    activity_url = URLField(blank=True,null=True)
    membership_url = URLField(blank=True,null=True)
    ttl = IntegerField(blank=True,null=True)
    group = OneToOneField(Group,related_name='connector')
    modify_time = DateTimeField(auto_now=True)
    create_time = DateTimeField(auto_now_add=True)
    
    def __unicode__(self):
        return "%s=%s for %s" % (self.attribute,self.value,self.group.name)
    
    class Meta:
        unique_together = ('attribute','value')
    
    def fetch_updates(self):
        return ([],[])
    
    def fetch_all(self):
        return []
    
    def fetch_meta(self):
        return {}
    
    def update(self):
        # pull JSON to get display and ttl (?)
        (added,removed) = self.fetch_updates()
        for user in added:
            if not self.group in user.groups:
                add_member.send(sender=self.group,user=user)
                user.groups.append(self.group)
        for user in removed:
            if self.group in user.groups:     
                remove_member.send(sender=self.group,user=user)
                user.groups.remove(self.group)
    
add_member = Signal(providing_args=['user'])
remove_member = Signal(providing_args=['user'])

def co_import_from_request(request):
    for attribute in CO_ATTRIBUTES:
        logging.debug(attribute)
        values = request.META.get(attribute)                 
        if values and values != "(null)":
            logging.debug("%s: %s" % (attribute,values))
            co_import_av(request.user,attribute,values.split(';'))

def co_import_av(user,attribute,values):
    for value in values:
        gco = GroupConnector.objects.filter(attribute=attribute,value=value)
        if not gco:
            group,created = Group.objects.get_or_create(name=value)
            gco = GroupConnector.objects.create(attribute=attribute,value=value,group=group)
            
            meta = gco.fetch_meta()
            changed = False
            for attr in ('name','ttl','activity_url','membership_url'):
                if meta.has_key(attr):
                    setattr(gco,attr,meta.get(attr))
                    changed = True
            
            #members = gco.fetch_all()
            #for user in members:
            #    if not gco.group in user.groups:
            #        add_member.send(sender=gco.group,user=user)
            #        user.groups.apppend(gco.group)
            #        user.save()
            
            if changed:
                gco.save()
                gco.group.save()
    
    for gco in GroupConnector.objects.filter(attribute=attribute):
        if not gco.value in values:
            if gco.group in user.groups.all():
                remove_member.send(sender=gco.group,user=user)
                user.groups.remove(gco.group)
                user.save()
        else:
            if not gco.group in user.groups.all():
                add_member.send(sender=gco.group,user=user)
                user.groups.apppend(gco.group)
                user.save()