30 lines
620 B
Python
30 lines
620 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=True)
|
|
|
|
def paid(self):
|
|
return self.filter(paid=True)
|
|
|
|
def unpaid(self):
|
|
return self.filter(paid=False)
|
|
|
|
def cancelled(self):
|
|
return self.filter(cancelled=True)
|