22 lines
751 B
Python
22 lines
751 B
Python
from django.conf import settings
|
|
from django.db import models
|
|
from django.db.models import Q
|
|
from django.utils import timezone
|
|
|
|
from organizations.managers import CurrentOrganizationManager
|
|
|
|
|
|
class MessageManager(models.Manager):
|
|
def get_for_now(self):
|
|
now = timezone.now()
|
|
return self.filter(enabled=True).filter(
|
|
Q(start_datetime__isnull=True) | Q(start_datetime__lte=now)
|
|
).filter(Q(end_datetime__isnull=True) | Q(end_datetime__gte=now))
|
|
|
|
|
|
class OrganizationMessageManager(MessageManager, CurrentOrganizationManager):
|
|
def get_queryset(self):
|
|
return super(OrganizationMessageManager, self).get_queryset().filter(
|
|
**{self._get_field_name() + '__id': settings.ORGANIZATION_ID}
|
|
)
|