66 lines
1.7 KiB
Python
66 lines
1.7 KiB
Python
import datetime
|
|
|
|
from django.db import models
|
|
|
|
import reversion
|
|
|
|
from django.contrib.auth.models import Permission, User
|
|
|
|
|
|
TEAM_ACCESS_CHOICES = [
|
|
("open", "open"),
|
|
("application", "by application"),
|
|
("invitation", "by invitation")
|
|
]
|
|
|
|
|
|
class Team(models.Model):
|
|
|
|
slug = models.SlugField(unique=True)
|
|
name = models.CharField(max_length=100)
|
|
description = models.TextField(blank=True)
|
|
access = models.CharField(max_length=20, choices=TEAM_ACCESS_CHOICES)
|
|
permissions = models.ManyToManyField(Permission, blank=True)
|
|
created = models.DateTimeField(default=datetime.datetime.now, editable=False)
|
|
|
|
def __unicode__(self):
|
|
return self.name
|
|
|
|
def get_state_for_user(self, user):
|
|
try:
|
|
return self.memberships.get(user=user).state
|
|
except Membership.DoesNotExist:
|
|
return None
|
|
|
|
def applicants(self):
|
|
return self.memberships.filter(state="applied")
|
|
|
|
def invitees(self):
|
|
return self.memberships.filter(state="invited")
|
|
|
|
def members(self):
|
|
return self.memberships.filter(state="member")
|
|
|
|
def managers(self):
|
|
return self.memberships.filter(state="manager")
|
|
|
|
|
|
MEMBERSHIP_STATE_CHOICES = [
|
|
("applied", "applied"),
|
|
("invited", "invited"),
|
|
("declined", "declined"),
|
|
("rejected", "rejected"),
|
|
("member", "member"),
|
|
("manager", "manager"),
|
|
]
|
|
|
|
|
|
class Membership(models.Model):
|
|
|
|
user = models.ForeignKey(User, related_name="memberships")
|
|
team = models.ForeignKey(Team, related_name="memberships")
|
|
state = models.CharField(max_length=20, choices=MEMBERSHIP_STATE_CHOICES)
|
|
message = models.TextField(blank=True)
|
|
|
|
|
|
reversion.register(Membership)
|