summaryrefslogtreecommitdiff
path: root/coip/apps/membership/models.py
blob: b2895130c9a4a056ba17106bcfa461c2bec3bbf5 (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
99
100
101
102
'''
Created on Jun 23, 2010

@author: leifj
'''
from django.db import models
from django.contrib.auth.models import User
from coip.apps.name.models import Name
import datetime
import logging
from coip.apps.entity.models import Entity
import tagging
from django.core.mail import send_mail
from coip.settings import NOREPLY

class Membership(models.Model):
    '''
    Membership in a namespace/group
    '''
    user = models.ForeignKey(User,blank=True,null=True,related_name='user')
    entity = models.ForeignKey(Entity,blank=True,null=True,related_name='entity')
    name = models.ForeignKey(Name,related_name='memberships')
    enabled = models.BooleanField()
    hidden = models.BooleanField()
    timecreated = models.DateTimeField(auto_now_add=True)
    lastupdated = models.DateTimeField(auto_now=True)
    expires = models.DateTimeField(blank=True,null=True)
    
    def __unicode__(self):
        return "%s in %s" % (self.user,self.name)
    
    def valid(self):
        return self.enabled and datetime.date.today() > self.expires
    
    def status(self):
        if self.valid():
            return "active"
        else:
            return "inactive";
        
    def is_user(self):
        return self.user != None
    
    def is_entity(self):
        return self.entity != None
    
    def send_notification(self,what):
        if not self.user or not self.user.email:
            return
                
        send_mail('You have been %s \'%s\'' % (what,self.name.short),
                  '''
You have been %s \'%s\'.

To view information about \'%s\' open this link in your browser:
%s

''' % (what,self.name.shortname(),self.name.shortname(),self.name.url()),
                  NOREPLY,
                  [self.user.email], 
                  fail_silently=False)
        return
    
def add_member(name,member_name,hidden=False):
    if isinstance(member_name,User):
        (m,created)  = Membership.objects.get_or_create(user=member_name,name=name)
    else:
        (m,created)  = Membership.objects.get_or_create(entity=member_name,name=name)
        
    if created or not m.enabled or m.hidden != hidden:
        m.enabled = True
        m.hidden = hidden
        m.save()
    
    return m.send_notification("added to")
        
def disable_member(name,member_name):
    if isinstance(member_name,User):
        m = Membership.objects.get(name=name,user=member_name)
    else:
        m = Membership.objects.get(name=name,entity=member_name)
    if m:
        m.enabled = False
        m.save()
        m.send_notification("temporarily removed from")
        
def remove_member(name,member_name):
    if isinstance(member_name,User):
        m = Membership.objects.get(name=name,user=member_name)
    else:
        m = Membership.objects.get(name=name,entity=member_name)
    if m:
        m.send_notification("removed from")
        m.delete()
        
def has_member(name,member_name):
    if isinstance(member_name,User):
        return Membership.objects.filter(name=name,user=member_name)
    else:
        return Membership.objects.filter(name=name,entity=member_name)

tagging.register(Membership)