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)