add NoteManager and NoteQuerySet, and basic tests

This commit is contained in:
Arthur Gerbaud
2016-08-07 11:50:01 +02:00
parent 2a8082e5b1
commit 9646c9f8ab
3 changed files with 129 additions and 6 deletions

View File

@@ -1,9 +1,32 @@
from django.db.models import Manager
from django.db.models.query import QuerySet
class NoteQuerySet(QuerySet):
def _ordered_by_field(self, field, reverse=False):
return self.order_by( '%s%s' % ( "-" if reverse else "",
field
)
)
def by_date(self, reverse=False):
return self._ordered_by_field('created_date', reverse=reverse)
def by_time(self, reverse=False):
return self._ordered_by_field('created_time', reverse=reverse)
class NoteManager(Manager):
def by_date(self):
return self.get_queryset().order_by('created_date')
def get_queryset(self):
return NoteQuerySet(self.model)
get_query_set = get_queryset
def by_date(self, **kwargs):
return self.get_queryset().by_date(**kwargs)
def by_time(self, **kwargs):
return self.get_queryset().by_time(**kwargs)
def by_time(self):
return self.get_queryset().order_by('created_time')