from psycopg2.extras import DateTimeTZRange

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)