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
|
'''
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
from pprint import pformat
import logging
from coip.apps.service.models import Service
class Membership(models.Model):
'''
Membership in a namespace/group
'''
user = models.ForeignKey(User,blank=True,null=True,related_name='user')
service = models.ForeignKey(Service,blank=True,null=True,related_name='service')
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_service(self):
return self.service != None
def add_member(name,userorservice,hidden=False):
if isinstance(userorservice,User):
(m,created) = Membership.objects.get_or_create(user=userorservice,name=name)
else:
(m,created) = Membership.objects.get_or_create(service=userorservice,name=name)
if created or not m.enabled or m.hidden != hidden:
m.enabled = True
m.hidden = hidden
m.save()
def disable_member(name,userorservice):
if isinstance(userorservice,User):
m = Membership.objects.get(name=name,user=userorservice)
else:
m = Membership.objects.get(name=name,service=userorservice)
if m:
m.enabled = False
m.save()
def remove_member(name,userorservice):
if isinstance(userorservice,User):
m = Membership.objects.get(name=name,user=userorservice)
else:
m = Membership.objects.get(name=name,service=userorservice)
if m:
m.delete()
def has_member(name,userorservice):
if isinstance(userorservice,User):
return Membership.objects.filter(name=name,user=userorservice)
else:
return Membership.objects.filter(name=name,service=userorservice)
|
