datatracker/ietf/nomcom/managers.py
Henrik Levkowetz 5f2da7ba9e Merged [6486] from rjsparks@nostrum.com:
Slightly more than minimal changes to have questionnaire reminders go only to those nominees who have accepted but have not yet sent in a questionnaire for a given position
 - Legacy-Id: 6573
Note: SVN reference [6486] has been migrated to Git commit 2a1b4c7775
2013-10-30 21:17:46 +00:00

103 lines
2.6 KiB
Python

from django.db import models
from django.db.models.query import QuerySet
class MixinManager(object):
def __getattr__(self, attr, *args):
try:
return getattr(self.__class__, attr, *args)
except AttributeError:
return getattr(self.get_query_set(), attr, *args)
class NomineePositionQuerySet(QuerySet):
def get_by_nomcom(self, nomcom):
return self.filter(position__nomcom=nomcom)
def by_state(self, state):
return self.filter(state=state)
def accepted(self):
""" only accepted objects """
return self.by_state('accepted')
def pending(self):
""" only pending objects """
return self.by_state('pending')
def declined(self):
""" only declined objects """
return self.by_state('declined')
def not_duplicated(self):
return self.filter(nominee__duplicated__isnull=True)
def with_questionnaire_response(self):
return self.filter(nominee__feedback__type='questio',nominee__feedback__positions=models.F('position')).distinct()
def without_questionnaire_response(self):
with_ids = self.with_questionnaire_response().values_list('id', flat=True)
return self.exclude(id__in=with_ids)
class NomineePositionManager(models.Manager, MixinManager):
def get_query_set(self):
return NomineePositionQuerySet(self.model)
class NomineeManagerQuerySet(QuerySet):
def get_by_nomcom(self, nomcom):
return self.filter(nomcom=nomcom)
def not_duplicated(self):
return self.filter(duplicated__isnull=True)
class NomineeManager(models.Manager, MixinManager):
def get_query_set(self):
return NomineeManagerQuerySet(self.model)
class PositionQuerySet(QuerySet):
def get_by_nomcom(self, nomcom):
return self.filter(nomcom=nomcom)
def opened(self):
""" only opened positions """
return self.filter(is_open=True)
def closed(self):
""" only closed positions """
return self.filter(is_open=False)
class PositionManager(models.Manager, MixinManager):
def get_query_set(self):
return PositionQuerySet(self.model)
class FeedbackQuerySet(QuerySet):
def get_by_nomcom(self, nomcom):
return self.filter(nomcom=nomcom)
def by_type(self, type):
return self.filter(type=type)
def comments(self):
return self.by_type('comment')
def questionnaires(self):
return self.by_type('questio')
def nominations(self):
return self.by_type('nomina')
class FeedbackManager(models.Manager, MixinManager):
def get_query_set(self):
return FeedbackQuerySet(self.model)