25 lines
585 B
Python
25 lines
585 B
Python
from django.db.models import QuerySet
|
|
from django.utils import timezone
|
|
|
|
|
|
class ProductQuerySet(QuerySet):
|
|
def available(self):
|
|
return self.filter(available_in__contains=timezone.now(), category__public=True)
|
|
|
|
|
|
class OrderQuerySet(QuerySet):
|
|
def not_cancelled(self):
|
|
return self.filter(cancelled=False)
|
|
|
|
def open(self):
|
|
return self.filter(open__isnull=False)
|
|
|
|
def paid(self):
|
|
return self.filter(paid=True)
|
|
|
|
def unpaid(self):
|
|
return self.filter(paid=False)
|
|
|
|
def cancelled(self):
|
|
return self.filter(cancelled=True)
|