Merge pull request #54 from mikefranklin/master

add waitlisted property
pinax · May 11, 2017 · c3f2dae · c3f2dae
2 parents 5bdccb3 + 0d7732a
commit c3f2dae
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pinax/teams/models.py b/pinax/teams/models.py
@@ -90,6 +90,10 @@ def declines(self):
     def rejections(self):
         return self.memberships.filter(state=BaseMembership.STATE_REJECTED)
 
+    @property
+    def waitlisted(self):
+        return self.memberships.filter(state=BaseMembership.STATE_WAITLISTED)
+
     @property
     def acceptances(self):
         return self.memberships.filter(state__in=[
@@ -229,6 +233,7 @@ class BaseMembership(models.Model):
     STATE_DECLINED = "declined"
     STATE_REJECTED = "rejected"
     STATE_ACCEPTED = "accepted"
+    STATE_WAITLISTED = "waitlisted"
     STATE_AUTO_JOINED = "auto-joined"
 
     ROLE_MEMBER = "member"
@@ -241,6 +246,7 @@ class BaseMembership(models.Model):
         (STATE_DECLINED, _("declined")),
         (STATE_REJECTED, _("rejected")),
         (STATE_ACCEPTED, _("accepted")),
+        (STATE_WAITLISTED, _("waitlisted")),
         (STATE_AUTO_JOINED, _("auto joined"))
     ]